tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

更新 models.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-03 15:03:05 +08:00
committed by GitHub
parent 25ea864b56
commit d4ecc90e16
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@@ -860,7 +860,7 @@ class SynthesizerTrn(nn.Module):
hidden_channels, 256, 3, 0.5, gin_channels=gin_channels 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) self.emb_g = nn.Embedding(n_speakers, gin_channels)
else: else:
self.ref_enc = ReferenceEncoder(spec_channels, gin_channels) self.ref_enc = ReferenceEncoder(spec_channels, gin_channels)