Просмотр исходного кода

fix: restore bench max tokens default

pull/1/head
orangels 1 месяц назад
Родитель
Сommit
986dc4e088
1 измененных файлов: 4 добавлений и 4 удалений
  1. +4
    -4
      src/cli.rs

+ 4
- 4
src/cli.rs Просмотреть файл

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


#[test] #[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"]) let cli = Cli::try_parse_from(["lq_token_test", "bench", "gpqa-diamond"])
.expect("parse gpqa bench"); .expect("parse gpqa bench");


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


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


#[test] #[test]


Загрузка…
Отмена
Сохранить