pub struct T5EncoderModel { /* private fields */ }Implementations§
Source§impl T5EncoderModel
impl T5EncoderModel
pub fn load(vb: VarBuilder<'_>, cfg: &Config) -> Result<Self>
pub fn forward(&mut self, input_ids: &Tensor) -> Result<Tensor>
pub fn forward_dt( &mut self, input_ids: &Tensor, dtype: Option<DType>, ) -> Result<Tensor>
pub fn device(&self) -> &Device
pub fn clear_kv_cache(&mut self)
Trait Implementations§
Source§impl Clone for T5EncoderModel
impl Clone for T5EncoderModel
Source§fn clone(&self) -> T5EncoderModel
fn clone(&self) -> T5EncoderModel
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 T5EncoderModel
impl !RefUnwindSafe for T5EncoderModel
impl Send for T5EncoderModel
impl Sync for T5EncoderModel
impl Unpin for T5EncoderModel
impl !UnwindSafe for T5EncoderModel
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