Struct Mamba2BidiLayersConfig 

pub struct Mamba2BidiLayersConfig {
    pub n_real_layers: usize,
    pub n_virtual_layers: Option<(usize, BidiSchedule)>,
    pub mamba_block: Mamba2Config,
    pub ignore_first_residual: bool,
    pub ignore_last_residual: bool,
    pub outputs_merge: Vec<OutputMergeConfig>,
}

Fields

n_real_layers: usize

n_virtual_layers: Option<(usize, BidiSchedule)>

mamba_block: Mamba2Config

ignore_first_residual: bool

ignore_last_residual: bool

outputs_merge: Vec<OutputMergeConfig>

Shape

• [n_real_layers / 2]

Implementations

impl Mamba2BidiLayersConfig

pub fn new( n_real_layers: usize, mamba_block: Mamba2Config, outputs_merge: Vec<OutputMergeConfig>, ) -> Self

Create a new instance of the config.

Arguments

Required Arguments

n_real_layers

mamba_block

outputs_merge

Shape

• [n_real_layers / 2]

Default Arguments

n_virtual_layers

• Defaults to "None"

ignore_first_residual

• Defaults to false

ignore_last_residual

• Defaults to false

impl Mamba2BidiLayersConfig

pub fn with_n_virtual_layers( self, n_virtual_layers: Option<(usize, BidiSchedule)>, ) -> Self

Sets the value for the field n_virtual_layers.

• Defaults to "None"

pub fn with_ignore_first_residual(self, ignore_first_residual: bool) -> Self

Sets the value for the field ignore_first_residual.

• Defaults to false

pub fn with_ignore_last_residual(self, ignore_last_residual: bool) -> Self

Sets the value for the field ignore_last_residual.

• Defaults to false

impl Mamba2BidiLayersConfig

pub fn init<B: Backend>(&self, device: &B::Device) -> Mamba2BidiLayers<B>

Returns the initialized model.

Trait Implementations

impl Clone for Mamba2BidiLayersConfig

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for Mamba2BidiLayersConfig

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for Mamba2BidiLayersConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Mamba2BidiLayersConfig

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Mamba2BidiLayersConfig

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

Formats the value using the given formatter.

impl Serialize for Mamba2BidiLayersConfig

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.