Update modeling_InternLM_XComposer.py

modeling_InternLM_XComposer.py

@@ -25,7 +25,8 @@ logger = logging.get_logger(__name__)
 class InternLMXComposerForCausalLM(PreTrainedModel):
     config_class = InternLMXComposerConfig
     _auto_class = "AutoModelForCausalLM"
-
+    _no_split_modules = ["InternLMDecoderLayer"]
+    
     meta_instruction = """meta instruction
 You are an AI assistant whose name is 浦语.
 - 浦语 is a conversational language model that is developed by Shanghai AI Laboratory (上海人工智能实验室). It is designed to be helpful, honest, and harmless.