فهرست منبع

imatrix : fix arg parser for imatrix (#9366)

* imatrix : fix arg parser

* beautify printing first arg
Xuan Son Nguyen 1 سال پیش
والد
کامیت
00b02bb249
2فایلهای تغییر یافته به همراه4 افزوده شده و 2 حذف شده
  1. 3 1
      common/common.cpp
  2. 1 1
      examples/imatrix/imatrix.cpp

+ 3 - 1
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() ? ", " : "");

+ 1 - 1
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;
     }