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

Delightful TTS VCTK recipe fixes (#2808) 

* fix: wrong import class

* fix: formatter name missing

* feat: get rid of clearml
This commit is contained in:
AWAS666 2023-07-31 10:27:42 +02:00 committed by GitHub
parent d124f78430
commit 9e74b51aa6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
recipes/vctk/delightful_tts/train_delightful_tts.py
View File

@ -1,22 +1,20 @@
import os import os
from clearml import Task
from trainer import Trainer, TrainerArgs from trainer import Trainer, TrainerArgs
from TTS.config.shared_configs import BaseDatasetConfig from TTS.config.shared_configs import BaseDatasetConfig
from TTS.tts.configs.delightful_tts_config import DelightfulTtsAudioConfig, DelightfulTTSConfig from TTS.tts.configs.delightful_tts_config import DelightfulTtsAudioConfig, DelightfulTTSConfig
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.tts.models.delightful_tts import DelightfulTtsArgs, DelightfulTTSE2e, VocoderConfig from TTS.tts.models.delightful_tts import DelightfulTtsArgs, DelightfulTTS, VocoderConfig
from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.speakers import SpeakerManager
from TTS.tts.utils.text.tokenizer import TTSTokenizer from TTS.tts.utils.text.tokenizer import TTSTokenizer
from TTS.utils.audio.processor import AudioProcessor from TTS.utils.audio.processor import AudioProcessor
task = Task.init(project_name="delightful-tts", task_name="vctk")
data_path = "/raid/datasets/vctk_v092_48khz_removed_silence_silero_vad" data_path = "/raid/datasets/vctk_v092_48khz_removed_silence_silero_vad"
output_path = os.path.dirname(os.path.abspath(__file__)) output_path = os.path.dirname(os.path.abspath(__file__))
dataset_config = BaseDatasetConfig(dataset_name="vctk", meta_file_train="", path=data_path, language="en-us") dataset_config = BaseDatasetConfig(dataset_name="vctk", formatter="vctk", meta_file_train="", path=data_path, language="en-us")
audio_config = DelightfulTtsAudioConfig() audio_config = DelightfulTtsAudioConfig()
@ -25,7 +23,7 @@ model_args = DelightfulTtsArgs()
vocoder_config = VocoderConfig() vocoder_config = VocoderConfig()
something_tts_config = DelightfulTTSConfig( something_tts_config = DelightfulTTSConfig(
run_name="delightful_tts_e2e_ljspeech", run_name="delightful_tts_vctk",
run_description="Train like in delightful tts paper.", run_description="Train like in delightful tts paper.",
model_args=model_args, model_args=model_args,
audio=audio_config, audio=audio_config,
@ -75,7 +73,7 @@ speaker_manager.set_ids_from_data(train_samples + eval_samples, parse_key="speak
config.model_args.num_speakers = speaker_manager.num_speakers config.model_args.num_speakers = speaker_manager.num_speakers
model = DelightfulTTSE2e( model = DelightfulTTS(
ap=ap, config=config, tokenizer=tokenizer, speaker_manager=speaker_manager, emotion_manager=None ap=ap, config=config, tokenizer=tokenizer, speaker_manager=speaker_manager, emotion_manager=None
) )