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 d-vector multispeaker training bug

This commit is contained in:
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 # 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") and config.use_speaker_embedding: if hasattr(self, "speaker_manager"):
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_d_vector() d_vector = self.speaker_manager.get_random_d_vector()
else: else:
d_vector = self.speaker_manager.get_d_vector_by_speaker(speaker_name) d_vector = self.speaker_manager.get_d_vector_by_speaker(speaker_name)
else: elif config.use_speaker_embedding:
if speaker_name is None: if speaker_name is None:
speaker_id = self.speaker_manager.get_random_speaker_id() speaker_id = self.speaker_manager.get_random_speaker_id()
else: else:
@ -284,7 +284,7 @@ class BaseTTS(BaseModel):
use_noise_augment=False if is_eval else config.use_noise_augment, use_noise_augment=False if is_eval else config.use_noise_augment,
verbose=verbose, verbose=verbose,
speaker_id_mapping=speaker_id_mapping, 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, language_id_mapping=language_id_mapping,
) )