Update modeling_internlm2.py

modeling_internlm2.py

@@ -1113,9 +1113,9 @@ class InternLM2ForCausalLM(InternLM2PreTrainedModel):
                     img_embeds, atts_img, img_target = self.img2emb(torch.zeros(1,3,self.im_size,self.im_size).to(image.device).to(image.dtype))
                     to_regress_embeds += img_embeds.sum() * 0
                     im_mask = torch.zeros(to_regress_embeds.shape[:2]).cuda()
-                    temp_max_length = 2048
+                    temp_max_length = self.max_length
 
-            temp_max_length = 2048
+            temp_max_length = self.max_length
             inputs_embeds = to_regress_embeds[:, :temp_max_length]
             attention_mask = attention_mask[:, :temp_max_length]
             targets = targets[:, :temp_max_length]