Skip to content

Commit

Permalink
Delightful TTS VCTK recipe fixes (#2808)
Browse files Browse the repository at this point in the history
* fix: wrong import class

* fix: formatter name missing

* feat: get rid of clearml
  • Loading branch information
AWAS666 committed Jul 31, 2023
1 parent d124f78 commit 9e74b51
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions recipes/vctk/delightful_tts/train_delightful_tts.py
Original file line number Diff line number Diff line change
@@ -1,22 +1,20 @@
import os

from clearml import Task
from trainer import Trainer, TrainerArgs

from TTS.config.shared_configs import BaseDatasetConfig
from TTS.tts.configs.delightful_tts_config import DelightfulTtsAudioConfig, DelightfulTTSConfig
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.text.tokenizer import TTSTokenizer
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"
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()

Expand All @@ -25,7 +23,7 @@
vocoder_config = VocoderConfig()

something_tts_config = DelightfulTTSConfig(
run_name="delightful_tts_e2e_ljspeech",
run_name="delightful_tts_vctk",
run_description="Train like in delightful tts paper.",
model_args=model_args,
audio=audio_config,
Expand Down Expand Up @@ -75,7 +73,7 @@
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
)

Expand Down

0 comments on commit 9e74b51

Please sign in to comment.