From f15e70d906aca5bb0e5ca7e4f1f857853ce5dd66 Mon Sep 17 00:00:00 2001 From: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com> Date: Thu, 3 Apr 2025 14:28:45 -0700 Subject: [PATCH] [TPU] Switch Test to Non-Sliding Window (#15981) Signed-off-by: Robert Shaw Co-authored-by: Robert Shaw --- tests/tpu/test_compilation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/tpu/test_compilation.py b/tests/tpu/test_compilation.py index 27328d4542d9a..2a71f460f78ef 100644 --- 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,