Struct LinformerModel 

pub struct LinformerModel { /* private fields */ }

Linformer model

Implementations

impl LinformerModel

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

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

pub fn device(&self) -> Device

impl LinformerModel

pub fn load_from_path(&mut self, model_path: impl AsRef<Path>) -> Result<()>

Enhanced weight loading from local path with support for multiple formats

pub fn load_from_huggingface(&mut self, model_name: &str) -> Result<()>

Enhanced weight loading from HuggingFace Hub with automatic download

Trait Implementations

impl Model for LinformerModel

type Config = LinformerConfig

type Input = (Vec<u32>, Option<Vec<u32>>, Option<Vec<u32>>)

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.