imatrix : fix arg parser for imatrix (#9366)

* imatrix : fix arg parser * beautify printing first arg
ggerganov · Sep 8, 2024 · 00b02bb · 00b02bb
1 parent a876861
commit 00b02bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/common/common.cpp b/common/common.cpp
@@ -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() ? ", " : "");

diff --git a/examples/imatrix/imatrix.cpp b/examples/imatrix/imatrix.cpp
@@ -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;
     }