Struct StableLMModel 

pub struct StableLMModel {
    pub config: StableLMConfig,
    pub embeddings: StableLMEmbeddings,
    pub layers: Vec<StableLMDecoderLayer>,
    pub norm: RMSNorm,
    /* private fields */
}

StableLM Base Model

Fields

config: StableLMConfig

embeddings: StableLMEmbeddings

layers: Vec<StableLMDecoderLayer>

norm: RMSNorm

Implementations

impl StableLMModel

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

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

pub fn device(&self) -> &Device

pub fn forward_with_outputs( &self, input_ids: &Tensor, ) -> Result<StableLMOutputs>

Trait Implementations

impl Model for StableLMModel

type Config = StableLMConfig

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.