diff options
| author | Vladimir Zorin <vladimir@deviant.guru> | 2023-05-28 20:14:24 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-05-28 20:14:24 +0300 |
| commit | 337aea11390221bc925e4acb1f603f1649af2735 (patch) | |
| tree | 5ced14826ee129c30c6c4a1509dcfc10acbbc5b8 /examples/common.cpp | |
| parent | bb051d9723d628414b9e929e5264e23262a2f1b2 (diff) | |
examples : add --alias option to gpt_params to set use friendly model name (#1614)
Diffstat (limited to 'examples/common.cpp')
| -rw-r--r-- | examples/common.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
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; |