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

Add new nosp token

Browse source
This commit is contained in:
Zero 2023-08-28 17:04:08 -06:00
parent 5c140c14d4
commit 851fca36d7
No known key found for this signature in database
GPG key ID: 3861E636EA1E0E2B
1 changed files with 13 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/whisper_ctx.rs
View file

@ -405,6 +405,15 @@ impl WhisperContext {
unsafe { whisper_rs_sys::whisper_token_sot(self.ctx) }
}
/// Get the ID of the solm token.
///
/// # C++ equivalent
/// `whisper_token whisper_token_solm(struct whisper_context * ctx)`
#[inline]
pub fn token_solm(&self) -> WhisperToken {
unsafe { whisper_rs_sys::whisper_token_solm(self.ctx) }
}
/// Get the ID of the prev token.
///
/// # C++ equivalent
@ -414,13 +423,13 @@ impl WhisperContext {
unsafe { whisper_rs_sys::whisper_token_prev(self.ctx) }
}
/// Get the ID of the solm token.
/// Get the ID of the nosp token.
///
/// # C++ equivalent
/// `whisper_token whisper_token_solm(struct whisper_context * ctx)`
/// `whisper_token whisper_token_nosp(struct whisper_context * ctx)`
#[inline]
pub fn token_solm(&self) -> WhisperToken {
unsafe { whisper_rs_sys::whisper_token_solm(self.ctx) }
pub fn token_nosp(&self) -> WhisperToken {
unsafe { whisper_rs_sys::whisper_token_nosp(self.ctx) }
}
/// Get the ID of the not token.