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

fixy

Browse Source
This commit is contained in:
Andres Marafioti
2024-08-29 10:34:10 +02:00
parent 2cb9464b8f
commit 2f266b2d95
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/melo_handler.py
View File

@@ -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()