From b752731e8344d9ecfa5923da024f448d6a7c12da Mon Sep 17 00:00:00 2001
From: Christian Meter <cmeter@googlemail.com>
Date: Thu, 13 Feb 2025 18:04:52 +0100
Subject: [PATCH] Re-add suppress_nst function with shorter fn name

---
 src/whisper_params.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/whisper_params.rs b/src/whisper_params.rs
index 34b4adc..36a709e 100644
--- a/src/whisper_params.rs
+++ b/src/whisper_params.rs
@@ -299,6 +299,15 @@ impl<'a, 'b> FullParams<'a, 'b> {
         self.fp.suppress_blank = suppress_blank;
     }
 
+    /// Set suppress_non_speech_tokens.
+    /// See <https://github.com/openai/whisper/blob/7858aa9c08d98f75575035ecd6481f462d66ca27/whisper/tokenizer.py#L224-L253>
+    /// for more information.
+    ///
+    /// Defaults to false.
+    pub fn set_suppress_nst(&mut self, suppress_nst: bool) {
+        self.fp.suppress_nst = suppress_nst;
+    }
+
     /// Set initial decoding temperature.
     /// See <https://ai.stackexchange.com/a/32478> for more information.
     ///