Struct WhisperModelWrapper 

pub struct WhisperModelWrapper { /* private fields */ }

Implementations

impl WhisperModelWrapper

pub fn new( vb: VarBuilder<'_>, config: Config, mel_filters: Vec<f32>, device: CandleDevice, dtype: DType, ) -> Result<Self>

Load from VarBuilder + config.

pub fn from_model_dir( model_dir: &Path, device: CandleDevice, dtype: DType, ) -> Result<Self>

Load from model directory.

pub fn pcm_to_mel_tensor(&self, pcm: &[f32]) -> Result<Tensor>

PCM → mel spectrogram tensor (matching Python whisper exactly). Pads or truncates PCM to exactly 30 seconds (N_SAMPLES = 480000).

pub fn encode(&self, mel: &Tensor) -> Result<Tensor>

Encode a mel segment → encoder hidden states.

pub fn decode_step( &self, tokens: &[u32], encoder_out: &Tensor, ) -> Result<Vec<f32>>

Run one decode pass: tokens → logits (includes KV cache, final_linear). On first call pass full initial_tokens; on subsequent calls pass only the new token.

pub fn reset_decoder(&self)

Reset decoder KV cache (call between segments).

pub fn config(&self) -> &Config

pub fn device(&self) -> &CandleDevice