Fix insanely_fast_whisper crash

jhj0517 · Oct 4, 2024 · c8e54db · c8e54db
1 parent 45fac7d
commit c8e54db
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/whisper/insanely_fast_whisper_inference.py b/modules/whisper/insanely_fast_whisper_inference.py
@@ -78,7 +78,8 @@ def transcribe(self,
             kwargs = {
                 "no_speech_threshold": params.no_speech_threshold,
                 "temperature": params.temperature,
-                "compression_ratio_threshold": params.compression_ratio_threshold
+                "compression_ratio_threshold": params.compression_ratio_threshold,
+                "logprob_threshold": params.log_prob_threshold,
             }
 
             if self.current_model_size.endswith(".en"):