From 1afd2445d3594e3e9c59dbb8b51df6debc55f669 Mon Sep 17 00:00:00 2001
From: rchan <rchan@turing.ac.uk>
Date: Wed, 4 Sep 2024 17:12:09 +0100
Subject: [PATCH] add language arg to lightning whisper handler
---
STT/lightning_whisper_mlx_handler.py | 1 +
s2s_pipeline.py | 1 -
2 files changed, 1 insertion(+), 1 deletion(-)
diff --git a/STT/lightning_whisper_mlx_handler.py b/STT/lightning_whisper_mlx_handler.py
index e5cd540..4785b73 100644
--- a/STT/lightning_whisper_mlx_handler.py
+++ b/STT/lightning_whisper_mlx_handler.py
@@ -22,6 +22,7 @@ class LightningWhisperSTTHandler(BaseHandler):
device="cuda",
torch_dtype="float16",
compile_mode=None,
+ language=None,
gen_kwargs={},
):
if len(model_name.split("/")) > 1:
diff --git a/s2s_pipeline.py b/s2s_pipeline.py
index 8da8298..9e5bb04 100644
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -299,7 +299,6 @@ def main():
setup_args=(should_listen,),
setup_kwargs=vars(parler_tts_handler_kwargs),
)
-
elif module_kwargs.tts == "melo":
try:
from TTS.melo_handler import MeloTTSHandler
--
GitLab