From 73d27ebd45584020aafa9447ef8a22012b4ddb0e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 24 Nov 2021 18:42:44 +0100 Subject: [PATCH] Fix GlowTTS --- TTS/tts/models/glow_tts.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/TTS/tts/models/glow_tts.py b/TTS/tts/models/glow_tts.py index 9e779f8e..3e5226aa 100644 --- a/TTS/tts/models/glow_tts.py +++ b/TTS/tts/models/glow_tts.py @@ -14,6 +14,7 @@ from TTS.tts.models.base_tts import BaseTTS from TTS.tts.utils.helpers import generate_path, maximum_path, sequence_mask from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.synthesis import synthesis +from TTS.tts.utils.text.tokenizer import TTSTokenizer from TTS.tts.utils.visual import plot_alignment, plot_spectrogram from TTS.utils.io import load_fsspec @@ -513,3 +514,22 @@ class GlowTTS(BaseTTS): def on_train_step_start(self, trainer): """Decide on every training step wheter enable/disable data depended initialization.""" self.run_data_dep_init = trainer.total_steps_done < self.data_dep_init_steps + + @staticmethod + def init_from_config(config: Coqpit): + """Initialize model from config.""" + + # init characters + if config.use_phonemes: + from TTS.tts.utils.text.characters import IPAPhonemes + characters = IPAPhonemes().init_from_config(config) + else: + from TTS.tts.utils.text.characters import Graphemes + characters = Graphemes().init_from_config(config) + config.num_chars = characters.num_chars + + from TTS.utils.audio import AudioProcessor + ap = AudioProcessor.init_from_config(config) + tokenizer = TTSTokenizer.init_from_config(config) + speaker_manager = SpeakerManager.init_from_config(config) + return GlowTTS(config, ap, tokenizer, speaker_manager) \ No newline at end of file