Struct State

pub struct State { /* private fields */ }

Implementations

impl State

pub fn new( model: LmModel, max_step_idx: usize, audio_lp: LogitsProcessor, text_lp: LogitsProcessor, pad_mult: Option<f32>, repetition_penalty: Option<(usize, f32)>, cfg_alpha: Option<f64>, config: Config, ) -> Self

pub fn step_idx(&self) -> usize

pub fn config(&self) -> &Config

pub fn user_rating(&self) -> u32

pub fn set_user_rating(&mut self, grade: u32)

pub fn step_( &mut self, text_token: Option<u32>, input_audio_tokens: &[u32], force_text_token: Option<u32>, ca_src: Option<&CaSrc>, conditions: Option<&Condition>, ) -> Result<u32>

pub fn step_without_ca_src( &mut self, text_token: u32, input_audio_tokens: &[u32], force_text_token: Option<u32>, ) -> Result<u32>

pub fn step( &mut self, text_token: u32, input_audio_tokens: &[u32], force_text_token: Option<u32>, ca_src: Option<&CaSrc>, ) -> Result<u32>

pub fn audio_tokens(&self, include_all: bool) -> &[Vec<u32>]

If include_all is set, all the time steps are returned. Otherwise only the timesteps that have been generated are handled.

pub fn text_tokens(&self, include_all: bool) -> &[u32]

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