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

Generate bindings + rustfmt

Browse source
This commit is contained in:
arizhih 2024-05-16 15:29:38 +02:00
parent f6826200fc
commit dcfcbced18
2 changed files with 1673 additions and 252 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/whisper_ctx.rs
View file

@ -476,22 +476,22 @@ pub struct WhisperContextParameters {
/// (in that case, GPU is always enabled). /// (in that case, GPU is always enabled).
pub use_gpu: bool, pub use_gpu: bool,
/// Enable flash attention, default false /// Enable flash attention, default false
pub flash_attn : bool, pub flash_attn: bool,
/// GPU device id, default 0 /// GPU device id, default 0
pub gpu_device: c_int, pub gpu_device: c_int,
/// [EXPERIMENTAL] Enable Token-level timestamps with DTW, default 0 /// [EXPERIMENTAL] Enable Token-level timestamps with DTW, default 0
pub dtw_token_timestamps: bool, pub dtw_token_timestamps: bool,
/// Preset id for DTW, default whisper_alignment_heads_preset_WHISPER_AHEADS_NONE /// Preset id for DTW, default whisper_alignment_heads_preset_WHISPER_AHEADS_NONE
pub dtw_aheads_preset : whisper_rs_sys::whisper_alignment_heads_preset, pub dtw_aheads_preset: whisper_rs_sys::whisper_alignment_heads_preset,
/// Number of top text layers used from model. Only with whisper_alignment_heads_preset_WHISPER_AHEADS_N_TOP_MOST preset. /// Number of top text layers used from model. Only with whisper_alignment_heads_preset_WHISPER_AHEADS_N_TOP_MOST preset.
pub dtw_n_top : c_int, pub dtw_n_top: c_int,
/// Custom aheads, only with whisper_alignment_heads_preset_WHISPER_AHEADS_CUSTOM preset /// Custom aheads, only with whisper_alignment_heads_preset_WHISPER_AHEADS_CUSTOM preset
/// See details https://github.com/ggerganov/whisper.cpp/pull/1485#discussion_r1519681143 /// See details https://github.com/ggerganov/whisper.cpp/pull/1485#discussion_r1519681143
pub dtw_aheads: whisper_rs_sys::whisper_aheads, pub dtw_aheads: whisper_rs_sys::whisper_aheads,
/// Memory size for DTW /// Memory size for DTW
/// ///
/// **Warning**: Might be removed in next version of whisper.cpp /// **Warning**: Might be removed in next version of whisper.cpp
pub dtw_mem_size : usize pub dtw_mem_size: usize,
} }
#[allow(clippy::derivable_impls)] // this impl cannot be derived #[allow(clippy::derivable_impls)] // this impl cannot be derived
@ -499,13 +499,16 @@ impl Default for WhisperContextParameters {
fn default() -> Self { fn default() -> Self {
Self { Self {
use_gpu: cfg!(feature = "_gpu"), use_gpu: cfg!(feature = "_gpu"),
flash_attn : false, flash_attn: false,
gpu_device: 0, gpu_device: 0,
dtw_token_timestamps: false, dtw_token_timestamps: false,
dtw_aheads_preset : whisper_rs_sys::whisper_alignment_heads_preset_WHISPER_AHEADS_NONE, dtw_aheads_preset: whisper_rs_sys::whisper_alignment_heads_preset_WHISPER_AHEADS_NONE,
dtw_n_top: -1, dtw_n_top: -1,
dtw_aheads : whisper_rs_sys::whisper_aheads { n_heads: 0, heads: std::ptr::null() }, dtw_aheads: whisper_rs_sys::whisper_aheads {
dtw_mem_size : 1024 * 1024 * 128 n_heads: 0,
heads: std::ptr::null(),
},
dtw_mem_size: 1024 * 1024 * 128,
} }
} }
} }
@ -529,7 +532,10 @@ impl WhisperContextParameters {
self.dtw_token_timestamps = dtw_token_timestamps; self.dtw_token_timestamps = dtw_token_timestamps;
self self
} }
pub fn dtw_aheads_preset(&mut self, dtw_aheads_preset: whisper_rs_sys::whisper_alignment_heads_preset) -> &mut Self { pub fn dtw_aheads_preset(
&mut self,
dtw_aheads_preset: whisper_rs_sys::whisper_alignment_heads_preset,
) -> &mut Self {
self.dtw_aheads_preset = dtw_aheads_preset; self.dtw_aheads_preset = dtw_aheads_preset;
self self
} }

1901
sys/src/bindings.rs
View file

File diff suppressed because it is too large Load diff