Update modeling_perceive_sampler.py

modeling_perceive_sampler.py

@@ -663,6 +663,9 @@ class BertModel(BertPreTrainedModel):
     argument and :obj:`add_cross_attention` set to :obj:`True`; an :obj:`encoder_hidden_states` is then expected as an
     input to the forward pass.
     """
+
+    _no_split_modules = ["InternLMDecoderLayer"]
+    
     def __init__(self, config, add_pooling_layer=False):
         super().__init__(config)
         self.config = config