coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove the call to get_speaker_manager

pull/1032/head
Edresson 2021-11-22 09:18:29 -03:00 committed by Eren Gölge
parent b769b49e34
commit 34749f8727
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
TTS/bin/extract_tts_spectrograms.py
View File

@ -12,7 +12,7 @@ from tqdm import tqdm
from TTS.config import load_config from TTS.config import load_config
from TTS.tts.datasets import TTSDataset, load_tts_samples from TTS.tts.datasets import TTSDataset, load_tts_samples
from TTS.tts.models import setup_model from TTS.tts.models import setup_model
from TTS.tts.utils.speakers import get_speaker_manager from TTS.tts.utils.speakers import SpeakerManager
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.generic_utils import count_parameters from TTS.utils.generic_utils import count_parameters
@ -234,8 +234,13 @@ def main(args): # pylint: disable=redefined-outer-name
# use eval and training partitions # use eval and training partitions
meta_data = meta_data_train + meta_data_eval meta_data = meta_data_train + meta_data_eval
# parse speakers # init speaker manager
speaker_manager = get_speaker_manager(c, args, meta_data_train) if config.use_speaker_embedding:
speaker_manager = SpeakerManager(data_items=meta_data)
elif config.use_d_vector_file:
speaker_manager = SpeakerManager(d_vectors_file_path=c.d_vector_file)
else:
speaker_manager = None
# setup model # setup model
model = setup_model(c) model = setup_model(c)