imatrix : fix arg parser for imatrix (#9366)

* imatrix : fix arg parser

* beautify printing first arg
This commit is contained in:
Xuan Son Nguyen 2024-09-08 12:12:17 +02:00 committed by GitHub
parent a876861455
commit 00b02bb249
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 2 deletions

View File

@ -600,7 +600,9 @@ std::string llama_arg::to_string() {
if (args.size() == 1) {
ss << arg;
} else {
ss << format("%-7s", arg) << ", ";
// first arg is usually abbreviation, we need padding to make it more beautiful
auto tmp = std::string(arg) + ", ";
ss << format("%-7s", tmp.c_str());
}
} else {
ss << arg << (arg != args.back() ? ", " : "");

View File

@ -577,7 +577,7 @@ int main(int argc, char ** argv) {
params.logits_all = true;
params.verbosity = 1;
auto options = gpt_params_parser_init(params, LLAMA_EXAMPLE_COMMON, print_usage);
auto options = gpt_params_parser_init(params, LLAMA_EXAMPLE_IMATRIX, print_usage);
if (!gpt_params_parse(argc, argv, params, options)) {
return 1;
}