Skip to content

Commit

Permalink
[Bugfix] Reshape the dimensions of the input image embeddings in Qwen…
Browse files Browse the repository at this point in the history
…2VL (vllm-project#9071)

Signed-off-by: Amit Garg <mitgarg17495@gmail.com>
  • Loading branch information
whyiug authored and garg-amit committed Oct 28, 2024
1 parent c81fc8c commit e22cd70
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions vllm/model_executor/models/qwen2_vl.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)}")
Expand Down

0 comments on commit e22cd70

Please sign in to comment.