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

feat: large v3 turbo dtw preset

Browse source
This commit is contained in:
newfla 2024-11-23 19:50:38 +01:00
parent 71ab110e04
commit c1c2377456
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)]