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