huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adapt unit tests to commit 28821bf

Browse Source 
Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
This commit is contained in:
Raphael Glon 2023-12-14 16:42:52 +01:00
parent 492c95dcbf
commit 29f87920a8
No known key found for this signature in database
GPG Key ID: 4D4CC6881E12A0C3
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
router/src/validation.rs
View File

@ -556,6 +556,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_p: Some(1.0), top_p: Some(1.0),
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -570,6 +571,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_p: Some(0.99), top_p: Some(0.99),
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -584,6 +586,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_p: None, top_p: None,
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -616,6 +619,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_n_tokens: Some(5), top_n_tokens: Some(5),
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -630,6 +634,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_n_tokens: Some(4), top_n_tokens: Some(4),
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -641,6 +646,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_n_tokens: Some(0), top_n_tokens: Some(0),
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })
@ -652,6 +658,7 @@ mod tests {
inputs: "Hello".to_string(), inputs: "Hello".to_string(),
parameters: GenerateParameters { parameters: GenerateParameters {
top_n_tokens: None, top_n_tokens: None,
max_new_tokens: Some(5),
..default_parameters() ..default_parameters()
}, },
}) })