···

       
13
       
13
       
 
       


     

       
14
       
14
       
 
       
stdenv.mkDerivation rec {

     

       
15
       
15
       
 
       
  pname = "whisper-cpp";

     

       
16
       
16
       
-
       
  version = "1.5.2";

     

       
16
       
16
       
+
       
  version = "1.5.4";

     

       
17
       
17
       
 
       


     

       
18
       
18
       
 
       
  src = fetchFromGitHub {

     

       
19
       
19
       
 
       
    owner = "ggerganov";

     

       
20
       
20
       
 
       
    repo = "whisper.cpp";

     

       
21
       
21
       
 
       
    rev = "refs/tags/v${version}" ;

     

       
22
       
22
       
-
       
    hash = "sha256-7pJbROifDajBJUE07Nz8tARB901fWCB+TS4okcnEsvc=";

     

       
22
       
22
       
+
       
    hash = "sha256-9H2Mlua5zx2WNXbz2C5foxIteuBgeCNALdq5bWyhQCk=";

     

       
23
       
23
       
 
       
  };

     

       
24
       
24
       
 
       


     

       
25
       
25
       
 
       
  # The upstream download script tries to download the models to the

     

···

       
1
       
1
       
-
       
diff --git a/models/download-ggml-model.sh b/models/download-ggml-model.sh

     

       
2
       
2
       
-
       
index 749b409..831f4c0 100755

     

       
3
       
1
       
 
       
--- a/models/download-ggml-model.sh

     

       
4
       
2
       
 
       
+++ b/models/download-ggml-model.sh

     

       
5
       
3
       
 
       
@@ -9,18 +9,6 @@

     
···

       
16
       
14
       
 
       
-    fi

     

       
17
       
15
       
 
       
-}

     

       
18
       
16
       
 
       
-

     

       
19
       
19
       
-
       
-models_path="$(get_script_path)"

     

       
17
       
17
       
+
       
-models_path="${2:-$(get_script_path)}"

     

       
20
       
18
       
 
       
-

     

       
21
       
19
       
 
       
 # Whisper models

     

       
22
       
20
       
 
       
 models=(

     

       
23
       
21
       
 
       
     "tiny.en"

     

       
22
       
22
       
+
       
@@ -56,8 +44,8 @@ function list_models {

     

       
23
       
23
       
+
       
     printf "\n\n"

     

       
24
       
24
       
+
       
 }

     

       
25
       
25
       
+
       
 

     

       
26
       
26
       
+
       
-if [ "$#" -lt 1 ] || [ "$#" -gt 2 ]; then

     

       
27
       
27
       
+
       
-    printf "Usage: $0 <model> [models_path]\n"

     

       
28
       
28
       
+
       
+if [ "$#" -ne 1 ]; then

     

       
29
       
29
       
+
       
+    printf "Usage: $0 <model>\n"

     

       
30
       
30
       
+
       
     list_models

     

       
31
       
31
       
+
       
 

     

       
32
       
32
       
+
       
     exit 1

     

       
24
       
33
       
 
       
@@ -82,8 +70,6 @@ fi

     

       
25
       
34
       
 
       
 

     

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

     
···

       
34
       
43
       
 
       
     exit 1

     

       
35
       
44
       
 
       
 fi

     

       
36
       
45
       
 
       
 

     

       
37
       
37
       
-
       
-printf "Done! Model '$model' saved in 'models/ggml-$model.bin'\n"

     

       
46
       
46
       
+
       
-printf "Done! Model '$model' saved in '$models_path/ggml-$model.bin'\n"

     

       
38
       
47
       
 
       
+printf "Done! Model '$model' saved in 'ggml-$model.bin'\n"

     

       
39
       
48
       
 
       
 printf "You can now use it like this:\n\n"

     

       
40
       
40
       
-
       
-printf "  $ ./main -m models/ggml-$model.bin -f samples/jfk.wav\n"

     

       
49
       
49
       
+
       
-printf "  $ ./main -m $models_path/ggml-$model.bin -f samples/jfk.wav\n"

     

       
41
       
50
       
 
       
+printf "  $ whisper-cpp -m ggml-$model.bin -f samples/jfk.wav\n"

     

       
42
       
51
       
 
       
 printf "\n"