From 7e1a8aa249c74e52525832f93276ee9edd054910 Mon Sep 17 00:00:00 2001 From: Benjamin Bossan Date: Fri, 17 Nov 2023 15:23:09 +0100 Subject: [PATCH] Fix torch install in HF integratin test workflow For some reason, the existing install instruction started failing recently. Try a different one akin to what is used for the unit tests. --- .github/workflows/test-hf-integration.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test-hf-integration.yml b/.github/workflows/test-hf-integration.yml index be1e967f7..0327f915a 100644 --- a/.github/workflows/test-hf-integration.yml +++ b/.github/workflows/test-hf-integration.yml @@ -20,7 +20,8 @@ jobs: - name: Install Requirements run: | python -m pip install -r requirements.txt - python -m pip install transformers tokenizers huggingface_hub torch + python -m pip install torch -f https://download.pytorch.org/whl/torch_stable.html + python -m pip install transformers tokenizers huggingface_hub python -m pip install . - name: run HF integration test env: