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

Update data loader tests

This commit is contained in:
Eren Gölge 2021-12-01 10:06:02 +01:00
parent 93957d58a1
commit 90cc45dd4e
1 changed files with 0 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -69,9 +69,6 @@ class TTSDataset(Dataset):
samples (list): List of dataset samples. samples (list): List of dataset samples.
tokenizer (TTSTokenizer): tokenizer to convert text to sequence IDs. If None init internally else
use the given. Defaults to None.
tokenizer (TTSTokenizer): tokenizer to convert text to sequence IDs. If None init internally else tokenizer (TTSTokenizer): tokenizer to convert text to sequence IDs. If None init internally else
use the given. Defaults to None. use the given. Defaults to None.
@ -205,20 +202,6 @@ class TTSDataset(Dataset):
token_ids = self.tokenizer.text_to_ids(text) token_ids = self.tokenizer.text_to_ids(text)
return np.array(token_ids, dtype=np.int32) return np.array(token_ids, dtype=np.int32)
@staticmethod
def _parse_sample(item):
language_name = None
attn_file = None
if len(item) == 5:
text, wav_file, speaker_name, language_name, attn_file = item
elif len(item) == 4:
text, wav_file, speaker_name, language_name = item
elif len(item) == 3:
text, wav_file, speaker_name = item
else:
token_ids = self.tokenizer.text_to_ids(text)
return token_ids
def load_data(self, idx): def load_data(self, idx):
item = self.samples[idx] item = self.samples[idx]