add language arg to lightning whisper handler

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

View File

@@ -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:

View File

@@ -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