mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-03 05:07:02 +08:00
[Models][Qwen3VL] Optimise _validate_and_reshape_mm_tensor (#26426)
Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
parent
4bdf7ac593
commit
0426e3c5e1
@ -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)
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user