Update models.py

2023-08-20 15:54:10 +08:00
parent 8f0c3340bc
commit 8791e93f87
1 changed files with 2 additions and 2 deletions
--- a/models.py
+++ b/models.py
@@ -507,8 +507,8 @@ class SynthesizerTrn(nn.Module):
                 upsample_rates,
                 upsample_initial_channel,
                 upsample_kernel_sizes,
-                 n_speakers=0,
+                 n_speakers=256,
-                 gin_channels=0,
+                 gin_channels=256,
                 use_sdp=True,
                 n_flow_layer = 4,
                 n_layers_trans_flow = 3,