Merge pull request #192 from newfla/feat_v3_turbo_preset

feat: large v3 turbo dtw preset
tazz4843 · Nov 29, 2024 · d3fe24a · d3fe24a
2 parents 71ab110 + c1c2377
commit d3fe24a
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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)]