File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -22,7 +22,7 @@ function get_script_path() {
2222models_path=" $( get_script_path) "
2323
2424# Whisper models
25- models=( " tiny.en" " tiny" " base.en" " base" " small.en" " small" " medium.en" " medium" " large-v1" " large" )
25+ models=( " tiny.en" " tiny" " base.en" " base" " small.en" " small.en-tdrz " " small " " medium.en" " medium" " large-v1" " large" )
2626
2727# list available models
2828function list_models {
@@ -50,6 +50,12 @@ if [[ ! " ${models[@]} " =~ " ${model} " ]]; then
5050 exit 1
5151fi
5252
53+ # check if model contains `tdrz` and update the src and pfx accordingly
54+ if [[ $model == * " tdrz" * ]]; then
55+ src=" https://huggingface.co/akashmjn/tinydiarize-whisper.cpp"
56+ pfx=" resolve/main/ggml"
57+ fi
58+
5359# download ggml model
5460
5561printf " Downloading ggml model $model from '$src ' ...\n"
You can’t perform that action at this time.
0 commit comments