Update configuration_internlm_xcomposer2.py

configuration_internlm_xcomposer2.py

@@ -71,8 +71,19 @@ class InternLMXcomposer2Config(PretrainedConfig):
         Example:
 
     """
-    model_type = "internlm2"
+    model_type = "internlmxcomposer2"
     _auto_class = "AutoConfig"
+    
+    layers_block_name = "model.layers"
+    outside_layer_modules = [
+        'vit', 'vision_proj', 'model.tok_embeddings', 'model.norm', 'output',
+    ]
+    inside_layer_modules = [
+        ["attention.wqkv.linear"],
+        ["attention.wo.linear"],
+        ["feed_forward.w1.linear", "feed_forward.w3.linear"],
+        ["feed_forward.w2.linear"],
+    ]
 
     def __init__(  # pylint: disable=W0102
         self,