diff --git a/TTS/vocoder/datasets/gan_dataset.py b/TTS/vocoder/datasets/gan_dataset.py index 08fdb0a3cf..a782067e1b 100644 --- a/TTS/vocoder/datasets/gan_dataset.py +++ b/TTS/vocoder/datasets/gan_dataset.py @@ -115,8 +115,8 @@ def load_item(self, idx): audio, mel = self.cache[idx] else: audio = self.ap.load_wav(wavpath) - audio, _ = self._pad_short_samples(audio) mel = self.ap.melspectrogram(audio) + audio, mel = self._pad_short_samples(audio, mel) else: # load precomputed features