[Bugfix] Reshape the dimensions of the input image embeddings in Qwen…

…2VL (#9071)
vllm-project · Oct 4, 2024 · 3d826d2 · 3d826d2
1 parent 0e36fd4
commit 3d826d2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/vllm/model_executor/models/qwen2_vl.py b/vllm/model_executor/models/qwen2_vl.py
@@ -967,6 +967,9 @@ def _parse_and_validate_image_input(
                                            image_grid_thw=image_grid_thw)
 
         if image_embeds is not None:
+            image_embeds = self._validate_and_reshape_mm_tensor(
+                image_embeds, "image embeds")
+
             if not isinstance(image_embeds, torch.Tensor):
                 raise ValueError("Incorrect type of image embeddings. "
                                  f"Got type: {type(image_embeds)}")