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() {
22
22
models_path=" $( get_script_path) "
23
23
24
24
# 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" )
26
26
27
27
# list available models
28
28
function list_models {
@@ -50,6 +50,12 @@ if [[ ! " ${models[@]} " =~ " ${model} " ]]; then
50
50
exit 1
51
51
fi
52
52
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
+
53
59
# download ggml model
54
60
55
61
printf " Downloading ggml model $model from '$src ' ...\n"
You can’t perform that action at this time.
0 commit comments