Struct DepFormer

pub struct DepFormer { /* private fields */ }

Implementations

impl DepFormer

pub fn new( text_vocab_size: usize, audio_vocab_size: usize, main_transformer_dim: usize, cfg: &DepFormerConfig, vb: MaybeQuantizedVarBuilder<'_>, ) -> Result<Self>

pub fn sample( &mut self, xs: &Tensor, text_token: Option<u32>, forced_audio_tokens: &[Option<u32>], lp: &mut LogitsProcessor, ) -> Result<Vec<u32>>

Run a transformer sampling step, getting a token id per codebook.

• xs is the previous layer hidden state.

pub fn sample_cfg( &mut self, xs: &Tensor, cfg_alpha: f64, text_token: Option<u32>, forced_audio_tokens: &[Option<u32>], lp: &mut LogitsProcessor, ) -> Result<Vec<u32>>

Trait Implementations

impl Clone for DepFormer

fn clone(&self) -> DepFormer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DepFormer

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

Formats the value using the given formatter.