Skip to content

Commit

Permalink
add HuggingFace mirror to download ggml model
Browse files Browse the repository at this point in the history
  • Loading branch information
akashmjn committed Jun 26, 2023
1 parent 72deb41 commit 7f0dc9b
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion models/download-ggml-model.sh
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ function get_script_path() {
models_path="$(get_script_path)"

# Whisper models
models=( "tiny.en" "tiny" "base.en" "base" "small.en" "small" "medium.en" "medium" "large-v1" "large" )
models=( "tiny.en" "tiny" "base.en" "base" "small.en" "small.en-tdrz" "small" "medium.en" "medium" "large-v1" "large" )

# list available models
function list_models {
Expand Down Expand Up @@ -50,6 +50,12 @@ if [[ ! " ${models[@]} " =~ " ${model} " ]]; then
exit 1
fi

# check if model contains `tdrz` and update the src and pfx accordingly
if [[ $model == *"tdrz"* ]]; then
src="https://huggingface.co/akashmjn/tinydiarize-whisper.cpp"
pfx="resolve/main/ggml"
fi

# download ggml model

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

0 comments on commit 7f0dc9b

Please sign in to comment.