update

2023-07-23 15:37:50 +08:00
parent 8fc674fe2c
commit 1fa3ce1d03
4 changed files with 6 additions and 6 deletions
--- a/models.py
+++ b/models.py
@@ -515,7 +515,7 @@ class SynthesizerTrn(nn.Module):
        if n_speakers > 1:
            self.emb_g = nn.Embedding(n_speakers, gin_channels)
        else:
-            self.ref_enc = ReferenceEncoder(spec_channels)
+            self.ref_enc = ReferenceEncoder(spec_channels, gin_channels)

    def forward(self, x, x_lengths, y, y_lengths, sid, tone, language, bert):