alihan/speech-to-speech
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

catch exception if language is not supported

Browse Source
This commit is contained in:
Andres Marafioti
2024-08-29 10:50:18 +02:00
parent 712005aff0
commit 3fff1d1da0
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/melo_handler.py
View File

@@ -62,9 +62,12 @@ class MeloTTSHandler(BaseHandler):
console.print(f"[green]ASSISTANT: {llm_sentence}")
if language_id is not None and self.language != language_id:
self.language = language_id
self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[self.language], device=self.device)
self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[self.language]]
try:
self.model = TTS(language=WHISPER_LANGUAGE_TO_MELO_LANGUAGE[language_id], device=self.device)
self.speaker_id = self.model.hps.data.spk2id[WHISPER_LANGUAGE_TO_MELO_SPEAKER[language_id]]
self.language = language_id
except KeyError:
console.print(f"[red]Language {language_id} not supported by Melo. Using {self.language} instead.")
if self.device == "mps":
import time