From b051d76dabb8ed80bd997d48fb451442062f3bc5 Mon Sep 17 00:00:00 2001
From: Charles Chen <pychen96@gmail.com>
Date: Sat, 15 Nov 2025 01:36:03 -0800
Subject: [PATCH] Fix: add missing get_embed_and_head in MiniMax M2 for Eagle3
 (#13297)

---
 python/sglang/srt/models/minimax_m2.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/sglang/srt/models/minimax_m2.py b/python/sglang/srt/models/minimax_m2.py
index 2053ddbc4..7ef5e282e 100644
--- a/python/sglang/srt/models/minimax_m2.py
+++ b/python/sglang/srt/models/minimax_m2.py
@@ -821,6 +821,9 @@ class MiniMaxM2ForCausalLM(nn.Module):
         else:
             self.model.layers_to_capture = [val + 1 for val in layer_ids]
 
+    def get_embed_and_head(self):
+        return self.model.embed_tokens.weight, self.lm_head.weight
+
     @torch.no_grad()
     def forward(
         self,