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: