diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 662877b3d6..c301da87a0 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -187,6 +187,7 @@ def wav_to_mel(y, n_fft, num_mels, sample_rate, hop_length, win_length, fmin, fm # CONFIGS ############################# + @dataclass class VitsAudioConfig(Coqpit): fft_size: int = 1024