]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
Fix: main get n_threads from cli
authorАртём Земляк <redacted>
Wed, 5 Oct 2022 02:47:48 +0000 (09:47 +0700)
committerАртём Земляк <redacted>
Wed, 5 Oct 2022 02:47:48 +0000 (09:47 +0700)
main.cpp

index eb5bb4339fe6e7f6136a6af5f5cb834a77a7929b..1885eb6d40624c272fef9bbb066ebd0653dc9241 100644 (file)
--- a/main.cpp
+++ b/main.cpp
@@ -187,6 +187,7 @@ int main(int argc, char ** argv) {
         wparams.print_special_tokens = params.print_special_tokens;
         wparams.translate            = params.translate;
         wparams.language             = params.language.c_str();
+        wparams.n_threads            = params.n_threads;
 
         if (whisper_full(ctx, wparams, pcmf32.data(), pcmf32.size()) != 0) {
             fprintf(stderr, "%s: failed to process audio\n", argv[0]);