Update models.py

2023-08-20 14:45:58 +08:00
parent b990832a99
commit 1f9558c157
1 changed files with 2 additions and 2 deletions
--- a/models.py
+++ b/models.py
@@ -510,7 +510,7 @@ class SynthesizerTrn(nn.Module):
                 n_speakers=0,
                 gin_channels=0,
                 use_sdp=True,
-                 _flow_layer = 4,
+                 n_flow_layer = 4,
                 n_layers_trans_flow = 3,
                 use_transformer_flow = True,
                 **kwargs):