[BugFix] Qwen3-Omni / Qwen2.5-Omni thinker batching with audio input (#451)

Signed-off-by: yinpeiqi <yinpeiqi809@gmail.com>
This commit is contained in:
Peiqi Yin authored and GitHub committed 2025-12-24 18:43:00 +08:00
1 parent cbaf508665
commit da7f25e14b
1 file changed
+2 -1
@@ -90,7 +90,8 @@ class Qwen2_5OmniConditionalGenerationMixin(Qwen2_5OmniConditionalGenerationMixi
):
feature_attention_mask = feature_attention_mask.reshape(-1, feature_attention_mask.shape[-1])
elif feature_attention_mask is not None and isinstance(feature_attention_mask, list):
feature_attention_mask = torch.cat(feature_attention_mask, dim=-1)
for i in range(len(feature_attention_mask)):
feature_attention_mask[i] = feature_attention_mask[i].reshape(-1)
return Qwen2_5OmniAudioFeatureInputs(
type="audio_features",
input_features=input_audio_features,