root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-25 02:44:36 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

bounds checking for input prefix (#492)

Browse Source
This commit is contained in:
anzz1 2023-03-25 14:42:09 +02:00 committed by GitHub
parent fbd4d38c64
commit e899bf54b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils.cpp
View File

@ -156,7 +156,11 @@ bool gpt_params_parse(int argc, char ** argv, gpt_params & params) {
} else if (arg == "--random-prompt") { } else if (arg == "--random-prompt") {
params.random_prompt = true; params.random_prompt = true;
} else if (arg == "--in-prefix") { } else if (arg == "--in-prefix") {
params.input_prefix = argv[++i]; if (++i >= argc) {
invalid_param = true;
break;
}
params.input_prefix = argv[i];
} else { } else {
fprintf(stderr, "error: unknown argument: %s\n", arg.c_str()); fprintf(stderr, "error: unknown argument: %s\n", arg.c_str());
gpt_print_usage(argc, argv, params); gpt_print_usage(argc, argv, params);