diff --git a/src/whisper_params.rs b/src/whisper_params.rs index 923f646..f9a2c6e 100644 --- a/src/whisper_params.rs +++ b/src/whisper_params.rs @@ -206,6 +206,15 @@ impl<'a, 'b> FullParams<'a, 'b> { self.fp.speed_up = speed_up; } + /// # EXPERIMENTAL + /// + /// Enables debug mode, such as dumping the log mel spectrogram. + /// + /// Defaults to false. + pub fn set_debug_mode(&mut self, debug: bool) { + self.fp.debug_mode = debug; + } + /// # EXPERIMENTAL /// /// Overwrite the audio context size. 0 = default. @@ -216,6 +225,16 @@ impl<'a, 'b> FullParams<'a, 'b> { self.fp.audio_ctx = audio_ctx; } + /// # EXPERIMENTAL + /// + /// Enable tinydiarize support. + /// Experimental speaker turn detection. + /// + /// Defaults to false. + pub fn set_tdrz_enable(&mut self, tdrz_enable: bool) { + self.fp.tdrz_enable = tdrz_enable; + } + /// Set tokens to provide the model as initial input. /// /// These tokens are prepended to any existing text content from a previous call.