diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2023-03-25 21:51:41 +0200 |
---|---|---|
committer | Georgi Gerganov <ggerganov@gmail.com> | 2023-03-25 21:51:41 +0200 |
commit | 79b2b266db6b198b5af450982c3cd61120fac951 (patch) | |
tree | 81859f0020c4ba4d55bfc2307cf7c9a8cd75595c /examples/common.cpp | |
parent | e2d490dafd860eaaaf9aa8008ab790527d556daf (diff) |
If n_predict == -1, generate forever
Diffstat (limited to 'examples/common.cpp')
-rw-r--r-- | examples/common.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/common.cpp b/examples/common.cpp index 866a6b0..2ab000f 100644 --- a/examples/common.cpp +++ b/examples/common.cpp @@ -204,7 +204,7 @@ void gpt_print_usage(int /*argc*/, char ** argv, const gpt_params & params) { fprintf(stderr, " --in-prefix STRING string to prefix user inputs with (default: empty)\n"); fprintf(stderr, " -f FNAME, --file FNAME\n"); fprintf(stderr, " prompt file to start generation.\n"); - fprintf(stderr, " -n N, --n_predict N number of tokens to predict (default: %d)\n", params.n_predict); + fprintf(stderr, " -n N, --n_predict N number of tokens to predict (default: %d, -1 - infinity)\n", params.n_predict); fprintf(stderr, " --top_k N top-k sampling (default: %d)\n", params.top_k); fprintf(stderr, " --top_p N top-p sampling (default: %.1f)\n", params.top_p); fprintf(stderr, " --repeat_last_n N last n tokens to consider for penalize (default: %d)\n", params.repeat_last_n); |