Merge pull request ggerganov#101 from codesoda/grammar-fixes

KultivatorConsulting · Nov 24, 2023 · 1ad60ec · 1ad60ec
2 parents 2a17add + a07e27e
commit 1ad60ec
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/lib.rs b/src/lib.rs
@@ -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;
 

diff --git a/src/whisper_ctx.rs b/src/whisper_ctx.rs
@@ -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
     }