From 2f266b2d9587aa1967e48cf7f92583c3f94143d8 Mon Sep 17 00:00:00 2001 From: Andres Marafioti <andimarafioti@gmail.com> Date: Thu, 29 Aug 2024 10:34:10 +0200 Subject: [PATCH] fixy --- TTS/melo_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py index d5cd53c..6b5d831 100644 --- a/TTS/melo_handler.py +++ b/TTS/melo_handler.py @@ -43,8 +43,8 @@ class MeloTTSHandler(BaseHandler): self.should_listen = should_listen self.device = device self.language = "<|" + language + "|>" # 'Tokenize' the language code to do less operations - self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[language], device=device) - self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[speaker_to_id]] + self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[self.language], device=device) + self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER["<|" + speaker_to_id + "|>"]] self.blocksize = blocksize self.warmup() -- GitLab