Update models.py

Stardust-minus · web-flow · commit 8beaf3279b70 · 2023-10-04T21:47:43.000+08:00
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
