Update modeling_internlm_xcomposer2.py

raise value error for invalid <ImageHere> prompt format

modeling_internlm_xcomposer2.py

@@ -156,6 +156,7 @@ class InternLMXComposer2ForCausalLM(InternLM2PreTrainedModel):
             return_tensors='pt',
             padding='longest',
             truncation=True,
+            max_length=self.max_length,
             add_special_tokens=add_special).to(self.device)
 
         targets = self.mask_human_targets(to_regress_tokens.input_ids)
@@ -175,6 +176,9 @@ class InternLMXComposer2ForCausalLM(InternLM2PreTrainedModel):
         parts = prompt.split('<ImageHere>')
         wrap_embeds, wrap_im_mask = [], []
         temp_len = 0
+
+        if len(parts) != image_nums + 1:
+            raise ValueError('Invalid <ImageHere> prompt format.')
     
         for idx, part in enumerate(parts):
             if len(part) > 0: