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 tests.

This commit is contained in:
Nicolas Patry 2024-02-08 15:25:34 +00:00
parent 40f693b6b9
commit 81fa53f37b
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
router/src/validation.rs
View File

@ -107,7 +107,7 @@ impl Validation {
) -> Result<(String, usize, u32), ValidationError> { ) -> Result<(String, usize, u32), ValidationError> {
// If we have a fast tokenizer // If we have a fast tokenizer
if let Some((encoding, inputs)) = self.tokenize(inputs.clone(), truncate).await? { if let Some((encoding, inputs)) = self.tokenize(inputs.clone(), truncate).await? {
if self.batch_dimension { if !self.batch_dimension {
let input_length = encoding.len(); let input_length = encoding.len();
// Get total tokens // Get total tokens
@ -465,6 +465,7 @@ mod tests {
let max_input_length = 5; let max_input_length = 5;
let max_total_tokens = 6; let max_total_tokens = 6;
let workers = 1; let workers = 1;
let batch_dimension = false;
let validation = Validation::new( let validation = Validation::new(
workers, workers,
tokenizer, tokenizer,
@ -473,6 +474,7 @@ mod tests {
max_top_n_tokens, max_top_n_tokens,
max_input_length, max_input_length,
max_total_tokens, max_total_tokens,
batch_dimension
); );
let max_new_tokens = 10; let max_new_tokens = 10;
@ -494,6 +496,7 @@ mod tests {
let max_input_length = 5; let max_input_length = 5;
let max_total_tokens = 6; let max_total_tokens = 6;
let workers = 1; let workers = 1;
let batch_dimension = false;
let validation = Validation::new( let validation = Validation::new(
workers, workers,
tokenizer, tokenizer,
@ -502,6 +505,7 @@ mod tests {
max_top_n_tokens, max_top_n_tokens,
max_input_length, max_input_length,
max_total_tokens, max_total_tokens,
batch_dimension,
); );
let max_new_tokens = 10; let max_new_tokens = 10;
@ -523,6 +527,7 @@ mod tests {
let max_input_length = 5; let max_input_length = 5;
let max_total_tokens = 6; let max_total_tokens = 6;
let workers = 1; let workers = 1;
let batch_dimension = false;
let validation = Validation::new( let validation = Validation::new(
workers, workers,
tokenizer, tokenizer,
@ -531,6 +536,7 @@ mod tests {
max_top_n_tokens, max_top_n_tokens,
max_input_length, max_input_length,
max_total_tokens, max_total_tokens,
batch_dimension
); );
match validation match validation
.validate(GenerateRequest { .validate(GenerateRequest {
@ -557,6 +563,7 @@ mod tests {
let max_input_length = 5; let max_input_length = 5;
let max_total_tokens = 106; let max_total_tokens = 106;
let workers = 1; let workers = 1;
let batch_dimension = false;
let validation = Validation::new( let validation = Validation::new(
workers, workers,
tokenizer, tokenizer,
@ -565,6 +572,7 @@ mod tests {
max_top_n_tokens, max_top_n_tokens,
max_input_length, max_input_length,
max_total_tokens, max_total_tokens,
batch_dimension
); );
match validation match validation
.validate(GenerateRequest { .validate(GenerateRequest {
@ -620,6 +628,7 @@ mod tests {
let max_input_length = 5; let max_input_length = 5;
let max_total_tokens = 106; let max_total_tokens = 106;
let workers = 1; let workers = 1;
let batch_dimension = false;
let validation = Validation::new( let validation = Validation::new(
workers, workers,
tokenizer, tokenizer,
@ -628,6 +637,7 @@ mod tests {
max_top_n_tokens, max_top_n_tokens,
max_input_length, max_input_length,
max_total_tokens, max_total_tokens,
batch_dimension,
); );
match validation match validation
.validate(GenerateRequest { .validate(GenerateRequest {