huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 23:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set alias for max_completion_tokens in ChatRequest (#2932)

Browse Source
This commit is contained in:
Alvaro Bartolome 2025-01-23 14:18:47 +01:00 committed by GitHub
parent cc212154e0
commit 6ab02931cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
router/src/lib.rs
View File

@ -460,7 +460,7 @@ pub struct CompletionRequest {
pub prompt: Prompt, pub prompt: Prompt,
/// The maximum number of tokens that can be generated in the chat completion. /// The maximum number of tokens that can be generated in the chat completion.
#[serde(default, alias = "max_completion_tokens")] #[serde(default)]
#[schema(default = "1024", example = "32")] #[schema(default = "1024", example = "32")]
pub max_tokens: Option<u32>, pub max_tokens: Option<u32>,
@ -840,7 +840,7 @@ pub(crate) struct ChatRequest {
pub top_logprobs: Option<u32>, pub top_logprobs: Option<u32>,
/// The maximum number of tokens that can be generated in the chat completion. /// The maximum number of tokens that can be generated in the chat completion.
#[serde(default)] #[serde(default, alias = "max_completion_tokens")]
#[schema(default = "1024", example = "32")] #[schema(default = "1024", example = "32")]
pub max_tokens: Option<u32>, pub max_tokens: Option<u32>,