[V1][VLM] Fix edge case bug for InternVL2 (#11165)

Signed-off-by: Roger Wang <[email protected]>

Author: ywang96

vllm/model_executor/models/internvl.py

@@ -669,8 +669,11 @@ def _process_image_input(
         image_embeds = self.extract_feature(image_input["data"])
 
         patches_per_image = image_input["patches_per_image"]
+
+        # Only one image in the current batch
         if len(patches_per_image) == 1:
-            image_embeds = image_embeds.unsqueeze(0)
+            image_embeds = image_embeds.view(
+                -1, self.config.text_config.hidden_size).unsqueeze(0)
             return image_embeds
 
         # NOTE: Image embeddings are split into separate tensors for each image