From 63f0fa572d7db84d0792c30844de573c1cb6f660 Mon Sep 17 00:00:00 2001 From: Vinesh Janarthanan <36610342+VJHack@users.noreply.github.com> Date: Sun, 15 Sep 2024 20:35:01 -0500 Subject: [PATCH] Update common/arg.cpp no-context-shift only works for main example Co-authored-by: Georgi Gerganov --- common/arg.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/common/arg.cpp b/common/arg.cpp index f5faddf12..97391ef77 100644 --- a/common/arg.cpp +++ b/common/arg.cpp @@ -703,7 +703,7 @@ gpt_params_context gpt_params_parser_init(gpt_params & params, llama_example ex, [](gpt_params & params) { params.ctx_shift = false; } - )); + ).set_examples({LLAMA_EXAMPLE_MAIN})); add_opt(llama_arg( {"--chunks"}, "N", format("max number of chunks to process (default: %d, -1 = all)", params.n_chunks),