diff --git a/models.py b/models.py
index a8e1e9f..2364d62 100644
--- a/models.py
+++ b/models.py
@@ -579,7 +579,7 @@ class SynthesizerTrn(nn.Module):
                  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):