whisper.cpp updated + suppress_non_speech_tokens renamed to suppress_nst

mgonzs13 · Jan 7, 2025 · ca26ac0 · ca26ac0
1 parent 0ee1bca
commit ca26ac0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/whisper_bringup/launch/whisper.launch.py b/whisper_bringup/launch/whisper.launch.py
@@ -70,9 +70,7 @@ def run_whisper(context: LaunchContext, repo, file, model_path):
             "language": LaunchConfiguration("language", default="en"),
             "detect_language": LaunchConfiguration("detect_language", default=False),
             "suppress_blank": LaunchConfiguration("suppress_blank", default=True),
-            "suppress_non_speech_tokens": LaunchConfiguration(
-                "suppress_non_speech_tokens", default=False
-            ),
+            "suppress_nst": LaunchConfiguration("suppress_nst", default=False),
             "temperature": LaunchConfiguration("temperature", default=0.00),
             "max_initial_ts": LaunchConfiguration("max_initial_ts", default=1.00),
             "length_penalty": LaunchConfiguration("length_penalty", default=-1.00),

diff --git a/whisper_cpp_vendor/CMakeLists.txt b/whisper_cpp_vendor/CMakeLists.txt
@@ -11,7 +11,7 @@ find_package(ament_cmake REQUIRED)
 FetchContent_Declare(
   whisper
   GIT_REPOSITORY https://github.com/ggerganov/whisper.cpp.git
-  GIT_TAG        v1.7.3
+  GIT_TAG        v1.7.4
 )
 
 FetchContent_MakeAvailable(whisper)

diff --git a/whisper_ros/src/whisper_ros/whisper_base_node.cpp b/whisper_ros/src/whisper_ros/whisper_base_node.cpp
@@ -74,7 +74,7 @@ WhisperBaseNode::WhisperBaseNode()
                                          {"speed_up", false},
                                          {"detect_language", false},
                                          {"suppress_blank", true},
-                                         {"suppress_non_speech_tokens", false},
+                                         {"suppress_nst", false},
                                          {"use_gpu", true},
                                          {"flash_attn", false},
                                          {"dtw_token_timestamps", false},
@@ -126,8 +126,7 @@ WhisperBaseNode::on_configure(const rclcpp_lifecycle::State &) {
   this->get_parameter("detect_language", this->wparams.detect_language);
 
   this->get_parameter("suppress_blank", this->wparams.suppress_blank);
-  this->get_parameter("suppress_non_speech_tokens",
-                      this->wparams.suppress_non_speech_tokens);
+  this->get_parameter("suppress_nst", this->wparams.suppress_nst);
 
   this->get_parameter("temperature", this->wparams.temperature);
   this->get_parameter("max_initial_ts", this->wparams.max_initial_ts);