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

Merge pull request #192 from newfla/feat_v3_turbo_preset

Browse source 
feat: large v3 turbo dtw preset
This commit is contained in:
Niko 2024-11-29 06:13:55 +00:00 • committed by GitHub
commit d3fe24a512
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/whisper_ctx.rs
View file

@ -585,6 +585,10 @@ impl<'a> WhisperContextParameters<'a> {
dtw_aheads_preset = dtw_aheads_preset =
whisper_rs_sys::whisper_alignment_heads_preset_WHISPER_AHEADS_LARGE_V3; whisper_rs_sys::whisper_alignment_heads_preset_WHISPER_AHEADS_LARGE_V3;
} }
DtwModelPreset::LargeV3Turbo => {
dtw_aheads_preset =
whisper_rs_sys::whisper_alignment_heads_preset_WHISPER_AHEADS_LARGE_V3_TURBO;
}
}, },
} }
@ -649,6 +653,7 @@ pub enum DtwModelPreset {
LargeV1, LargeV1,
LargeV2, LargeV2,
LargeV3, LargeV3,
LargeV3Turbo,
} }
#[cfg(test)] #[cfg(test)]