XzJosh commited on
Commit
fd6b5d7
1 Parent(s): ca38eea

Upload models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -632,7 +632,7 @@ class SynthesizerTrn(nn.Module):
632
  self.sdp = StochasticDurationPredictor(hidden_channels, 192, 3, 0.5, 4, gin_channels=gin_channels)
633
  self.dp = DurationPredictor(hidden_channels, 256, 3, 0.5, gin_channels=gin_channels)
634
 
635
- if n_speakers > 1:
636
  self.emb_g = nn.Embedding(n_speakers, gin_channels)
637
  else:
638
  self.ref_enc = ReferenceEncoder(spec_channels, gin_channels)
 
632
  self.sdp = StochasticDurationPredictor(hidden_channels, 192, 3, 0.5, 4, gin_channels=gin_channels)
633
  self.dp = DurationPredictor(hidden_channels, 256, 3, 0.5, gin_channels=gin_channels)
634
 
635
+ if n_speakers >= 1:
636
  self.emb_g = nn.Embedding(n_speakers, gin_channels)
637
  else:
638
  self.ref_enc = ReferenceEncoder(spec_channels, gin_channels)