Merge pull request #84 from rchan26/main

Add language arg to lightning whisper handler

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:

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