From a66154a2cdf60e5a90503a14b98fa0d18a22678d Mon Sep 17 00:00:00 2001 From: Alvaro Moran Date: Wed, 11 Sep 2024 11:21:56 +0000 Subject: [PATCH] WIP --- .github/workflows/test-pytorch-xla-tpu-tgi.yml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi.yml b/.github/workflows/test-pytorch-xla-tpu-tgi.yml index 398cd931..1588066a 100644 --- a/.github/workflows/test-pytorch-xla-tpu-tgi.yml +++ b/.github/workflows/test-pytorch-xla-tpu-tgi.yml @@ -28,6 +28,6 @@ jobs: - name: quick test run: | - python -m pip install huggingface_hub - HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='google/gemma-2b', allow_patterns=['*.json','model*.safetensors'])" - + make test_installs tgi_server + find text-generation-inference/ -name "text_generation_server-*whl" -exec python -m pip install {} \; + HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} python -m pytest -sv text-generation-inference/tests -k gemma-2b