Fix test of flashinfer mla with nextn (#4237)

9dfafa74 · Baizhou Zhang · GitHub · f1d09a65 · 9dfafa74
Unverified Commit 9dfafa74 authored Mar 09, 2025 by Baizhou Zhang Committed by GitHub Mar 09, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/srt/test_mla_flashinfer.py test/srt/test_mla_flashinfer.py +1 -1

No files found.
--- 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",