diff --git a/python/sglang/srt/models/minimax_m2.py b/python/sglang/srt/models/minimax_m2.py index 9b50d3d07..30815d8fa 100644 --- a/python/sglang/srt/models/minimax_m2.py +++ b/python/sglang/srt/models/minimax_m2.py @@ -942,6 +942,18 @@ class MiniMaxM2Model(nn.Module): class MiniMaxM2ForCausalLM(nn.Module): """MiniMax M2 model for causal language modeling.""" + packed_modules_mapping = { + "qkv_proj": [ + "q_proj", + "k_proj", + "v_proj", + ], + "gate_up_proj": [ + "gate_proj", + "up_proj", + ], + } + def __init__( self, config: PretrainedConfig,