Struct Mamba2Block 

pub struct Mamba2Block { /* private fields */ }

Implementations

impl Mamba2Block

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

pub fn forward(&self, xs: &Tensor, state: &mut State) -> Result<Tensor>

pub fn forward_prefill( &self, xs: &Tensor, state: &mut State, chunk_size: usize, ) -> Result<Tensor>

Trait Implementations

impl Clone for Mamba2Block

fn clone(&self) -> Mamba2Block

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Mamba2Block

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

Formats the value using the given formatter.