From 509ec08e5751bc9c51160f5b731ea94ddd4906ee Mon Sep 17 00:00:00 2001
From: Xuan Son Nguyen <son@huggingface.co>
Date: Thu, 5 Sep 2024 21:03:50 +0200
Subject: [PATCH] bring back --n-predict

---
 common/common.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/common/common.cpp b/common/common.cpp
index 9e959b02b..d04fc5f7f 100644
--- a/common/common.cpp
+++ b/common/common.cpp
@@ -944,7 +944,7 @@ std::vector<llama_arg> gpt_params_parser_init(gpt_params & params, llama_example
         }
     ).set_env("LLAMA_ARG_CTX_SIZE"));
     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),
         [&params](int value) {
             params.n_predict = value;