fix naming to backbone

Signed-off-by: jenchen13 <[email protected]>

Author: jenchen13

modelopt/torch/export/plugins/mcore_nemotron.py

@@ -26,8 +26,6 @@
     NameRemapping,
     QKVMerging,
     QKVSlicing,
-    GatedMLPSlicing,
-    GatedMLPMerging,
 )
 
 # Example on adding a new CausalLM.
@@ -69,11 +67,11 @@
     "linear_fc2": NameRemapping("backbone.layers.{}.mixer.down_proj.", ROW_TP),
     # MoE
     "router": NameRemapping("model.layers.{}.mlp.gate.", REPLICATE),
-    "local_experts.linear_fc1": NameRemapping("model.layers.{}.mlp.experts.{}.up_proj", COL_ETP),
-    "local_experts.linear_fc2": NameRemapping("model.layers.{}.mlp.experts.{}.down_proj.", ROW_ETP),
-    "shared_experts.linear_fc1": NameRemapping("model.layers.{}.mlp.shared_experts.up_proj.", COL_TP),
+    "local_experts.linear_fc1": NameRemapping("backbone.layers.{}.mlp.experts.{}.up_proj", COL_ETP),
+    "local_experts.linear_fc2": NameRemapping("backbone.layers.{}.mlp.experts.{}.down_proj.", ROW_ETP),
+    "shared_experts.linear_fc1": NameRemapping("backbone.layers.{}.mlp.shared_experts.up_proj.", COL_TP),
     "shared_experts.linear_fc2": NameRemapping(
-        "model.layers.{}.mlp.shared_experts.down_proj.", ROW_TP
+        "backbone.layers.{}.mlp.shared_experts.down_proj.", ROW_TP
     ),
 
 }
@@ -101,10 +99,10 @@
     "linear_fc1": NameRemapping("backbone.layers.{}.mixer.up_proj."),
     "linear_fc2": NameRemapping("backbone.layers.{}.mixer.down_proj."),
     # MoE
-    "router": NameRemapping("model.layers.{}.mlp.gate."),
-    "local_experts.linear_fc1": NameRemapping("model.layers.{}.mlp.experts.{}.up_proj."),
-    "local_experts.linear_fc2": NameRemapping("model.layers.{}.mlp.experts.{}.down_proj."),
-    "shared_experts.linear_fc1": NameRemapping("model.layers.{}.mlp.shared_experts.up_proj."),
-    "shared_experts.linear_fc2": NameRemapping("model.layers.{}.mlp.shared_experts.down_proj."),
+    "router": NameRemapping("backbone.layers.{}.mlp.gate."),
+    "local_experts.linear_fc1": NameRemapping("backbone.layers.{}.mlp.experts.{}.up_proj."),
+    "local_experts.linear_fc2": NameRemapping("backbone.layers.{}.mlp.experts.{}.down_proj."),
+    "shared_experts.linear_fc1": NameRemapping("backbone.layers.{}.mlp.shared_experts.up_proj."),
+    "shared_experts.linear_fc2": NameRemapping("backbone.layers.{}.mlp.shared_experts.down_proj."),
 
 }