[BugFix] Qwen3-Omni / Qwen2.5-Omni thinker batching with audio input (#451)
Signed-off-by: yinpeiqi <yinpeiqi809@gmail.com>
This commit is contained in:
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,
|
||||
|
||||
Reference in new issue
Block a user