]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
node : add max_len params in node addon (#2760)
authorbillyct <redacted>
Mon, 3 Feb 2025 20:49:06 +0000 (04:49 +0800)
committerGitHub <redacted>
Mon, 3 Feb 2025 20:49:06 +0000 (22:49 +0200)
examples/addon.node/addon.cpp
examples/addon.node/index.js

index 4ada6ca508489e42729caf8fe01ebe1f38abbc01..656bfe3f1dcf69b247aebe7fdad8c77136549054 100644 (file)
@@ -330,6 +330,7 @@ Napi::Value whisper(const Napi::CallbackInfo& info) {
   bool no_timestamps = whisper_params.Get("no_timestamps").As<Napi::Boolean>();
   int32_t audio_ctx = whisper_params.Get("audio_ctx").As<Napi::Number>();
   bool comma_in_time = whisper_params.Get("comma_in_time").As<Napi::Boolean>();
+  int32_t max_len = whisper_params.Get("max_len").As<Napi::Number>();
 
   Napi::Value pcmf32Value = whisper_params.Get("pcmf32");
   std::vector<float> pcmf32_vec;
@@ -352,6 +353,7 @@ Napi::Value whisper(const Napi::CallbackInfo& info) {
   params.audio_ctx = audio_ctx;
   params.pcmf32 = pcmf32_vec;
   params.comma_in_time = comma_in_time;
+  params.max_len = max_len;
 
   Napi::Function callback = info[1].As<Napi::Function>();
   Worker* worker = new Worker(callback, params);
index 643ee756452de62f347eeace89a7ffbc4d8c7b3f..65fa31f8784979ed61df500b14faa597b32fb593 100644 (file)
@@ -18,6 +18,7 @@ const whisperParams = {
   translate: true,
   no_timestamps: false,
   audio_ctx: 0,
+  max_len: 0,
 };
 
 const arguments = process.argv.slice(2);