Update modeling_minicpmv.py

modeling_minicpmv.py

@@ -239,6 +239,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
         **kwargs
     ):
         batch = []
+        counter = 0
         for model_inputs in model_inputs_batch:
             bs = len(model_inputs["input_ids"])
             img_list = model_inputs["pixel_values"]
@@ -265,6 +266,8 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
                 input_embeds,
                 vision_hidden_states,
             ) = self.get_vllm_embedding(model_inputs)  
+            print(input_embeds.shape, f"INPUT_EMBEDS {counter}")
+            counter += 1
             batch.append(input_embeds)
             
         batch = torch.stack(batch)