feat: add support for mamba cp

kmehant · kmehant · commit 10154941f3d3 · 2025-11-14T17:27:36.000+05:30
Signed-off-by: Mehant Kammakomati &lt;mehant.kammakomati2@ibm.com&gt;
diff --git a/tuning/config/acceleration_configs/mcp.py b/tuning/config/acceleration_configs/mcp.py
@@ -32,7 +32,7 @@ class MCP:
 @dataclass
 class MCPConfig:
 
-    cp: MCP = None
+    mcp: MCP = None
 
     def __post_init__(self):
         # ensure nested dataclasses initialized
diff --git a/tuning/sft_trainer.py b/tuning/sft_trainer.py
@@ -200,7 +200,7 @@ def train(
             )
     if fast_moe_config is not None and fast_moe_config.fast_moe is None:
         fast_moe_config = None
-    if mcp_config is not None and mcp_config.cp is None:
+    if mcp_config is not None and mcp_config.mcp is None:
         mcp_config = None
     if fast_moe_config is not None:
         # If LoRA with ScatterMoE detected, raise warning

Original file line number	Diff line number	Diff line change
`@@ -200,7 +200,7 @@ def train(`
`200`	`200`	`)`
`201`	`201`	`if fast_moe_config is not None and fast_moe_config.fast_moe is None:`
`202`	`202`	`fast_moe_config = None`
`203`		`- if mcp_config is not None and mcp_config.cp is None:`
	`203`	`+ if mcp_config is not None and mcp_config.mcp is None:`
`204`	`204`	`mcp_config = None`
`205`	`205`	`if fast_moe_config is not None:`
`206`	`206`	`# If LoRA with ScatterMoE detected, raise warning`