From 46bf73a37f1aabe6f0b40365b0c7b2ba831905f5 Mon Sep 17 00:00:00 2001 From: saood06 Date: Sun, 23 Feb 2025 00:16:27 -0600 Subject: Add new sweep-bench benchmark (#225) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * examples : add new sweep-bench benchmark * Change documentation to reference ik_llama.cpp * Made it compile with ik_llama * Fix JSONL output --------- Co-authored-by: Stanisław Szymczyk --- common/common.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'common/common.cpp') diff --git a/common/common.cpp b/common/common.cpp index f7a6f76f..6bf6e4f9 100644 --- a/common/common.cpp +++ b/common/common.cpp @@ -1360,6 +1360,15 @@ bool gpt_params_find_arg(int argc, char ** argv, const std::string & arg, gpt_pa params.warmup = false; return true; } + if (arg == "--output-format") { + CHECK_ARG + std::string value(argv[i]); + /**/ if (value == "jsonl") { params.sweep_bench_output_jsonl = true; } + else if (value == "md") { params.sweep_bench_output_jsonl = false; } + else { invalid_param = true; } + return true; + } + #ifndef LOG_DISABLE_LOGS // Parse args for logging parameters if (log_param_single_parse(argv[i])) { -- cgit v1.2.3