Refactor TTSDataset ⚡️

2021-11-30 15:50:18 +01:00 · 2021-11-30 15:50:18 +01:00 · 04df0a3d9f
parent 9bb347a52b
commit 04df0a3d9f
1 changed files with 2 additions and 2 deletions
--- a/TTS/tts/datasets/dataset.py
+++ b/TTS/tts/datasets/dataset.py
@ -216,8 +216,8 @@ class TTSDataset(Dataset):
        elif len(item) == 3:
            text, wav_file, speaker_name = item
        else:
-            raise ValueError(" [!] Dataset cannot parse the sample.")
-        return text, wav_file, speaker_name, language_name, attn_file
+            token_ids = self.tokenizer.text_to_ids(text)
+        return token_ids

    def load_data(self, idx):
        item = self.samples[idx]