Skip to content

Commit f15e70d

Browse files
robertgshaw2-redhatRobert Shaw
and
Robert Shaw
authored
[TPU] Switch Test to Non-Sliding Window (#15981)
Signed-off-by: Robert Shaw <[email protected]> Co-authored-by: Robert Shaw <[email protected]>
1 parent b6be6f8 commit f15e70d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tpu/test_compilation.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def test_tpu_compilation():
3030
n=N,
3131
max_tokens=16)
3232

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

0 commit comments

Comments
 (0)