Update configuration_ernie4_5_vl.py

configuration_ernie4_5_vl.py

@@ -539,19 +539,19 @@ class Ernie4_5_VLMoEConfig(Ernie4_5_MoEConfig):
         "activation_function": "hidden_act",
     }
     base_model_tp_plan = {
-        "ernie.layers.*.self_attn.q_proj": "colwise_rep",
-        "ernie.layers.*.self_attn.k_proj": "colwise_rep",
-        "ernie.layers.*.self_attn.v_proj": "colwise_rep",
-        "ernie.layers.*.self_attn.o_proj": "rowwise_rep",
-        "ernie.layers.*.mlp.experts.*.gate_proj": "colwise",
-        "ernie.layers.*.mlp.experts.*.up_proj": "colwise",
-        "ernie.layers.*.mlp.experts.*.down_proj": "rowwise",
-        "ernie.layers.*.mlp_text.experts.*.gate_proj": "colwise",
-        "ernie.layers.*.mlp_text.experts.*.up_proj": "colwise",
-        "ernie.layers.*.mlp_text.experts.*.down_proj": "rowwise",
-        "ernie.layers.*.mlp.gate_proj": "colwise",
-        "ernie.layers.*.mlp.up_proj": "colwise",
-        "ernie.layers.*.mlp.down_proj": "rowwise",
+        "model.layers.*.self_attn.q_proj": "colwise_rep",
+        "model.layers.*.self_attn.k_proj": "colwise_rep",
+        "model.layers.*.self_attn.v_proj": "colwise_rep",
+        "model.layers.*.self_attn.o_proj": "rowwise_rep",
+        "model.layers.*.mlp.experts.*.gate_proj": "colwise",
+        "model.layers.*.mlp.experts.*.up_proj": "colwise",
+        "model.layers.*.mlp.experts.*.down_proj": "rowwise",
+        "model.layers.*.mlp_text.experts.*.gate_proj": "colwise",
+        "model.layers.*.mlp_text.experts.*.up_proj": "colwise",
+        "model.layers.*.mlp_text.experts.*.down_proj": "rowwise",
+        "model.layers.*.mlp.gate_proj": "colwise",
+        "model.layers.*.mlp.up_proj": "colwise",
+        "model.layers.*.mlp.down_proj": "rowwise"
     }
 
     def __init__(