Struct S4ForLanguageModeling 

pub struct S4ForLanguageModeling {
    pub s4: S4Model,
    pub lm_head: Linear,
    pub device: Device,
}

S4 Model for Language Modeling

Fields

s4: S4Model

lm_head: Linear

device: Device

Implementations

impl S4ForLanguageModeling

pub fn new_with_device(config: S4Config, device: Device) -> Result<Self>

pub fn new(config: S4Config) -> Result<Self>

pub fn device(&self) -> Device

Trait Implementations

impl Model for S4ForLanguageModeling

type Config = S4Config

type Input = Tensor

type Output = Tensor

fn forward(&self, input: Self::Input) -> Result<Self::Output>

Performs a forward pass through the model.

fn load_pretrained(&mut self, reader: &mut dyn Read) -> Result<()>

Loads pretrained weights into the model.

fn get_config(&self) -> &Self::Config

Returns a reference to the model’s configuration.

fn num_parameters(&self) -> usize

Returns the total number of parameters in the model.