vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bring back --n-predict

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-09-05 21:03:50 +02:00
parent b1657cb934
commit 509ec08e57
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
common/common.cpp
View file

@ -944,7 +944,7 @@ std::vector<llama_arg> gpt_params_parser_init(gpt_params & params, llama_example
} }
).set_env("LLAMA_ARG_CTX_SIZE")); ).set_env("LLAMA_ARG_CTX_SIZE"));
add_opt(llama_arg( add_opt(llama_arg(
{"-n", "--predict"}, "N", {"-n", "--predict", "--n-predict"}, "N",
format("number of tokens to predict (default: %d, -1 = infinity, -2 = until context filled)", params.n_predict), format("number of tokens to predict (default: %d, -1 = infinity, -2 = until context filled)", params.n_predict),
[&params](int value) { [&params](int value) {
params.n_predict = value; params.n_predict = value;