Fix coqui-ai#2062

rhyw · Nov 7, 2022 · b686c09 · b686c09
1 parent fcbfca8
commit b686c09
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/TTS/vocoder/configs/hifigan_config.py b/TTS/vocoder/configs/hifigan_config.py
@@ -22,14 +22,12 @@ class HifiganConfig(BaseGANVocoderConfig):
         generator_model_params (dict): Parameters of the generator model. Defaults to
             `
             {
-                "use_mel": True,
-                "sample_rate": 22050,
-                "n_fft": 1024,
-                "hop_length": 256,
-                "win_length": 1024,
-                "n_mels": 80,
-                "mel_fmin": 0.0,
-                "mel_fmax": None,
+                "upsample_factors": [8, 8, 2, 2],
+                "upsample_kernel_sizes": [16, 16, 4, 4],
+                "upsample_initial_channel": 512,
+                "resblock_kernel_sizes": [3, 7, 11],
+                "resblock_dilation_sizes": [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
+                "resblock_type": "1",
             }
             `
         batch_size (int):