Struct CommandRModel 

pub struct CommandRModel { /* private fields */ }

Command R Model

Implementations

impl CommandRModel

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

pub fn forward( &mut self, input_ids: &Tensor, attention_mask: Option<&Tensor>, position_ids: Option<&Tensor>, past_key_values: Option<&[(Tensor, Tensor)]>, ) -> Result<CommandRModelOutput>

Trait Implementations

impl Clone for CommandRModel

fn clone(&self) -> CommandRModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for CommandRModel

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

Formats the value using the given formatter.

impl Model for CommandRModel

type Config = CommandRConfig

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.