[Bugfix] Fix handling of image embeds in models (#29480)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2025-11-26 21:00:15 +08:00
committed by GitHub
parent 452a7c9f7c
commit e30859dff3
3 changed files with 4 additions and 15 deletions

View File

@@ -460,7 +460,7 @@ class LlavaNextForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsP
image_input: LlavaNextImageInputs,
) -> torch.Tensor | list[torch.Tensor]:
if image_input["type"] == "image_embeds":
return [image_input["data"]]
return image_input["data"]
patch_embeddings = self._process_image_pixels(image_input)