diff --git a/models.py b/models.py
index 452d9b3..fd4fdb5 100644
--- a/models.py
+++ b/models.py
@@ -507,8 +507,8 @@ class SynthesizerTrn(nn.Module):
                  upsample_rates,
                  upsample_initial_channel,
                  upsample_kernel_sizes,
-                 n_speakers=0,
-                 gin_channels=0,
+                 n_speakers=256,
+                 gin_channels=256,
                  use_sdp=True,
                  n_flow_layer = 4,
                  n_layers_trans_flow = 3,