Add support for tdrz and debug mode

This commit is contained in:
Zero 2023-08-28 17:04:38 -06:00
parent 851fca36d7
commit 6668c4e2aa
No known key found for this signature in database
GPG key ID: 3861E636EA1E0E2B

View file

@ -206,6 +206,15 @@ impl<'a, 'b> FullParams<'a, 'b> {
self.fp.speed_up = speed_up; 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 /// # EXPERIMENTAL
/// ///
/// Overwrite the audio context size. 0 = default. /// Overwrite the audio context size. 0 = default.
@ -216,6 +225,16 @@ impl<'a, 'b> FullParams<'a, 'b> {
self.fp.audio_ctx = audio_ctx; 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. /// Set tokens to provide the model as initial input.
/// ///
/// These tokens are prepended to any existing text content from a previous call. /// These tokens are prepended to any existing text content from a previous call.