layers_block_name = "model.layers"

modeling_InternLM_XComposer.py

@@ -37,6 +37,8 @@ class InternLMXComposerForCausalLM(PreTrainedModel):
         max_new_tokens=500,
     )
 
+    layers_block_name = "model.layers"
+
     def __init__(self, config):
         super().__init__(config)