Struct GenerateState 

pub struct GenerateState { /* private fields */ }

Multistream autoregressive state (ported from kyutai lm_generate_multistream).

Implementations

impl GenerateState

pub fn new( max_steps: usize, text_lp: LogitsProcessor, audio_lp: LogitsProcessor, cfg: GenerateConfig, ) -> GenerateState

pub fn config(&self) -> &GenerateConfig

pub fn step_idx(&self) -> usize

pub fn text_tokens(&self) -> &[u32]

pub fn step( &mut self, lm: &mut LmModel, text_token: u32, input_audio: &[u32], ) -> Result<u32, Error>

Advance one 12.5 Hz frame. input_audio is user codebooks (empty for one-way).

pub fn last_audio_frame(&self) -> Option<Vec<u32>>

Moshi output codebooks ready for Mimi decode (past acoustic delay).

pub fn reset(&mut self, lm: &mut LmModel)