huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 11:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

enable hf_transfer on tests

Browse Source
This commit is contained in:
OlivierDehaene 2023-03-03 17:44:50 +01:00
parent 5862add4d0
commit c02a5475ea
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/tests.yaml vendored
View File

@ -60,7 +60,7 @@ jobs:
- name: Run server tests - name: Run server tests
run: | run: |
pip install pytest pip install pytest
pytest -sv server/tests HF_HUB_ENABLE_HF_TRANSFER=1 pytest -sv server/tests
- name: Run Rust tests - name: Run Rust tests
run: | run: |
cargo test cargo test

2
Makefile
View File

@ -19,7 +19,7 @@ integration-tests: install-router install-launcher
cargo test cargo test
python-tests: python-tests:
cd server && pytest tests cd server && HF_HUB_ENABLE_HF_TRANSFER=1 pytest tests
run-bloom-560m: run-bloom-560m:
text-generation-launcher --model-id bigscience/bloom-560m --num-shard 2 text-generation-launcher --model-id bigscience/bloom-560m --num-shard 2