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

Get the number speaker from the Speaker Manager property

This commit is contained in:
Edresson 2021-11-22 18:52:59 -03:00 committed by Eren Gölge
parent 88f4369c63
commit 5782df8ffe
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/bin/train_tts.py
View File

@ -48,16 +48,15 @@ def main():
if config.use_speaker_embedding: if config.use_speaker_embedding:
speaker_manager = SpeakerManager(data_items=train_samples + eval_samples) speaker_manager = SpeakerManager(data_items=train_samples + eval_samples)
if hasattr(config, "model_args"): if hasattr(config, "model_args"):
config.model_args.num_speakers = len(speaker_manager.speaker_ids) config.model_args.num_speakers = speaker_manager.num_speakers
else: else:
config.num_speakers = len(speaker_manager.speaker_ids) config.num_speakers = speaker_manager.num_speakers
elif config.use_d_vector_file: elif config.use_d_vector_file:
speaker_manager = SpeakerManager(d_vectors_file_path=config.d_vector_file) speaker_manager = SpeakerManager(d_vectors_file_path=config.d_vector_file)
if hasattr(config, "model_args"): if hasattr(config, "model_args"):
config.model_args.num_speakers = len(speaker_manager.speaker_ids) config.model_args.num_speakers = speaker_manager.num_speakers
else: else:
config.num_speakers = len(speaker_manager.speaker_ids) config.num_speakers = speaker_manager.num_speakers
else: else:
speaker_manager = None speaker_manager = None