[BugFix]: Fix Qwen3-TTS code2wav fails when enforce_eager: false (#2868)

Signed-off-by: wuyuefeng <565948592@qq.com>
This commit is contained in:
yuefeng Wu authored and GitHub committed 2026-04-23 09:54:56 -04:00
1 parent 95a07f7732
commit c8efdbd671
1 file changed
+8 -2
@@ -324,12 +324,18 @@ class Qwen3TTSCode2Wav(nn.Module):
multimodal_outputs={"model_outputs": audios, "sr": srs},
)
def make_omni_output(self, model_outputs: torch.Tensor | OmniOutput, **kwargs: Any) -> OmniOutput:
def make_omni_output(self, model_outputs: torch.Tensor | OmniOutput | tuple, **kwargs: Any) -> OmniOutput:
if isinstance(model_outputs, OmniOutput):
return model_outputs
if isinstance(model_outputs, tuple) and len(model_outputs) == len(OmniOutput._fields):
return OmniOutput(*model_outputs)
if not (isinstance(model_outputs, tuple) and len(model_outputs) == 2):
raise TypeError(f"Qwen3TTSCode2Wav expected (audio_tensor, sr) outputs, got {type(model_outputs)}")
raise TypeError(
"Qwen3TTSCode2Wav expected OmniOutput, OmniOutput tuple, "
f"or (audio_tensor, sr) outputs, got {type(model_outputs)}"
)
audio_tensor, sr = model_outputs
return OmniOutput(