From 163fafba5162af9bedeaef67116eecdc910c30d1 Mon Sep 17 00:00:00 2001
From: Hanming Lu <69857889+hanming-lu@users.noreply.github.com>
Date: Fri, 5 Jun 2026 23:26:32 -0700
Subject: [PATCH] fix test_qwen3_next_models flaky (#27419)

---
 test/registered/models_e2e/test_qwen3_next_models.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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