Unverified Commit 0426e3c5 authored by Lukas Geiger's avatar Lukas Geiger Committed by GitHub
Browse files

[Models][Qwen3VL] Optimise `_validate_and_reshape_mm_tensor` (#26426)


Signed-off-by: default avatarLukas Geiger <lukas.geiger94@gmail.com>
parent 4bdf7ac5
......@@ -1304,7 +1304,7 @@ class Qwen3VLForConditionalGeneration(
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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment