Skip to content

Commit e22cd70

Browse files
whyiuggarg-amit
authored andcommitted
[Bugfix] Reshape the dimensions of the input image embeddings in Qwen2VL (vllm-project#9071)
Signed-off-by: Amit Garg <[email protected]>
1 parent c81fc8c commit e22cd70

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

vllm/model_executor/models/qwen2_vl.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -967,6 +967,9 @@ def _parse_and_validate_image_input(
967967
image_grid_thw=image_grid_thw)
968968

969969
if image_embeds is not None:
970+
image_embeds = self._validate_and_reshape_mm_tensor(
971+
image_embeds, "image embeds")
972+
970973
if not isinstance(image_embeds, torch.Tensor):
971974
raise ValueError("Incorrect type of image embeddings. "
972975
f"Got type: {type(image_embeds)}")

0 commit comments

Comments
 (0)