Skip to content

Commit 7f0dc9b

Browse files
committed
add HuggingFace mirror to download ggml model
1 parent 72deb41 commit 7f0dc9b

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

models/download-ggml-model.sh

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ function get_script_path() {
2222
models_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
2828
function list_models {
@@ -50,6 +50,12 @@ if [[ ! " ${models[@]} " =~ " ${model} " ]]; then
5050
exit 1
5151
fi
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

5561
printf "Downloading ggml model $model from '$src' ...\n"

0 commit comments

Comments
 (0)