fix eagle unit test (#3591)

2025-02-15 23:10:48 +08:00
parent 7711ac6ed0
commit 6718b10996
1 changed files with 2 additions and 0 deletions
--- a/test/srt/test_bench_serving.py
+++ b/test/srt/test_bench_serving.py
@@ -153,6 +153,8 @@ class TestBenchServing(unittest.TestCase):
                "64",
                "--mem-fraction-static",
                "0.7",
+                "--cuda-graph-max-bs",
+                "32",
            ],
        )