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

View File

@ -118,11 +118,14 @@ class Synthesizer(object):
speaker_manager = self._init_speaker_manager()
language_manager = self._init_language_manager()
self.tts_model = setup_tts_model(
config=self.tts_config,
speaker_manager=speaker_manager,
language_manager=language_manager,
)
if language_manager is not None:
self.tts_model = setup_tts_model(
config=self.tts_config,
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)
if use_cuda:
self.tts_model.cuda()
@ -251,7 +254,7 @@ class Synthesizer(object):
# handle multi-lingaul
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):
language_id = self.tts_model.language_manager.language_id_mapping[language_idx]