Struct LmModel 

pub struct LmModel { /* private fields */ }

Moshi temporal LM (Helium) + optional DepFormer depth decoder.

Implementations

impl LmModel

pub fn open( cfg: LmConfig, weights: HashMap<String, (Vec<f32>, Vec<usize>)>, ) -> Result<LmModel, Error>

pub fn config(&self) -> &LmConfig

pub fn reset_state(&mut self)

pub fn text_start_token(&self) -> u32

pub fn audio_pad_token(&self) -> u32

pub fn forward_step( &mut self, text_token: Option<u32>, audio_tokens: &[Option<u32>], ) -> Result<(ArrayBase<OwnedRepr<f32>, Dim<[usize; 1]>>, ArrayBase<OwnedRepr<f32>, Dim<[usize; 1]>>), Error>

Single streaming step: sum embeddings → temporal transformer → text logits + hidden.

pub fn depformer_sample( &mut self, hidden: &ArrayBase<OwnedRepr<f32>, Dim<[usize; 1]>>, text_token: Option<u32>, forced: &[Option<u32>], lp: &mut LogitsProcessor, ) -> Result<Option<Vec<u32>>, Error>