diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml index 27caff11..476ae443 100644 --- a/.github/workflows/tests.yaml +++ b/.github/workflows/tests.yaml @@ -60,7 +60,7 @@ jobs: - name: Run server tests run: | pip install pytest - pytest -sv server/tests + HF_HUB_ENABLE_HF_TRANSFER=1 pytest -sv server/tests - name: Run Rust tests run: | cargo test diff --git a/Makefile b/Makefile index 08217730..cf7b8675 100644 --- a/Makefile +++ b/Makefile @@ -19,7 +19,7 @@ integration-tests: install-router install-launcher cargo test python-tests: - cd server && pytest tests + cd server && HF_HUB_ENABLE_HF_TRANSFER=1 pytest tests run-bloom-560m: text-generation-launcher --model-id bigscience/bloom-560m --num-shard 2