WIP

huggingface · Sep 11, 2024 · a66154a · a66154a
1 parent ff1b371
commit a66154a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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