Bläddra i källkod

fix: restore bench max tokens default

pull/1/head
orangels 1 månad sedan
förälder
incheckning
986dc4e088
1 ändrade filer med 4 tillägg och 4 borttagningar
  1. +4
    -4
      src/cli.rs

+ 4
- 4
src/cli.rs Visa fil

@@ -138,7 +138,7 @@ pub enum BenchCommand {
limit: Option<usize>,
#[arg(long, num_args = 0..=1, default_missing_value = "true")]
stream: Option<bool>,
#[arg(long, default_value_t = 200_000)]
#[arg(long, default_value_t = 32_768)]
max_tokens: u32,
#[arg(long)]
debug_raw: bool,
@@ -170,7 +170,7 @@ pub enum BenchCommand {
limit: Option<usize>,
#[arg(long, num_args = 0..=1, default_missing_value = "true")]
stream: Option<bool>,
#[arg(long, default_value_t = 200_000)]
#[arg(long, default_value_t = 32_768)]
max_tokens: u32,
#[arg(long)]
debug_raw: bool,
@@ -1576,7 +1576,7 @@ mod tests {
}

#[test]
fn bench_command_defaults_max_tokens_to_200000() {
fn bench_command_defaults_max_tokens_to_32768() {
let cli = Cli::try_parse_from(["lq_token_test", "bench", "gpqa-diamond"])
.expect("parse gpqa bench");

@@ -1587,7 +1587,7 @@ mod tests {
panic!("expected gpqa-diamond bench command");
};

assert_eq!(max_tokens, 200_000);
assert_eq!(max_tokens, 32_768);
}

#[test]


Laddar…
Avbryt
Spara