Update models.py

2023-11-07 12:51:59 +08:00
parent 16af947b67
commit 59805037c1
1 changed files with 1 additions and 1 deletions
--- a/models.py
+++ b/models.py
@@ -899,7 +899,7 @@ class SynthesizerTrn(nn.Module):
            hidden_channels, 256, 3, 0.5, gin_channels=gin_channels
        )

-        if n_speakers > 1:
+        if n_speakers > =1:
            self.emb_g = nn.Embedding(n_speakers, gin_channels)
        else:
            self.ref_enc = ReferenceEncoder(spec_channels, gin_channels)