From c3f1bde10fb232057798456aa948b8d4f937a5d0 Mon Sep 17 00:00:00 2001 From: DarkLight1337 Date: Thu, 19 Dec 2024 18:24:18 +0000 Subject: [PATCH] Fix extra dimension Signed-off-by: DarkLight1337 --- vllm/multimodal/processing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/multimodal/processing.py b/vllm/multimodal/processing.py index 306b7f11f413b..78a031032337a 100644 --- a/vllm/multimodal/processing.py +++ b/vllm/multimodal/processing.py @@ -687,7 +687,7 @@ def _cached_call_fine( ) for k, v in processed_mm_item.items(): - processed_mm_items[k].append(v) + processed_mm_items[k].append(v[0]) # NOTE: Some processors do not accept mm-only input, in which case # we have to fallback to processing `prompt` and `mm_data` together