From 15451d0f1c7052ead86eaa7b68584f74049ca816 Mon Sep 17 00:00:00 2001
From: Remc <caland.remi@gmail.com>
Date: Mon, 4 Sep 2023 17:08:19 +0200
Subject: [PATCH] fix: correct defaut_asr_options with new options (patch 0.8)

---
 whisperx/asr.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/whisperx/asr.py b/whisperx/asr.py
index da30774..20cc0c5 100644
--- a/whisperx/asr.py
+++ b/whisperx/asr.py
@@ -63,11 +63,14 @@ def load_model(whisper_arch,
         "best_of": 5,
         "patience": 1,
         "length_penalty": 1,
+        "repetition_penalty": 1,
+        "no_repeat_ngram_size": 0,
         "temperatures": [0.0, 0.2, 0.4, 0.6, 0.8, 1.0],
         "compression_ratio_threshold": 2.4,
         "log_prob_threshold": -1.0,
         "no_speech_threshold": 0.6,
         "condition_on_previous_text": False,
+        "prompt_reset_on_temperature": 0.5,
         "initial_prompt": None,
         "prefix": None,
         "suppress_blank": True,