pub struct ChineseClipModel { /* private fields */ }Implementations§
Source§impl ChineseClipModel
impl ChineseClipModel
pub fn new(vs: VarBuilder<'_>, c: &ChineseClipConfig) -> Result<Self>
pub fn get_text_features( &self, input_ids: &Tensor, token_type_ids: Option<&Tensor>, attention_mask: Option<&Tensor>, ) -> Result<Tensor>
pub fn get_image_features(&self, pixel_values: &Tensor) -> Result<Tensor>
pub fn forward( &self, pixel_values: &Tensor, input_ids: &Tensor, token_type_ids: Option<&Tensor>, attention_mask: Option<&Tensor>, ) -> Result<(Tensor, Tensor)>
Trait Implementations§
Source§impl Clone for ChineseClipModel
impl Clone for ChineseClipModel
Source§fn clone(&self) -> ChineseClipModel
fn clone(&self) -> ChineseClipModel
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for ChineseClipModel
impl !RefUnwindSafe for ChineseClipModel
impl Send for ChineseClipModel
impl Sync for ChineseClipModel
impl Unpin for ChineseClipModel
impl !UnwindSafe for ChineseClipModel
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
§impl<T> Instrument for T
impl<T> Instrument for T
§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more