fix whisperx load error

matheusbach · Nov 20, 2023 · 5784370 · 5784370
1 parent dbc125c
commit 5784370
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/legen.py b/legen.py
@@ -124,7 +124,7 @@
         import whisperx_utils
 
         whisper_model = whisperx.load_model(
-            whisper_arch=args.model, device=torch_device, compute_type=compute_type)
+            whisper_arch=args.model, device=torch_device, compute_type=compute_type, asr_options={"repetition_penalty": 1, "prompt_reset_on_temperature": 0.5, "no_repeat_ngram_size": 2,})
     else:
         import whisper