[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/glm4_1v.py
+++ b/vllm/model_executor/models/glm4_1v.py
@@ -1378,7 +1378,7 @@ class Glm4vForConditionalGeneration(nn.Module, SupportsMultiModal,
                raise ValueError(f"{name} should be 2D or batched 3D tensor. "
                                 f"Got ndim: {mm_input.ndim} "
                                 f"(shape={mm_input.shape})")
-            return torch.concat(list(mm_input))
+            return mm_input.reshape(-1, mm_input.shape[-1])
        else:
            return torch.concat(mm_input)