pub struct MirostatV2Sampler { /* private fields */ }Expand description
Adaptive Mirostat V2 policy with backend-neutral state.
Implementations§
Source§impl MirostatV2Sampler
impl MirostatV2Sampler
Sourcepub fn new(tau: f32, eta: f32) -> Result<Self, SamplingConfigurationError>
pub fn new(tau: f32, eta: f32) -> Result<Self, SamplingConfigurationError>
Creates a sampler targeting tau bits of surprise.
Sourcepub fn penalties(
self,
repeat_penalty: f32,
repeat_last_n: i32,
frequency_penalty: f32,
presence_penalty: f32,
) -> Self
pub fn penalties( self, repeat_penalty: f32, repeat_last_n: i32, frequency_penalty: f32, presence_penalty: f32, ) -> Self
Sets penalties applied before adaptive truncation.
Sourcepub fn generated_tokens(&self) -> &[u32]
pub fn generated_tokens(&self) -> &[u32]
Accepted-token history.
Sourcepub fn accept_token(
&mut self,
token: u32,
probability: f32,
) -> Result<(), SamplingConfigurationError>
pub fn accept_token( &mut self, token: u32, probability: f32, ) -> Result<(), SamplingConfigurationError>
Records an externally accepted token and its normalized probability.
Trait Implementations§
Source§impl Clone for MirostatV2Sampler
impl Clone for MirostatV2Sampler
Source§fn clone(&self) -> MirostatV2Sampler
fn clone(&self) -> MirostatV2Sampler
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for MirostatV2Sampler
impl Debug for MirostatV2Sampler
Source§impl Default for MirostatV2Sampler
impl Default for MirostatV2Sampler
Source§impl<B: SamplingBackend> Sampler<B> for MirostatV2Sampler
impl<B: SamplingBackend> Sampler<B> for MirostatV2Sampler
Source§fn sample(
&mut self,
logits: &B::Logits,
temperature: f32,
random: Option<&mut B::RandomState>,
context: &B::Context,
) -> Result<B::Token, B::Error>
fn sample( &mut self, logits: &B::Logits, temperature: f32, random: Option<&mut B::RandomState>, context: &B::Context, ) -> Result<B::Token, B::Error>
Selects one token from raw model logits.
Source§fn uses_checkpoint_defaults(&self) -> bool
fn uses_checkpoint_defaults(&self) -> bool
Whether loaded checkpoint defaults should wrap this policy.
Source§impl<B: SamplingBackend> SpeculativeSampler<B> for MirostatV2Sampler
impl<B: SamplingBackend> SpeculativeSampler<B> for MirostatV2Sampler
Source§fn process_logits(
&mut self,
logits: &B::Logits,
temperature: f32,
history: &[u32],
context: &B::Context,
) -> Result<B::Logits, B::Error>
fn process_logits( &mut self, logits: &B::Logits, temperature: f32, history: &[u32], context: &B::Context, ) -> Result<B::Logits, B::Error>
Applies penalties, filters, and temperature.
Source§fn commit_token(
&mut self,
processed_logits: &B::Logits,
token: u32,
context: &B::Context,
) -> Result<(), B::Error>
fn commit_token( &mut self, processed_logits: &B::Logits, token: u32, context: &B::Context, ) -> Result<(), B::Error>
Commits a token selected from a processed target distribution.
Source§fn uses_checkpoint_defaults(&self) -> bool
fn uses_checkpoint_defaults(&self) -> bool
Whether loaded checkpoint defaults should wrap this policy.
Source§fn supports_exact_optimistic_promotion(&self) -> bool
fn supports_exact_optimistic_promotion(&self) -> bool
Whether optimistic draft work is an exact discardable fork.
Source§fn grammar_is_complete(&mut self) -> Result<bool, B::Error>
fn grammar_is_complete(&mut self) -> Result<bool, B::Error>
Whether the committed generation grammar is complete.
Auto Trait Implementations§
impl Freeze for MirostatV2Sampler
impl RefUnwindSafe for MirostatV2Sampler
impl Send for MirostatV2Sampler
impl Sync for MirostatV2Sampler
impl Unpin for MirostatV2Sampler
impl UnsafeUnpin for MirostatV2Sampler
impl UnwindSafe for MirostatV2Sampler
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more