Struct VoxtralEncoder 

pub struct VoxtralEncoder { /* private fields */ }

Implementations

impl VoxtralEncoder

pub fn new(cfg: &VoxtralEncoderConfig, vb: VarBuilder<'_>) -> Result<Self>

pub fn forward(&self, input_features: &Tensor) -> Result<Tensor>

pub fn forward_with_training( &self, input_features: &Tensor, training: bool, ) -> Result<Tensor>

pub fn get_intermediate_size(&self) -> usize

Get the output dimension of the first FC layer (needed for projector)

pub fn process_long_audio( &self, input_features: &Tensor, chunk_size: usize, overlap: usize, ) -> Result<Tensor>

Process long audio sequences in chunks to save memory

Trait Implementations

impl Clone for VoxtralEncoder

fn clone(&self) -> VoxtralEncoder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for VoxtralEncoder

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

Formats the value using the given formatter.