From 337aea11390221bc925e4acb1f603f1649af2735 Mon Sep 17 00:00:00 2001 From: Vladimir Zorin Date: Sun, 28 May 2023 20:14:24 +0300 Subject: examples : add --alias option to gpt_params to set use friendly model name (#1614) --- examples/common.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'examples/common.cpp') diff --git a/examples/common.cpp b/examples/common.cpp index 1308f841..478dbafc 100644 --- a/examples/common.cpp +++ b/examples/common.cpp @@ -251,6 +251,12 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) { break; } params.model = argv[i]; + } else if (arg == "-a" || arg == "--alias") { + if (++i >= argc) { + invalid_param = true; + break; + } + params.model_alias = argv[i]; } else if (arg == "--lora") { if (++i >= argc) { invalid_param = true; -- cgit v1.2.3