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

add language arg to lightning whisper handler

Browse Source
This commit is contained in:
rchan
2024-09-04 17:12:09 +01:00
parent 8afd078ab4
commit 1afd2445d3
2 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
STT/lightning_whisper_mlx_handler.py
View File

@@ -22,6 +22,7 @@ class LightningWhisperSTTHandler(BaseHandler):
device="cuda", device="cuda",
torch_dtype="float16", torch_dtype="float16",
compile_mode=None, compile_mode=None,
language=None,
gen_kwargs={}, gen_kwargs={},
): ):
if len(model_name.split("/")) > 1: if len(model_name.split("/")) > 1:

1
s2s_pipeline.py
View File

@@ -299,7 +299,6 @@ def main():
setup_args=(should_listen,), setup_args=(should_listen,),
setup_kwargs=vars(parler_tts_handler_kwargs), setup_kwargs=vars(parler_tts_handler_kwargs),
) )
elif module_kwargs.tts == "melo": elif module_kwargs.tts == "melo":
try: try:
from TTS.melo_handler import MeloTTSHandler from TTS.melo_handler import MeloTTSHandler