From c1c2377456408eb066c07b23d6b411b969ebe28e Mon Sep 17 00:00:00 2001 From: newfla Date: Sat, 23 Nov 2024 19:50:38 +0100 Subject: [PATCH] feat: large v3 turbo dtw preset --- src/whisper_ctx.rs | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/whisper_ctx.rs b/src/whisper_ctx.rs index 1c42a6c..0e7be5c 100644 --- a/src/whisper_ctx.rs +++ b/src/whisper_ctx.rs @@ -585,6 +585,10 @@ impl<'a> WhisperContextParameters<'a> { dtw_aheads_preset = 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, LargeV2, LargeV3, + LargeV3Turbo, } #[cfg(test)]