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

Fix zoo tests

This commit is contained in:
WeberJulian 2021-12-09 12:42:38 +01:00 committed by Eren Gölge
parent 846b4a14c6
commit b4bb0ace70
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
TTS/utils/synthesizer.py
View File

@ -118,11 +118,14 @@ class Synthesizer(object):
speaker_manager = self._init_speaker_manager() speaker_manager = self._init_speaker_manager()
language_manager = self._init_language_manager() language_manager = self._init_language_manager()
self.tts_model = setup_tts_model( if language_manager is not None:
config=self.tts_config, self.tts_model = setup_tts_model(
speaker_manager=speaker_manager, config=self.tts_config,
language_manager=language_manager, speaker_manager=speaker_manager,
) language_manager=language_manager,
)
else:
self.tts_model = setup_tts_model(config=self.tts_config, speaker_manager=speaker_manager)
self.tts_model.load_checkpoint(self.tts_config, tts_checkpoint, eval=True) self.tts_model.load_checkpoint(self.tts_config, tts_checkpoint, eval=True)
if use_cuda: if use_cuda:
self.tts_model.cuda() self.tts_model.cuda()
@ -251,7 +254,7 @@ class Synthesizer(object):
# handle multi-lingaul # handle multi-lingaul
language_id = None language_id = None
if self.tts_languages_file or hasattr(self.tts_model.language_manager, "language_id_mapping"): if self.tts_languages_file or (hasattr(self.tts_model, "language_manager") and self.tts_model.language_manager is not None):
if language_idx and isinstance(language_idx, str): if language_idx and isinstance(language_idx, str):
language_id = self.tts_model.language_manager.language_id_mapping[language_idx] language_id = self.tts_model.language_manager.language_id_mapping[language_idx]