WIP Investigate failure

huggingface · Oct 7, 2024 · 2823fea · 2823fea
1 parent 29ab034
commit 2823fea
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi-jetstream.yml b/.github/workflows/test-pytorch-xla-tpu-tgi-jetstream.yml
@@ -29,4 +29,6 @@ jobs:
 
       - name: Build and test TGI server
         run: |
-          HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} make tgi_test_jetstream
+          make test_installs jetstream_requirements 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 "jetstream and TinyLlama"