diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 0f8e111a..cb160273 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -204,6 +204,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