Spaces:
Running
Running
Upload models.py
Browse files
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
|
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)
|