Struct Model

pub struct Model {
    pub lm: LmModel,
    pub sample_rate: f64,
    pub max_duration_s: f64,
    /* private fields */
}

Fields

lm: LmModel

sample_rate: f64

max_duration_s: f64

Implementations

impl Model

pub fn new( cfg: &Config, vb_t5: VarBuilder<'_>, vb_lm: VarBuilder<'_>, vb_speaker_cond: Option<VarBuilder<'_>>, ) -> Result<Self>

impl Model

pub fn conditions( &mut self, token_ids: &Tensor, speaker_pcm: Option<&Tensor>, ) -> Result<Tensor>

pub fn sample( &mut self, conditions: &Tensor, cfg_alpha: f64, ) -> Result<Vec<Vec<u32>>>

pub fn sample_lp( &mut self, conditions: &Tensor, cfg_alpha: f64, lp: LogitsProcessor, ) -> Result<Vec<Vec<u32>>>

Trait Implementations

impl Clone for Model

fn clone(&self) -> Model

Returns a duplicate of the value.

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

Performs copy-assignment from source.