This commit is contained in:
tuna2134
2026-07-24 16:29:09 +09:00
parent 8125666e22
commit 0d6eebc3a8

View File

@@ -260,7 +260,7 @@ class JointMelSynthesizer(nn.Module):
@torch.inference_mode() @torch.inference_mode()
def infer(self, *args: Any, **kwargs: Any) -> tuple[torch.Tensor, ...]: def infer(self, *args: Any, **kwargs: Any) -> tuple[torch.Tensor, ...]:
mel, attn, mel_mask = self.acoustic_model.infer_mel(*args, **kwargs) mel, attn, mel_mask = self.acoustic_model.infer_mel(*args, **kwargs)
sid = args[2] sid = args[2] if len(args) > 2 else kwargs["sid"]
return self.vocoder(mel, sid), mel, attn, mel_mask return self.vocoder(mel, sid), mel, attn, mel_mask