Struct TrOCRModel 

pub struct TrOCRModel { /* private fields */ }

Implementations

impl TrOCRModel

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

pub fn encoder(&mut self) -> &mut TrOCREncoder

pub fn decoder(&mut self) -> &mut TrOCRForCausalLM

pub fn decode( &mut self, xs: &Tensor, encoder_xs: &Tensor, past_kv_len: usize, ) -> Result<Tensor>

pub fn reset_kv_cache(&mut self)

Trait Implementations

impl Clone for TrOCRModel

fn clone(&self) -> TrOCRModel

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TrOCRModel

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

Formats the value using the given formatter.