Struct Model 

pub struct Model {
    pub multi_modal_projector: MultiModalProjector,
    pub language_model: Model,
    pub vision_tower: Model,
    pub patch_size: usize,
    pub dtype: DType,
    pub pos: usize,
}

Fields

multi_modal_projector: MultiModalProjector

language_model: Model

vision_tower: Model

patch_size: usize

dtype: DType

pos: usize

Implementations

impl Model

pub fn new(cfg: &Config, vb: VarBuilder<'_>) -> Result<Self>

pub fn clear_kv_cache(&mut self)

pub fn encode_image(&self, image: &Tensor) -> Result<Tensor>

pub fn lm_forward(&mut self, input_ids: &Tensor) -> Result<Tensor>

pub fn lm_forward_embeds(&mut self, xs: &Tensor) -> Result<Tensor>

Trait Implementations

impl Clone for Model

fn clone(&self) -> Model

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Model

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

Formats the value using the given formatter.