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 the bug in multispeaker vits

This commit is contained in:
Edresson 2021-11-22 18:43:56 -03:00 committed by Eren Gölge
parent aa1a070d58
commit 88f4369c63
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/bin/train_tts.py
View File

@ -47,8 +47,17 @@ def main():
# init speaker manager
if config.use_speaker_embedding:
speaker_manager = SpeakerManager(data_items=train_samples + eval_samples)
if hasattr(config, "model_args"):
config.model_args.num_speakers = len(speaker_manager.speaker_ids)
else:
config.num_speakers = len(speaker_manager.speaker_ids)
elif config.use_d_vector_file:
speaker_manager = SpeakerManager(d_vectors_file_path=config.d_vector_file)
if hasattr(config, "model_args"):
config.model_args.num_speakers = len(speaker_manager.speaker_ids)
else:
config.num_speakers = len(speaker_manager.speaker_ids)
else:
speaker_manager = None

6
tests/tts_tests/test_vits_multilingual_train.py
View File

@ -53,9 +53,15 @@ config.audio.trim_db = 60
# active multilingual mode
config.model_args.use_language_embedding = True
config.use_language_embedding = True
# active multispeaker mode
config.model_args.use_speaker_embedding = True
config.use_speaker_embedding = True
# config.num_speakers=1
# config.model_args.num_speakers=1
config.model_args.use_d_vector_file = False
config.use_d_vector_file = False
# active language sampler
config.use_language_weighted_sampler = True