diff --git a/whisper_bringup/launch/whisper.launch.py b/whisper_bringup/launch/whisper.launch.py index c7561f0..5262e0b 100644 --- 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 index ac36fab..7415d5f 100644 --- 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 index 07a7f6a..9df0450 100644 --- 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);