huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-10 03:44:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt

Browse Source
This commit is contained in:
OlivierDehaene 2023-03-07 13:35:37 +01:00
parent feddbbc998
commit f3586b2308
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
router/src/lib.rs
View File

@ -14,18 +14,18 @@ use validation::Validation;
pub(crate) struct GenerateParameters { pub(crate) struct GenerateParameters {
#[serde(default)] #[serde(default)]
#[schema( #[schema(
exclusive_minimum = 0.0, exclusive_minimum = 0.0,
nullable = true, nullable = true,
default = "null", default = "null",
example = 0.5 example = 0.5
)] )]
pub temperature: Option<f32>, pub temperature: Option<f32>,
#[serde(default)] #[serde(default)]
#[schema( #[schema(
exclusive_minimum = 0.0, exclusive_minimum = 0.0,
nullable = true, nullable = true,
default = "null", default = "null",
example = 1.03 example = 1.03
)] )]
pub repetition_penalty: Option<f32>, pub repetition_penalty: Option<f32>,
#[serde(default)] #[serde(default)]
@ -33,11 +33,11 @@ pub(crate) struct GenerateParameters {
pub top_k: Option<i32>, pub top_k: Option<i32>,
#[serde(default)] #[serde(default)]
#[schema( #[schema(
exclusive_minimum = 0.0, exclusive_minimum = 0.0,
maximum = 1.0, maximum = 1.0,
nullable = true, nullable = true,
default = "null", default = "null",
example = 0.95 example = 0.95
)] )]
pub top_p: Option<f32>, pub top_p: Option<f32>,
#[serde(default)] #[serde(default)]