pub struct SamplingParams {Show 19 fields
pub temperature: Option<f32>,
pub max_new_tokens: Option<u32>,
pub top_p: Option<f32>,
pub top_k: Option<i32>,
pub frequency_penalty: Option<f32>,
pub presence_penalty: Option<f32>,
pub repetition_penalty: Option<f32>,
pub stop: Option<StringOrArray>,
pub ignore_eos: Option<bool>,
pub skip_special_tokens: Option<bool>,
pub json_schema: Option<String>,
pub regex: Option<String>,
pub ebnf: Option<String>,
pub min_p: Option<f32>,
pub min_new_tokens: Option<u32>,
pub stop_token_ids: Option<Vec<u32>>,
pub no_stop_trim: Option<bool>,
pub n: Option<u32>,
pub sampling_seed: Option<u64>,
}Expand description
Sampling parameters for text generation
Fields§
§temperature: Option<f32>Temperature for sampling (must be >= 0.0, no upper limit)
max_new_tokens: Option<u32>Maximum number of new tokens to generate (must be >= 0)
top_p: Option<f32>Top-p nucleus sampling (0.0 < top_p <= 1.0)
top_k: Option<i32>Top-k sampling (-1 to disable, or >= 1)
frequency_penalty: Option<f32>§presence_penalty: Option<f32>§repetition_penalty: Option<f32>§stop: Option<StringOrArray>§ignore_eos: Option<bool>§skip_special_tokens: Option<bool>§json_schema: Option<String>§regex: Option<String>§ebnf: Option<String>§min_p: Option<f32>§min_new_tokens: Option<u32>Minimum number of new tokens (validated in schema function for cross-field check with max_new_tokens)
stop_token_ids: Option<Vec<u32>>§no_stop_trim: Option<bool>§n: Option<u32>§sampling_seed: Option<u64>Trait Implementations§
Source§impl Clone for SamplingParams
impl Clone for SamplingParams
Source§fn clone(&self) -> SamplingParams
fn clone(&self) -> SamplingParams
Returns a duplicate of the value. Read more
1.0.0 · 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 SamplingParams
impl Debug for SamplingParams
Source§impl Default for SamplingParams
impl Default for SamplingParams
Source§fn default() -> SamplingParams
fn default() -> SamplingParams
Returns the “default value” for a type. Read more
Source§impl<'de> Deserialize<'de> for SamplingParams
impl<'de> Deserialize<'de> for SamplingParams
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Source§impl Serialize for SamplingParams
impl Serialize for SamplingParams
Source§impl Validate for SamplingParams
impl Validate for SamplingParams
Source§impl<'v_a> ValidateArgs<'v_a> for SamplingParams
impl<'v_a> ValidateArgs<'v_a> for SamplingParams
Auto Trait Implementations§
impl Freeze for SamplingParams
impl RefUnwindSafe for SamplingParams
impl Send for SamplingParams
impl Sync for SamplingParams
impl Unpin for SamplingParams
impl UnwindSafe for SamplingParams
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