Fix test of flashinfer mla with nextn (#4237)

2025-03-09 12:45:39 -07:00
parent f1d09a6541
commit 9dfafa743c
1 changed files with 1 additions and 1 deletions
--- a/test/srt/test_mla_flashinfer.py
+++ b/test/srt/test_mla_flashinfer.py
@@ -121,7 +121,7 @@ class TestFlashinferMLAMTP(unittest.TestCase):
                    "--speculative-draft",
                    "lmsys/sglang-ci-dsv3-test-NextN",
                    "--speculative-num-steps",
-                    "4",
+                    "3",
                    "--speculative-eagle-topk",
                    "1",
                    "--speculative-num-draft-tokens",