Skip to content
Snippets Groups Projects
Commit e2c9d968 authored by Andres Marafioti's avatar Andres Marafioti
Browse files

let whisper determine the language

parent 832ee799
No related branches found
No related tags found
No related merge requests found
......@@ -30,6 +30,7 @@ class WhisperSTTHandler(BaseHandler):
self.torch_dtype = getattr(torch, torch_dtype)
self.compile_mode = compile_mode
self.gen_kwargs = gen_kwargs
del self.gen_kwargs["language"]
self.processor = AutoProcessor.from_pretrained(model_name)
self.model = AutoModelForSpeechSeq2Seq.from_pretrained(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment