From 1a98836b94bfbe27de758ca6aea7c0d7e14fba7e Mon Sep 17 00:00:00 2001 From: orangels Date: Sat, 9 May 2026 18:30:40 +0800 Subject: [PATCH] fix: set bench max tokens default to 200k --- src/cli.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/cli.rs b/src/cli.rs index 1bc75f4..c20802a 100644 --- a/src/cli.rs +++ b/src/cli.rs @@ -138,7 +138,7 @@ pub enum BenchCommand { limit: Option, #[arg(long, num_args = 0..=1, default_missing_value = "true")] stream: Option, - #[arg(long, default_value_t = 20_000)] + #[arg(long, default_value_t = 200_000)] max_tokens: u32, #[arg(long)] debug_raw: bool, @@ -170,7 +170,7 @@ pub enum BenchCommand { limit: Option, #[arg(long, num_args = 0..=1, default_missing_value = "true")] stream: Option, - #[arg(long, default_value_t = 20_000)] + #[arg(long, default_value_t = 200_000)] max_tokens: u32, #[arg(long)] debug_raw: bool, @@ -1576,7 +1576,7 @@ mod tests { } #[test] - fn bench_command_defaults_max_tokens_to_20000() { + fn bench_command_defaults_max_tokens_to_200000() { 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, 20_000); + assert_eq!(max_tokens, 200_000); } #[test]