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

OverFlow with test sentences (#2253) 

* Fix typo in function definiton

* Swap hasattr out

hasattr(self, "speaker_manager")  and hasattr(self, "language_manager") seems to be redundant since BaseTTS defines both.
This commit is contained in:
thennal10 2023-03-01 13:41:30 +05:30 committed by GitHub
parent 16b9862252
commit d39bc74f57
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
TTS/tts/models/base_tts.py
View File

@ -111,7 +111,7 @@ class BaseTTS(BaseTrainerModel):
"""Prepare and return `aux_input` used by `forward()`""" """Prepare and return `aux_input` used by `forward()`"""
return {"speaker_id": None, "style_wav": None, "d_vector": None, "language_id": None} return {"speaker_id": None, "style_wav": None, "d_vector": None, "language_id": None}
def get_aux_input_from_test_setences(self, sentence_info): def get_aux_input_from_test_sentences(self, sentence_info):
if hasattr(self.config, "model_args"): if hasattr(self.config, "model_args"):
config = self.config.model_args config = self.config.model_args
else: else:
@ -134,7 +134,7 @@ class BaseTTS(BaseTrainerModel):
# get speaker id/d_vector # get speaker id/d_vector
speaker_id, d_vector, language_id = None, None, None speaker_id, d_vector, language_id = None, None, None
if hasattr(self, "speaker_manager"): if self.speaker_manager is not None:
if config.use_d_vector_file: if config.use_d_vector_file:
if speaker_name is None: if speaker_name is None:
d_vector = self.speaker_manager.get_random_embedding() d_vector = self.speaker_manager.get_random_embedding()
@ -147,7 +147,7 @@ class BaseTTS(BaseTrainerModel):
speaker_id = self.speaker_manager.name_to_id[speaker_name] speaker_id = self.speaker_manager.name_to_id[speaker_name]
# get language id # get language id
if hasattr(self, "language_manager") and config.use_language_embedding and language_name is not None: if self.language_manager is not None and config.use_language_embedding and language_name is not None:
language_id = self.language_manager.name_to_id[language_name] language_id = self.language_manager.name_to_id[language_name]
return { return {
@ -287,7 +287,7 @@ class BaseTTS(BaseTrainerModel):
loader = None loader = None
else: else:
# setup multi-speaker attributes # setup multi-speaker attributes
if hasattr(self, "speaker_manager") and self.speaker_manager is not None: if self.speaker_manager is not None:
if hasattr(config, "model_args"): if hasattr(config, "model_args"):
speaker_id_mapping = ( speaker_id_mapping = (
self.speaker_manager.name_to_id if config.model_args.use_speaker_embedding else None self.speaker_manager.name_to_id if config.model_args.use_speaker_embedding else None
@ -302,7 +302,7 @@ class BaseTTS(BaseTrainerModel):
d_vector_mapping = None d_vector_mapping = None
# setup multi-lingual attributes # setup multi-lingual attributes
if hasattr(self, "language_manager") and self.language_manager is not None: if self.language_manager is not None:
language_id_mapping = self.language_manager.name_to_id if self.args.use_language_embedding else None language_id_mapping = self.language_manager.name_to_id if self.args.use_language_embedding else None
else: else:
language_id_mapping = None language_id_mapping = None
@ -424,7 +424,7 @@ class BaseTTS(BaseTrainerModel):
print(f" > `speakers.pth` is saved to {output_path}.") print(f" > `speakers.pth` is saved to {output_path}.")
print(" > `speakers_file` is updated in the config.json.") print(" > `speakers_file` is updated in the config.json.")
if hasattr(self, "language_manager") and self.language_manager is not None: if self.language_manager is not None:
output_path = os.path.join(trainer.output_path, "language_ids.json") output_path = os.path.join(trainer.output_path, "language_ids.json")
self.language_manager.save_ids_to_file(output_path) self.language_manager.save_ids_to_file(output_path)
trainer.config.language_ids_file = output_path trainer.config.language_ids_file = output_path