Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Avoiding timeout for bloom tests. (#2693) 

* Avoiding timeout for bloom tests.

* Skip the test let's see if it's always the first tests that fails.

* Fail early.

* Pulling ?

* No early exit.
This commit is contained in:
Nicolas Patry 2024-10-26 05:35:28 +02:00 committed by GitHub
parent a6b02da971
commit 8a8794a672
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/build.yaml vendored
View File

@ -202,4 +202,5 @@ jobs:
export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}" export EXTRA_PYTEST="${{ needs.build-and-push.outputs.extra_pytest }}"
export HF_TOKEN=${{ secrets.HF_TOKEN }} export HF_TOKEN=${{ secrets.HF_TOKEN }}
echo $DOCKER_IMAGE echo $DOCKER_IMAGE
docker pull $DOCKER_IMAGE
pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST} pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST}

2
integration-tests/models/test_bloom_560m.py
View File

@ -3,7 +3,7 @@ import pytest
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
def bloom_560_handle(launcher): def bloom_560_handle(launcher):
with launcher("bigscience/bloom-560m") as handle: with launcher("bigscience/bloom-560m", num_shard=1) as handle:
yield handle yield handle