Struct WhisperConfig 

pub struct WhisperConfig {
    pub num_mel_bins: usize,
    pub max_source_positions: usize,
    pub d_model: usize,
    pub encoder_attention_heads: usize,
    pub encoder_layers: usize,
    pub vocab_size: usize,
    pub max_target_positions: usize,
    pub decoder_attention_heads: usize,
    pub decoder_layers: usize,
    pub suppress_tokens: Vec<u32>,
    pub begin_suppress_tokens: Vec<u32>,
}

Whisper model dimensions (HF field names; see OpenAI model.py comments in candle).

Fields

num_mel_bins: usize

max_source_positions: usize

d_model: usize

encoder_attention_heads: usize

encoder_layers: usize

vocab_size: usize

max_target_positions: usize

decoder_attention_heads: usize

decoder_layers: usize

suppress_tokens: Vec<u32>

begin_suppress_tokens: Vec<u32>

Suppressed on the first generated token only (HF begin_suppress_tokens).

Implementations

impl WhisperConfig

pub fn from_file(path: &Path) -> Result<WhisperConfig, Error>

pub fn head_dim(&self) -> usize

pub fn decoder_head_dim(&self) -> usize

pub fn encoder_seq_len(&self, mel_frames: usize) -> usize

Encoder sequence length after the two conv layers (stride-2 on the second).

pub fn tiny_synthetic() -> WhisperConfig

Tiny config for compile tests (no real weights required).

pub fn tiny() -> WhisperConfig

openai/whisper-tiny dimensions.

Trait Implementations

impl Clone for WhisperConfig

fn clone(&self) -> WhisperConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for WhisperConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for WhisperConfig

fn deserialize<__D>( __deserializer: __D, ) -> Result<WhisperConfig, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.