From b6fe0cca9901f917a0b1c60b662d6afe4015486d Mon Sep 17 00:00:00 2001
From: Liangsheng Yin <hnyls2002@gmail.com>
Date: Tue, 31 Mar 2026 17:12:20 -0700
Subject: [PATCH] Switch MooncakeSpec to EAGLE3 + Llama-3.1 (#21794)

---
 .../disaggregation/test_disaggregation_basic.py     | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/test/registered/disaggregation/test_disaggregation_basic.py b/test/registered/disaggregation/test_disaggregation_basic.py
index 6b621e3e0..aaaffdee2 100644
--- a/test/registered/disaggregation/test_disaggregation_basic.py
+++ b/test/registered/disaggregation/test_disaggregation_basic.py
@@ -13,9 +13,9 @@ from sglang.test.server_fixtures.disaggregation_fixture import (
     PDDisaggregationServerBase,
 )
 from sglang.test.test_utils import (
-    DEFAULT_DRAFT_MODEL_EAGLE,
+    DEFAULT_DRAFT_MODEL_EAGLE3,
     DEFAULT_MODEL_NAME_FOR_TEST,
-    DEFAULT_TARGET_MODEL_EAGLE,
+    DEFAULT_TARGET_MODEL_EAGLE3,
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,
     popen_launch_pd_server,
 )
@@ -291,8 +291,8 @@ class TestDisaggregationMooncakeSpec(PDDisaggregationServerBase):
     @classmethod
     def setUpClass(cls):
         super().setUpClass()
-        cls.model = DEFAULT_TARGET_MODEL_EAGLE
-        cls.draft_model = DEFAULT_DRAFT_MODEL_EAGLE
+        cls.model = DEFAULT_TARGET_MODEL_EAGLE3
+        cls.draft_model = DEFAULT_DRAFT_MODEL_EAGLE3
         cls.spec_args = [
             "--speculative-algorithm",
             "EAGLE",
@@ -306,6 +306,7 @@ class TestDisaggregationMooncakeSpec(PDDisaggregationServerBase):
             "16",
             "--cuda-graph-max-bs",
             "8",
+            "--dtype=float16",
         ]
         print(f"{cls.base_host=} {cls.lb_port=} {cls.prefill_port=} {cls.decode_port=}")
 
@@ -365,14 +366,14 @@ class TestDisaggregationMooncakeSpec(PDDisaggregationServerBase):
             data_path=None,
             num_questions=200,
             max_new_tokens=512,
-            parallel=2,
+            parallel=128,
             host=f"http://{self.base_host}",
             port=int(self.lb_port),
         )
         metrics = run_eval_few_shot_gsm8k(args)
         print(f"Evaluation metrics: {metrics}")
 
-        self.assertGreater(metrics["accuracy"], 0.20)
+        self.assertGreater(metrics["accuracy"], 0.74)
 
 
 class TestDisaggregationSimulatedRetract(PDDisaggregationServerBase):