From 08aed05796f28244bf70e26ffa9f8ba88031e938 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=CE=9D=CE=B1=CF=81=CE=BF=CF=85=CF=83=CE=AD=C2=B7=CE=BC?= =?UTF-8?q?=C2=B7=CE=B3=CE=B9=CE=BF=CF=85=CE=BC=CE=B5=CE=BC=CE=AF=C2=B7?= =?UTF-8?q?=CE=A7=CE=B9=CE=BD=CE=B1=CE=BA=CE=AC=CE=BD=CE=BD=CE=B1?= <40709280+NaruseMioShirakana@users.noreply.github.com> Date: Thu, 8 Feb 2024 21:39:29 +0800 Subject: [PATCH] Add files via upload --- GPT_SoVITS/module/models_onnx.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/GPT_SoVITS/module/models_onnx.py b/GPT_SoVITS/module/models_onnx.py index 35fd291..232fd74 100644 --- a/GPT_SoVITS/module/models_onnx.py +++ b/GPT_SoVITS/module/models_onnx.py @@ -896,9 +896,6 @@ class SynthesizerTrn(nn.Module): refer_mask = torch.ones_like(refer[:1,:1,:]) ge = self.ref_enc(refer * refer_mask, refer_mask) - y_lengths = torch.LongTensor([codes.size(2) * 2]).to(codes.device) - text_lengths = torch.LongTensor([text.size(-1)]).to(text.device) - quantized = self.quantizer.decode(codes) if self.semantic_frame_rate == "25hz": dquantized = torch.cat([quantized, quantized]).permute(1, 2, 0) @@ -907,6 +904,7 @@ class SynthesizerTrn(nn.Module): x, m_p, logs_p, y_mask = self.enc_p( quantized, text, ge ) + z_p = m_p + torch.randn_like(m_p) * torch.exp(logs_p) z = self.flow(z_p, y_mask, g=ge, reverse=True)