diff --git a/models.py b/models.py index 7504b18..d53b430 100644 --- a/models.py +++ b/models.py @@ -860,7 +860,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)