diff --git a/STT/lightning_whisper_mlx_handler.py b/STT/lightning_whisper_mlx_handler.py index e5cd540a968d4a9e29ef6b1fe1c796f9f89bd571..4785b73853275e5a308ae9da48667b0b656297df 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 8da829834e85c856458a571bf3c7242500d8ae6b..9e5bb0493f6195e1d7fc31f5f4a87dc5b0da014f 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