Skip to content

Commit d57c94a

Browse files
committed
fix: include logit_bias in all ValidGenerateRequest's
1 parent aab9533 commit d57c94a

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

backends/v2/src/queue.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ use text_generation_router::validation::{
1212
use tokio::sync::{mpsc, oneshot};
1313
use tokio::time::Instant;
1414
use 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,

backends/v3/src/queue.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ use text_generation_router::validation::{
1616
use tokio::sync::{mpsc, oneshot};
1717
use tokio::time::Instant;
1818
use 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,

0 commit comments

Comments
 (0)