[CI/Build] Try to make beam search test less flaky (#34885)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2026-02-19 19:16:58 +08:00
parent 1391378861
commit 23210a911e
1 changed files with 1 additions and 1 deletions
--- a/tests/samplers/test_beam_search.py
+++ b/tests/samplers/test_beam_search.py
@@ -27,7 +27,7 @@ EXTRA_ENGINE_KWARGS: dict = (
        max_num_seqs=1,
    )
    if current_platform.is_rocm()
-    else {}
+    else dict(async_scheduling=False, max_num_seqs=1)
 )

 # FIXME(zhuohan): The test can not pass if we: