Struct MirostatSampler 

pub struct MirostatSampler {
    pub params: MirostatParams,
    pub state: MirostatState,
}

Fields

params: MirostatParams

state: MirostatState

Implementations

impl MirostatSampler

pub fn new(params: MirostatParams) -> Self

Trait Implementations

impl Clone for MirostatSampler

fn clone(&self) -> MirostatSampler

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MirostatSampler

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

Formats the value using the given formatter.

impl Default for MirostatSampler

fn default() -> MirostatSampler

Returns the “default value” for a type.

impl Sampler for MirostatSampler

fn init(&mut self, _model_tokens: &[u32])

Initialize the sampler state.

fn transform(&self, _output: &mut [f32])

Update the raw model output.

fn sample(&mut self, probs: &[f32]) -> u32

Select one token from the distribution, and also update the state.