diff --git a/transformers/run_whisper.sh b/transformers/run_whisper.sh index b3813ff..dd348b9 100755 --- a/transformers/run_whisper.sh +++ b/transformers/run_whisper.sh @@ -2,7 +2,7 @@ export PYTHONPATH="..":$PYTHONPATH -MODEL_IDs=("openai/whisper-tiny.en" "openai/whisper-small.en" "openai/whisper-base.en" "openai/whisper-medium.en" "openai/whisper-large" "openai/whisper-large-v2" "openai/whisper-large-v3" "distil-whisper/distil-medium.en" "distil-whisper/distil-large-v2" "distil-whisper/distil-large-v3" "nyrahealth/CrisperWhisper") +MODEL_IDs=("openai/whisper-tiny.en" "openai/whisper-small.en" "openai/whisper-base.en" "openai/whisper-medium.en" "openai/whisper-large" "openai/whisper-large-v2" "openai/whisper-large-v3" "distil-whisper/distil-medium.en" "distil-whisper/distil-large-v2" "distil-whisper/distil-large-v3" "shunyalabs/pingala-v1-universal" "nyrahealth/CrisperWhisper") BATCH_SIZE=64 num_models=${#MODEL_IDs[@]}