Merge branch 'ysimonson-update-whisper'
This commit is contained in:
commit
298e0ddabd
3 changed files with 6 additions and 4 deletions
|
|
@ -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
|
||||
// these are just provided for convenience
|
||||
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)
|
||||
.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");
|
||||
|
||||
// now we can run the model
|
||||
state
|
||||
.full(params, &samples[..])
|
||||
.full(params, &mono_samples[..])
|
||||
.expect("failed to run model");
|
||||
|
||||
// fetch the results
|
||||
|
|
|
|||
|
|
@ -51,7 +51,8 @@ pub fn convert_integer_to_float_audio(
|
|||
/// ```
|
||||
/// # use whisper_rs::convert_stereo_to_mono_audio;
|
||||
/// 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> {
|
||||
let (input, []) = input.as_chunks::<2>() else {
|
||||
|
|
|
|||
|
|
@ -801,7 +801,7 @@ impl<'a, 'b> FullParams<'a, 'b> {
|
|||
/// # Examples
|
||||
/// ```
|
||||
/// # 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!");
|
||||
/// // ... further usage of params ...
|
||||
/// ```
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue