chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #458 from remic33/correct_default_asr_options

Browse Source 
fix: correct defaut_asr_options with new options (patch 0.8)
This commit is contained in:
Max Bain
2023-09-04 09:22:16 -06:00
committed by GitHub
parent 8c4a21b66d 15451d0f1c
commit 15d9e08d3e
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
whisperx/asr.py
View File

@ -63,11 +63,14 @@ def load_model(whisper_arch,
"best_of": 5, "best_of": 5,
"patience": 1, "patience": 1,
"length_penalty": 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], "temperatures": [0.0, 0.2, 0.4, 0.6, 0.8, 1.0],
"compression_ratio_threshold": 2.4, "compression_ratio_threshold": 2.4,
"log_prob_threshold": -1.0, "log_prob_threshold": -1.0,
"no_speech_threshold": 0.6, "no_speech_threshold": 0.6,
"condition_on_previous_text": False, "condition_on_previous_text": False,
"prompt_reset_on_temperature": 0.5,
"initial_prompt": None, "initial_prompt": None,
"prefix": None, "prefix": None,
"suppress_blank": True, "suppress_blank": True,