[Models] Optimise and simplify _validate_and_reshape_mm_tensor (#24742)

Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
2025-09-12 16:37:37 +01:00
parent 684b6870e1
commit 57f94e88ea
9 changed files with 12 additions and 10 deletions
--- a/vllm/model_executor/models/qwen2_5_omni_thinker.py
+++ b/vllm/model_executor/models/qwen2_5_omni_thinker.py
@@ -551,6 +551,8 @@ class Qwen2_5OmniConditionalGenerationMixin:
            raise ValueError(f"Incorrect type of {name}. "
                             f"Got type: {type(mm_input)}")
        if isinstance(mm_input, torch.Tensor):
+            if dim == 0:
+                return mm_input.reshape(-1, *mm_input.shape[2:])
            return torch.concat(list(mm_input), dim=dim)
        else:
            return torch.concat(mm_input, dim=dim)