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

Fix d-vector multispeaker training bug

pull/1032/head
Edresson 2021-08-26 14:47:53 -03:00 committed by Eren Gölge
parent 08da902af3
commit d7042ecfd8
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -134,13 +134,13 @@ class BaseTTS(BaseModel):
# get speaker id/d_vector
speaker_id, d_vector, language_id = None, None, None
if hasattr(self, "speaker_manager") and config.use_speaker_embedding:
if hasattr(self, "speaker_manager"):
if config.use_d_vector_file:
if speaker_name is None:
d_vector = self.speaker_manager.get_random_d_vector()
else:
d_vector = self.speaker_manager.get_d_vector_by_speaker(speaker_name)
else:
elif config.use_speaker_embedding:
if speaker_name is None:
speaker_id = self.speaker_manager.get_random_speaker_id()
else:
@ -284,7 +284,7 @@ class BaseTTS(BaseModel):
use_noise_augment=False if is_eval else config.use_noise_augment,
verbose=verbose,
speaker_id_mapping=speaker_id_mapping,
d_vector_mapping=d_vector_mapping if config.use_d_vector_file else None,
d_vector_mapping=d_vector_mapping,
language_id_mapping=language_id_mapping,
)