Merge branch 'ysimonson-update-whisper'

This commit is contained in:
Niko 2025-12-06 11:47:33 -08:00
commit 298e0ddabd
No known key found for this signature in database
3 changed files with 6 additions and 4 deletions

View file

@ -51,15 +51,16 @@ fn main() {
// note that you don't need to use these, you can do it yourself or any other way you want // note that you don't need to use these, you can do it yourself or any other way you want
// these are just provided for convenience // these are just provided for convenience
let mut inter_samples = vec![Default::default(); samples.len()]; let mut inter_samples = vec![Default::default(); samples.len()];
let mut mono_samples = vec![Default::default(); samples.len() / 2];
whisper_rs::convert_integer_to_float_audio(&samples, &mut inter_samples) whisper_rs::convert_integer_to_float_audio(&samples, &mut inter_samples)
.expect("failed to convert audio data"); .expect("failed to convert audio data");
let samples = whisper_rs::convert_stereo_to_mono_audio(&inter_samples) whisper_rs::convert_stereo_to_mono_audio(&inter_samples, &mut mono_samples)
.expect("failed to convert audio data"); .expect("failed to convert audio data");
// now we can run the model // now we can run the model
state state
.full(params, &samples[..]) .full(params, &mono_samples[..])
.expect("failed to run model"); .expect("failed to run model");
// fetch the results // fetch the results

View file

@ -51,7 +51,8 @@ pub fn convert_integer_to_float_audio(
/// ``` /// ```
/// # use whisper_rs::convert_stereo_to_mono_audio; /// # use whisper_rs::convert_stereo_to_mono_audio;
/// let samples = [0.0f32; 1024]; /// let samples = [0.0f32; 1024];
/// let mono = convert_stereo_to_mono_audio(&samples).expect("should be no half samples missing"); /// let mut mono_samples = [0.0f32; 512];
/// convert_stereo_to_mono_audio(&samples, &mut mono_samples).expect("should be no half samples missing");
/// ``` /// ```
pub fn convert_stereo_to_mono_audio(input: &[f32], output: &mut [f32]) -> Result<(), WhisperError> { pub fn convert_stereo_to_mono_audio(input: &[f32], output: &mut [f32]) -> Result<(), WhisperError> {
let (input, []) = input.as_chunks::<2>() else { let (input, []) = input.as_chunks::<2>() else {

View file

@ -801,7 +801,7 @@ impl<'a, 'b> FullParams<'a, 'b> {
/// # Examples /// # Examples
/// ``` /// ```
/// # use whisper_rs::{FullParams, SamplingStrategy}; /// # use whisper_rs::{FullParams, SamplingStrategy};
/// let mut params = FullParams::new(SamplingStrategy::default()); /// let mut params = FullParams::new(SamplingStrategy::Greedy { best_of: 5 });
/// params.set_initial_prompt("Hello, world!"); /// params.set_initial_prompt("Hello, world!");
/// // ... further usage of params ... /// // ... further usage of params ...
/// ``` /// ```