diff --git a/TTS/tts/utils/generic_utils.py b/TTS/tts/utils/generic_utils.py
index 94cd5668..1fed8a4e 100644
--- a/TTS/tts/utils/generic_utils.py
+++ b/TTS/tts/utils/generic_utils.py
@@ -106,7 +106,7 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
                         hidden_channels=192,
                         filter_channels=768,
                         filter_channels_dp=256,
-                        out_channels=80,
+                        out_channels=c.audio['num_mels'],
                         kernel_size=3,
                         num_heads=2,
                         num_layers_enc=6,