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

Make style

This commit is contained in:
Eren Gölge 2021-11-30 15:56:15 +01:00
parent 7575367b9f
commit 98057a00ae
3 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -286,8 +286,12 @@ class TTSDataset(Dataset):
"""
# sort items based on the sequence length in ascending order
text_ignore_idx, text_keep_idx = self.sort_and_filter_by_length(self.text_lengths, self.min_text_len, self.max_text_len)
audio_ignore_idx, audio_keep_idx = self.sort_and_filter_by_length(self.audio_lengths, self.min_audio_len, self.max_audio_len)
text_ignore_idx, text_keep_idx = self.sort_and_filter_by_length(
self.text_lengths, self.min_text_len, self.max_text_len
)
audio_ignore_idx, audio_keep_idx = self.sort_and_filter_by_length(
self.audio_lengths, self.min_audio_len, self.max_audio_len
)
keep_idx = list(set(audio_keep_idx) | set(text_keep_idx))
ignore_idx = list(set(audio_ignore_idx) | set(text_ignore_idx))