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

Merge branch 'master' into add_initial_prompt_param

Browse source
This commit is contained in:
Chris Raethke 2023-11-27 12:57:18 +10:00 • committed by GitHub
commit 41d73af64c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

1
src/lib.rs
View file

@ -14,6 +14,7 @@ pub use standalone::*;
pub use utilities::*;
pub use whisper_ctx::WhisperContext;
pub use whisper_ctx::WhisperContextParameters;
pub use whisper_grammar::{WhisperGrammarElement, WhisperGrammarElementType};
pub use whisper_params::{FullParams, SamplingStrategy};
pub use whisper_state::WhisperState;

2
src/whisper_ctx.rs
View file

@ -551,7 +551,7 @@ impl WhisperContextParameters {
pub fn new() -> Self {
Self::default()
}
pub fn use_gpu(mut self, use_gpu: bool) -> Self {
pub fn use_gpu(&mut self, use_gpu: bool) -> &mut Self {
self.use_gpu = use_gpu;
self
}