Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: default max_new_tokens to 100

This commit is contained in:
OlivierDehaene 2023-12-13 09:19:19 +01:00
parent 88aae2595d
commit 28821bfd5d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/openapi.json
View File

@ -523,7 +523,7 @@
"max_new_tokens": {
"type": "integer",
"format": "int32",
"default": "null",
"default": "20",
"example": "20",
"nullable": true,
"minimum": 0

4
router/src/lib.rs
View File

@ -107,7 +107,7 @@ pub(crate) struct GenerateParameters {
#[schema(default = "false", example = true)]
pub do_sample: bool,
#[serde(default = "default_max_new_tokens")]
#[schema(nullable = true, default = "null", example = "20")]
#[schema(nullable = true, default = "100", example = "20")]
pub max_new_tokens: Option<u32>,
#[serde(default)]
#[schema(nullable = true, default = "null", example = false)]
@ -141,7 +141,7 @@ pub(crate) struct GenerateParameters {
}
fn default_max_new_tokens() -> Option<u32> {
None
Some(100)
}
fn default_parameters() -> GenerateParameters {