Fix oom error for large page size (#4913)

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
2025-03-30 21:34:21 -07:00
parent 4a63bc32b7
commit a169b9f813
2 changed files with 21 additions and 13 deletions
--- a/test/srt/test_eagle_infer.py
+++ b/test/srt/test_eagle_infer.py
@@ -144,7 +144,7 @@ class TestEAGLEEngine(CustomTestCase):
        if engine.server_args.model_path == DEFAULT_EAGLE_TARGET_MODEL_FOR_TEST:
            self.assertGreater(acc_length, 3.6)
        else:
-            self.assertGreater(acc_length, 2.6)
+            self.assertGreater(acc_length, 2.5)


 class TestEAGLEEngineTokenMap(TestEAGLEEngine):