From 7cfea0df390c154c1026f77d3682e2733ca4aca8 Mon Sep 17 00:00:00 2001 From: QiliangCui Date: Fri, 25 Jul 2025 13:22:01 -0700 Subject: [PATCH] [TPU][Test] Rollback PR-21550. (#21619) Signed-off-by: Qiliang Cui --- tests/v1/tpu/test_basic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/v1/tpu/test_basic.py b/tests/v1/tpu/test_basic.py index dd89059ded524..865b58bc7f4b0 100644 --- a/tests/v1/tpu/test_basic.py +++ b/tests/v1/tpu/test_basic.py @@ -59,7 +59,7 @@ def test_basic( # actually test chunked prompt max_num_batched_tokens=1024, max_model_len=8192, - gpu_memory_utilization=0.95, + gpu_memory_utilization=0.7, max_num_seqs=max_num_seqs, tensor_parallel_size=tensor_parallel_size) as vllm_model: vllm_outputs = vllm_model.generate_greedy(example_prompts,