WIP test more models

huggingface · Oct 7, 2024 · 38e4e28 · 38e4e28
1 parent 341d950
commit 38e4e28
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" --run-slow