Add batch warmup to sweep-bench (#375)

Co-authored-by: Iwan Kawrakow <iwan.kawrakow@gmail.com>
This commit is contained in:
Kawrakow
2025-05-12 07:50:26 +03:00
committed by GitHub
parent 2e585d4508
commit ceb8f513e4
3 changed files with 22 additions and 1 deletions

View File

@@ -1468,6 +1468,10 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa
params.warmup = false;
return true;
}
if (arg == "--warmup-batch" || arg == "-wb") {
params.batch_warmup = true;
return true;
}
if (arg == "--output-format") {
CHECK_ARG
std::string value(argv[i]);