File tree Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Expand file tree Collapse file tree 2 files changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -12,6 +12,7 @@ use text_generation_router::validation::{
1212use tokio:: sync:: { mpsc, oneshot} ;
1313use tokio:: time:: Instant ;
1414use tracing:: { info_span, instrument, Span } ;
15+ use std:: collections:: HashMap ;
1516
1617/// Queue entry
1718#[ derive( Debug ) ]
@@ -429,6 +430,7 @@ mod tests {
429430 frequency_penalty : 0.0 ,
430431 watermark : false ,
431432 grammar : None ,
433+ logit_bias : HashMap :: new ( ) ,
432434 } ,
433435 stopping_parameters : ValidStoppingParameters {
434436 ignore_eos_token : false ,
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ use text_generation_router::validation::{
1616use tokio:: sync:: { mpsc, oneshot} ;
1717use tokio:: time:: Instant ;
1818use tracing:: { info_span, instrument, Instrument , Span } ;
19+ use std:: collections:: HashMap ;
1920
2021/// Queue entry
2122#[ derive( Debug ) ]
@@ -577,6 +578,7 @@ mod tests {
577578 frequency_penalty : 0.0 ,
578579 watermark : false ,
579580 grammar : None ,
581+ logit_bias : HashMap :: new ( ) ,
580582 } ,
581583 stopping_parameters : ValidStoppingParameters {
582584 ignore_eos_token : false ,
You can’t perform that action at this time.
0 commit comments