Struct Mimi

pub struct Mimi { /* private fields */ }

Implementations

impl Mimi

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

pub fn batched( batch_size: usize, cfg: Config, vb: VarBuilder<'_>, ) -> Result<Self>

pub fn config(&self) -> &Config

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

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

pub fn encode_step( &mut self, xs: &StreamTensor, m: &StreamMask, ) -> Result<StreamTensor>

pub fn decode(&mut self, codes: &Tensor) -> Result<Tensor>

pub fn decode_step( &mut self, codes: &StreamTensor, m: &StreamMask, ) -> Result<StreamTensor>

pub fn reset_state(&mut self)

pub fn reset_batch_idx( &mut self, batch_idx: usize, batch_size: usize, ) -> Result<()>

Trait Implementations

impl Clone for Mimi

fn clone(&self) -> Mimi

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Mimi

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

Formats the value using the given formatter.