Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor TTSDataset ⚡️

This commit is contained in:
Eren Gölge 2021-11-30 15:50:18 +01:00
parent 9bb347a52b
commit 04df0a3d9f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/datasets/dataset.py
View File

@ -216,8 +216,8 @@ class TTSDataset(Dataset):
elif len(item) == 3: elif len(item) == 3:
text, wav_file, speaker_name = item text, wav_file, speaker_name = item
else: else:
raise ValueError(" [!] Dataset cannot parse the sample.") token_ids = self.tokenizer.text_to_ids(text)
return text, wav_file, speaker_name, language_name, attn_file return token_ids
def load_data(self, idx): def load_data(self, idx):
item = self.samples[idx] item = self.samples[idx]