[TPU] Switch Test to Non-Sliding Window (#15981)

Signed-off-by: Robert Shaw <robshaw@redhat.com> Co-authored-by: Robert Shaw <robshaw@redhat.com>
2025-04-03 14:28:45 -07:00
parent b6be6f8d1e
commit f15e70d906
1 changed files with 1 additions and 1 deletions
--- a/tests/tpu/test_compilation.py
+++ b/tests/tpu/test_compilation.py
@@ -30,7 +30,7 @@ def test_tpu_compilation():
                                         n=N,
                                         max_tokens=16)

-        llm = LLM(model="Qwen/Qwen2.5-1.5B-Instruct",
+        llm = LLM(model="Qwen/Qwen2-1.5B-Instruct",
                  max_num_batched_tokens=256,
                  max_model_len=256,
                  max_num_seqs=32,