WIP test more models

huggingface · Oct 7, 2024 · b8a798c · b8a798c
1 parent 341d950
commit b8a798c
Showing 1 changed file with 1 addition 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
@@ -32,4 +32,4 @@ jobs:
           make test_installs jetstream_requirements tgi_server
           find text-generation-inference/ -name "text_generation_server-*whl" -exec python -m pip install {} \;
           JETSTREAM_PT=1 HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_TPU_CI }} \
-            python -m pytest -sv text-generation-inference/tests -k "jetstream and TinyLlama and greedy"
+            python -m pytest -sv text-generation-inference/tests -k "jetstream and greedy" --runslow