Struct ClipModel

pub struct ClipModel { /* private fields */ }

Implementations

impl ClipModel

pub fn new(vs: VarBuilder<'_>, c: &ClipConfig) -> Result<Self>

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

pub fn get_image_features(&self, pixel_values: &Tensor) -> Result<Tensor>

pub fn forward( &self, pixel_values: &Tensor, input_ids: &Tensor, ) -> Result<(Tensor, Tensor)>

Trait Implementations

impl Clone for ClipModel

fn clone(&self) -> ClipModel

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for ClipModel

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

Formats the value using the given formatter.