subsai
subsai copied to clipboard
whisperX results in "unexpected keyword argument error for 3 fields
Command I input to windows powershell:
subsai .\file --model m-bain/whisperX --model-configs '{\"model_type\": \"large-v2\", \"device\": \"cuda\"}' --format srt
And result is:
TypeError: <lambda>() got an unexpected keyword argument 'repetition_penalty'
Reading the source code for whisperX there is a repetition_penalty field that isn't present in your code:
https://github.com/m-bain/whisperX/blob/b1a98b78c9152ace9f9801593b5fa0c7d5d96b0f/whisperx/asr.py#L66
default_asr_options = { "beam_size": 5, "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, "suppress_tokens": [-1], "without_timestamps": True, "max_initial_timestamp": 0.0, "word_timestamps": False, "prepend_punctuations": ""'“¿([{-", "append_punctuations": ""'.。,,!!??::”)]}、", "suppress_numerals": False, }
But you don't have all of these as is, so may not be required