examples : add --alias option to gpt_params to set use friendly model name (#1614)

This commit is contained in:
Vladimir Zorin 2023-05-28 20:14:24 +03:00 committed by GitHub
parent bb051d9723
commit 337aea1139
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 21 additions and 2 deletions

View file

@ -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;