add language arg to lightning whisper handler
This commit is contained in:
@@ -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:
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user