diff --git a/test/registered/models_e2e/test_qwen3_next_models.py b/test/registered/models_e2e/test_qwen3_next_models.py index ed64ec962..809fdd9d9 100644 --- a/test/registered/models_e2e/test_qwen3_next_models.py +++ b/test/registered/models_e2e/test_qwen3_next_models.py @@ -40,7 +40,7 @@ class TestQwen3NextLazyExtraBuffer( model = QWEN3_NEXT_MODEL cache_chunk_size = 64 gsm8k_accuracy_thres = 0.93 - kl_div_thres = 0.001 + kl_div_thres = 0.002 other_args = _make_args(page_size=1, track_interval=2) @@ -50,7 +50,7 @@ class TestQwen3NextLazyExtraBufferLargePage( model = QWEN3_NEXT_MODEL cache_chunk_size = 64 gsm8k_accuracy_thres = 0.93 - kl_div_thres = 0.001 + kl_div_thres = 0.002 other_args = _make_args(page_size=2, track_interval=2) @@ -61,7 +61,7 @@ class TestQwen3NextLazyExtraBufferAllocFail( model = QWEN3_NEXT_MODEL cache_chunk_size = 64 gsm8k_accuracy_thres = 0.93 - kl_div_thres = 0.001 + kl_div_thres = 0.002 other_args = _make_args(page_size=1, track_interval=2) @classmethod @@ -84,7 +84,7 @@ class TestQwen3NextLazyExtraBufferLargePageAllocFail( model = QWEN3_NEXT_MODEL cache_chunk_size = 64 gsm8k_accuracy_thres = 0.93 - kl_div_thres = 0.001 + kl_div_thres = 0.002 other_args = _make_args(page_size=2, track_interval=2) @classmethod