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

Merge branch 'dev' of github.com:mozilla/TTS into dev

pull/10/head
Eren Golge 2019-07-18 16:42:36 +02:00
parent 3c80f7acd9 63c0085256
commit d5cefe0546
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
datasets/preprocess.py
View File

@ -70,7 +70,7 @@ def mozilla(root_path, meta_file):
return items return items
def mailabs(root_path, meta_files): def mailabs(root_path, meta_files=None):
"""Normalizes M-AI-Labs meta data files to TTS format""" """Normalizes M-AI-Labs meta data files to TTS format"""
speaker_regex = re.compile("by_book/(male|female|mix)/(?P<speaker_name>[^/]+)/") speaker_regex = re.compile("by_book/(male|female|mix)/(?P<speaker_name>[^/]+)/")
if meta_files is None: if meta_files is None:
@ -164,4 +164,4 @@ def libri_tts(root_path, meta_files=None):
items.append([text, wav_file, speaker_name]) items.append([text, wav_file, speaker_name])
for item in items: for item in items:
assert os.path.exists(item[1]), f" [!] wav file is not exist - {item[1]}" assert os.path.exists(item[1]), f" [!] wav file is not exist - {item[1]}"
return items return items

2
utils/synthesis.py
View File

@ -108,7 +108,7 @@ def synthesis(model,
# preprocess the given text # preprocess the given text
inputs = text_to_seqvec(text, CONFIG, use_cuda) inputs = text_to_seqvec(text, CONFIG, use_cuda)
speaker_id = id_to_torch(speaker_id) speaker_id = id_to_torch(speaker_id)
if use_cuda: if speaker_id is not None and use_cuda:
speaker_id = speaker_id.cuda() speaker_id = speaker_id.cuda()
# synthesize voice # synthesize voice
decoder_output, postnet_output, alignments, stop_tokens = run_model( decoder_output, postnet_output, alignments, stop_tokens = run_model(