Skip to content

Commit 10b34e3

Browse files
[Bugfix] Fixed the issue of not being able to input video and image simultaneously (#15387)
Signed-off-by: chaunceyjiang <[email protected]>
1 parent b5269db commit 10b34e3

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

vllm/entrypoints/chat_utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -556,11 +556,11 @@ def all_mm_data(self) -> Optional[MultiModalDataDict]:
556556
raise ValueError(\
557557
"Only one message can have {'type': 'image_embeds'}")
558558
mm_inputs["image"] = image_embeds_lst[0]
559-
elif "image" in items_by_modality:
559+
if "image" in items_by_modality:
560560
mm_inputs["image"] = items_by_modality["image"] # A list of images
561-
elif "audio" in items_by_modality:
561+
if "audio" in items_by_modality:
562562
mm_inputs["audio"] = items_by_modality["audio"] # A list of audios
563-
elif "video" in items_by_modality:
563+
if "video" in items_by_modality:
564564
mm_inputs["video"] = items_by_modality["video"] # A list of videos
565565
return mm_inputs
566566

@@ -589,11 +589,11 @@ async def all_mm_data(self) -> Optional[MultiModalDataDict]:
589589
raise ValueError(
590590
"Only one message can have {'type': 'image_embeds'}")
591591
mm_inputs["image"] = image_embeds_lst[0]
592-
elif "image" in items_by_modality:
592+
if "image" in items_by_modality:
593593
mm_inputs["image"] = items_by_modality["image"] # A list of images
594-
elif "audio" in items_by_modality:
594+
if "audio" in items_by_modality:
595595
mm_inputs["audio"] = items_by_modality["audio"] # A list of audios
596-
elif "video" in items_by_modality:
596+
if "video" in items_by_modality:
597597
mm_inputs["video"] = items_by_modality["video"] # A list of videos
598598
return mm_inputs
599599

0 commit comments

Comments
 (0)