Refactor TTSDataset ⚡️

2026-09-11 12:00:24 +08:00 · 2022-02-25 11:05:06 +01:00
parent 9bb347a52b
commit 04df0a3d9f
1 changed files with 2 additions and 2 deletions
@@ -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]