From f802a931a399de8d563facc38c2d618f7cde0b25 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Tue, 7 Dec 2021 12:54:39 +0000 Subject: [PATCH] Pass samples to init_from_config in SpeakerManager --- TTS/tts/utils/speakers.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/TTS/tts/utils/speakers.py b/TTS/tts/utils/speakers.py index c556db79..ba48f27c 100644 --- a/TTS/tts/utils/speakers.py +++ b/TTS/tts/utils/speakers.py @@ -319,23 +319,27 @@ class SpeakerManager: raise NotImplementedError @staticmethod - def init_from_config(config: "Coqpit"): + def init_from_config(config: "Coqpit", samples: Union[List[List], List[Dict]] = None) -> "SpeakerManager": """Initialize a speaker manager from config Args: config (Coqpit): Config object. + samples (Union[List[List], List[Dict]], optional): List of data samples to parse out the speaker names. + Defaults to None. Returns: SpeakerEncoder: Speaker encoder object. """ speaker_manager = None - if hasattr(config, "use_speaker_embedding") and config.use_speaker_embedding is True: + if hasattr(config, "use_speaker_embedding") and config.use_speaker_embedding: + if samples: + speaker_manager = SpeakerManager(data_items=samples) if config.get("speaker_file", None): speaker_manager = SpeakerManager(speaker_id_file_path=config.speaker_file) if config.get("speakers_file", None): speaker_manager = SpeakerManager(speaker_id_file_path=config.speakers_file) - if hasattr(config, "use_d_vector_file") and config.use_speaker_embedding is True: + if hasattr(config, "use_d_vector_file") and config.use_d_vector_file: if config.get("speakers_file", None): speaker_manager = SpeakerManager(d_vectors_file_path=config.speaker_file) if config.get("d_vector_file", None):