Update models.py

fishaudio · Oct 4, 2023 · 8beaf32 · 8beaf32
1 parent 10b1e6f
commit 8beaf32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models.py b/models.py
@@ -763,7 +763,7 @@ def __init__(
         gin_channels=256,
         use_sdp=True,
         n_flow_layer=4,
-        n_layers_trans_flow=3,
+        n_layers_trans_flow=6,
         flow_share_parameter=False,
         use_transformer_flow=True,
         **kwargs