Fix extra dimension

Signed-off-by: DarkLight1337 <[email protected]>
vllm-project · Dec 19, 2024 · c3f1bde · c3f1bde
1 parent b84ff42
commit c3f1bde
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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