[Models] Optimise and simplify _validate_and_reshape_mm_tensor (#24742)

Signed-off-by: Lukas Geiger <lukas.geiger94@gmail.com>
This commit is contained in:
Lukas Geiger
2025-09-12 16:37:37 +01:00
committed by GitHub
parent 684b6870e1
commit 57f94e88ea
9 changed files with 12 additions and 10 deletions

View File

@@ -342,7 +342,7 @@ class Qwen2AudioForConditionalGeneration(nn.Module, SupportsMultiModal,
raise ValueError(f"Incorrect type of {name}. "
f"Got type: {type(mm_input)}")
if isinstance(mm_input, torch.Tensor):
return torch.concat(list(mm_input))
return mm_input.reshape(-1, *mm_input.shape[2:])
else:
return torch.concat(mm_input)