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

Pass samples to init_from_config in SpeakerManager

This commit is contained in:
Eren Gölge 2021-12-07 12:54:39 +00:00
parent bde68d9f25
commit f802a931a3
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
TTS/tts/utils/speakers.py
View File

@ -319,23 +319,27 @@ class SpeakerManager:
raise NotImplementedError raise NotImplementedError
@staticmethod @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 """Initialize a speaker manager from config
Args: Args:
config (Coqpit): Config object. 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: Returns:
SpeakerEncoder: Speaker encoder object. SpeakerEncoder: Speaker encoder object.
""" """
speaker_manager = None 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): if config.get("speaker_file", None):
speaker_manager = SpeakerManager(speaker_id_file_path=config.speaker_file) speaker_manager = SpeakerManager(speaker_id_file_path=config.speaker_file)
if config.get("speakers_file", None): if config.get("speakers_file", None):
speaker_manager = SpeakerManager(speaker_id_file_path=config.speakers_file) 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): if config.get("speakers_file", None):
speaker_manager = SpeakerManager(d_vectors_file_path=config.speaker_file) speaker_manager = SpeakerManager(d_vectors_file_path=config.speaker_file)
if config.get("d_vector_file", None): if config.get("d_vector_file", None):