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::{
12
12
use tokio:: sync:: { mpsc, oneshot} ;
13
13
use tokio:: time:: Instant ;
14
14
use tracing:: { info_span, instrument, Span } ;
15
+ use std:: collections:: HashMap ;
15
16
16
17
/// Queue entry
17
18
#[ derive( Debug ) ]
@@ -429,6 +430,7 @@ mod tests {
429
430
frequency_penalty : 0.0 ,
430
431
watermark : false ,
431
432
grammar : None ,
433
+ logit_bias : HashMap :: new ( ) ,
432
434
} ,
433
435
stopping_parameters : ValidStoppingParameters {
434
436
ignore_eos_token : false ,
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ use text_generation_router::validation::{
16
16
use tokio:: sync:: { mpsc, oneshot} ;
17
17
use tokio:: time:: Instant ;
18
18
use tracing:: { info_span, instrument, Instrument , Span } ;
19
+ use std:: collections:: HashMap ;
19
20
20
21
/// Queue entry
21
22
#[ derive( Debug ) ]
@@ -577,6 +578,7 @@ mod tests {
577
578
frequency_penalty : 0.0 ,
578
579
watermark : false ,
579
580
grammar : None ,
581
+ logit_bias : HashMap :: new ( ) ,
580
582
} ,
581
583
stopping_parameters : ValidStoppingParameters {
582
584
ignore_eos_token : false ,
You can’t perform that action at this time.
0 commit comments