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

change return types

Browse source
This commit is contained in:
0/0 2022-12-14 17:04:38 -07:00
parent 8317096daa
commit 462f6c7b13
No known key found for this signature in database
GPG key ID: 3861E636EA1E0E2B
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/whisper_ctx.rs
View file

@ -194,7 +194,7 @@ impl WhisperContext {
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `whisper_token whisper_sample_best(struct whisper_context * ctx, bool need_timestamp)` /// `whisper_token whisper_sample_best(struct whisper_context * ctx, bool need_timestamp)`
pub fn sample_best(&mut self) -> Result<WhisperToken, WhisperError> { pub fn sample_best(&mut self) -> Result<WhisperTokenData, WhisperError> {
if !self.decode_once { if !self.decode_once {
return Err(WhisperError::DecodeNotComplete); return Err(WhisperError::DecodeNotComplete);
} }
@ -210,11 +210,11 @@ impl WhisperContext {
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `whisper_token whisper_sample_timestamp(struct whisper_context * ctx)` /// `whisper_token whisper_sample_timestamp(struct whisper_context * ctx)`
pub fn sample_timestamp(&mut self) -> Result<WhisperToken, WhisperError> { pub fn sample_timestamp(&mut self, is_initial: bool) -> Result<WhisperTokenData, WhisperError> {
if !self.decode_once { if !self.decode_once {
return Err(WhisperError::DecodeNotComplete); return Err(WhisperError::DecodeNotComplete);
} }
let ret = unsafe { whisper_rs_sys::whisper_sample_timestamp(self.ctx) }; let ret = unsafe { whisper_rs_sys::whisper_sample_timestamp(self.ctx, is_initial) };
Ok(ret) Ok(ret)
} }