Skip to content

GitLab

Unverified Commit 23210a91 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

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


Signed-off-by: default avatarDarkLight1337 <tlleungac@connect.ust.hk>
parent 13913788
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/samplers/test_beam_search.py
View file @ 23210a91
...@@ -27,7 +27,7 @@ EXTRA_ENGINE_KWARGS: dict = ( ...@@ -27,7 +27,7 @@ EXTRA_ENGINE_KWARGS: dict = (
max_num_seqs=1, max_num_seqs=1,
) )
if current_platform.is_rocm() if current_platform.is_rocm()
else {} else dict(async_scheduling=False, max_num_seqs=1)
) )
# FIXME(zhuohan): The test can not pass if we: # FIXME(zhuohan): The test can not pass if we:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment