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,