Struct StepArg

pub struct StepArg {
    pub backtrack: u32,
    pub tokens: Vec<TokenId>,
    pub sampled: Option<TokenId>,
}

Fields

backtrack: u32

Sampling result for the previous iteration. For simple sampled token ‘t’, backtrack==0 and tokens==[t]. For first request, backtrack==0 and tokens==[] (prompt is passed separately, before). Can be more complex when splices are used.

tokens: Vec<TokenId>

sampled: Option<TokenId>

The token that was sampled (after applying the mask), before any splicing.

Implementations

impl StepArg

pub fn empty() -> Self

pub fn save_tokens(&self, acc_tokens: &mut Vec<TokenId>)

pub fn from_splice(s: &Splice, sampled: Option<TokenId>) -> Self

pub fn from_sampled_token(tok: TokenId) -> Self

Trait Implementations

impl Clone for StepArg

fn clone(&self) -> StepArg

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for StepArg

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for StepArg

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for StepArg

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.