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

Update train_tts for multi-speaker

This commit is contained in:
Eren Gölge 2021-10-21 16:15:41 +00:00
parent 2b7d159383
commit a409e0f8f8
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
TTS/bin/train_tts.py
View File

@ -4,6 +4,7 @@ from TTS.config import load_config, register_config
from TTS.trainer import Trainer, TrainingArgs from TTS.trainer import Trainer, TrainingArgs
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models import setup_model from TTS.tts.models import setup_model
from TTS.tts.utils.speakers import SpeakerManager
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
@ -43,8 +44,16 @@ def main():
# setup audio processor # setup audio processor
ap = AudioProcessor(**config.audio) ap = AudioProcessor(**config.audio)
# init speaker manager
if config.use_speaker_embedding:
speaker_manager = SpeakerManager(data_items=train_samples + eval_samples)
elif config.use_d_vector_file:
speaker_manager = SpeakerManager(d_vectors_file_path=config.d_vector_file)
else:
speaker_manager = None
# init the model from config # init the model from config
model = setup_model(config) model = setup_model(config, speaker_manager)
# init the trainer and 🚀 # init the trainer and 🚀
trainer = Trainer( trainer = Trainer(