diff --git a/.github/workflows/test-pytorch-xla-tpu-tgi-jetstream.yml b/.github/workflows/test-pytorch-xla-tpu-tgi-jetstream.yml index 9f45f1e..0819532 100644 --- 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