Enum LogitsReturnPolicy 

pub enum LogitsReturnPolicy {
    FullLogits,
    GreedyArgmax {
        token_mask: Option<TokenSelectionMask>,
        repetition_penalty: Option<GreedyRepetitionPenalty>,
    },
}

Variants

FullLogits

GreedyArgmax

Fields

token_mask: Option<TokenSelectionMask>

repetition_penalty: Option<GreedyRepetitionPenalty>

Implementations

impl LogitsReturnPolicy

pub fn requires_full_logits(&self) -> bool

Trait Implementations

impl Clone for LogitsReturnPolicy

fn clone(&self) -> LogitsReturnPolicy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for LogitsReturnPolicy

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

Formats the value using the given formatter.

impl Default for LogitsReturnPolicy

fn default() -> Self

Returns the “default value” for a type.