From 46ebea8f6b2e902e81aadd55a0260783031c84e4 Mon Sep 17 00:00:00 2001 From: 0hujun <96733800+0hujun@users.noreply.github.com> Date: Tue, 21 Apr 2026 11:23:17 +0800 Subject: [PATCH 1/2] fix: Add is_mtp parameter to _set_moe_state avoid type error --- src/mcore_bridge/model/gpts/minimax_m2.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/mcore_bridge/model/gpts/minimax_m2.py b/src/mcore_bridge/model/gpts/minimax_m2.py index 9814c84..657cefa 100644 --- a/src/mcore_bridge/model/gpts/minimax_m2.py +++ b/src/mcore_bridge/model/gpts/minimax_m2.py @@ -80,6 +80,7 @@ def _set_moe_state( hf_prefix: str, layer_idx: int, to_mcore: bool, + is_mtp: bool, ): if to_mcore: hf_state_dict = { From 32199a8f7c5b286e517d5fbeb2baf8ac940157ae Mon Sep 17 00:00:00 2001 From: 0hujun <96733800+0hujun@users.noreply.github.com> Date: Tue, 21 Apr 2026 11:32:14 +0800 Subject: [PATCH 2/2] fix: Add is_mtp parameter to _set_moe_state avoid type error --- src/mcore_bridge/model/gpts/minimax_m2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mcore_bridge/model/gpts/minimax_m2.py b/src/mcore_bridge/model/gpts/minimax_m2.py index 657cefa..ebdfae3 100644 --- a/src/mcore_bridge/model/gpts/minimax_m2.py +++ b/src/mcore_bridge/model/gpts/minimax_m2.py @@ -80,14 +80,14 @@ def _set_moe_state( hf_prefix: str, layer_idx: int, to_mcore: bool, - is_mtp: bool, + is_mtp: bool = False, ): if to_mcore: hf_state_dict = { k.replace('.w1.', '.gate_proj.').replace('.w3.', '.up_proj.').replace('.w2.', '.down_proj.'): v for k, v in hf_state_dict.items() } - hf_state_dict = super()._set_moe_state(mg_mlp, hf_state_dict, hf_prefix, layer_idx, to_mcore) + hf_state_dict = super()._set_moe_state(mg_mlp, hf_state_dict, hf_prefix, layer_idx, to_mcore, is_mtp) if not to_mcore: hf_state_dict = { k.replace('.gate_proj.', '.w1.').replace('.up_proj.', '.w3.').replace('.down_proj.', '.w2.'): v