updated

vllm-project · Nov 3, 2024 · 29e7c31 · 29e7c31
1 parent f833512
commit 29e7c31
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/entrypoints/openai/test_accuracy.py b/tests/entrypoints/openai/test_accuracy.py
@@ -30,7 +30,10 @@
 MAX_WAIT_SECONDS = None
 
 if current_platform.is_tpu():
-    MORE_ARGS_LIST = [[]]
+    MORE_ARGS_LIST = [
+        [], # Default
+        # ["--num-scheduler-steps", "8"], # Multi-step << currently fails
+    ]
     MAX_WAIT_SECONDS = 600