Update modeling_internlm_xcomposer2.py

modeling_internlm_xcomposer2.py

@@ -349,8 +349,7 @@ class InternLMXComposer2ForCausalLM(InternLM2PreTrainedModel):
             im_mask = kwargs.get('im_mask', None)
             if im_mask is None and inputs_embeds is not None:
                 im_mask = torch.zeros(inputs_embeds.shape[:2]).to(
-                    inputs_embeds.device)
-                im_mask[:, 1:1 + self.im_seq_len] = 1
+                    inputs_embeds.device) 
                 im_mask = im_mask.bool()
 
         output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions