harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Re-add suppress_nst function with shorter fn name

Browse source
This commit is contained in:
Christian Meter 2025-02-13 18:04:52 +01:00
parent f9fe2acf95
commit b752731e83
No known key found for this signature in database
GPG key ID: DC36932EEAE54549
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/whisper_params.rs
View file

@ -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.
///