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,