diff --git a/models.py b/models.py
index dd9e0c0..f392136 100644
--- a/models.py
+++ b/models.py
@@ -763,7 +763,7 @@ class SynthesizerTrn(nn.Module):
         gin_channels=256,
         use_sdp=True,
         n_flow_layer=4,
-        n_layers_trans_flow=6,
+        n_layers_trans_flow=4,
         flow_share_parameter=False,
         use_transformer_flow=True,
         **kwargs