tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-06-28 12:25:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

run: allow to customize prompt by env var LLAMA_PROMPT_PREFIX (#12041)

Browse Source 
Signed-off-by: Florent Benoit <fbenoit@redhat.com>
This commit is contained in:
Florent BENOIT
2025-02-23 18:15:51 +01:00
committed by GitHub
parent f777a73e18
commit 7ad0779f5d
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/run/run.cpp
View File

@ -977,7 +977,8 @@ static int generate(LlamaData & llama_data, const std::string & prompt, std::str
} }
static int read_user_input(std::string & user_input) { static int read_user_input(std::string & user_input) {
static const char * prompt_prefix = "> "; static const char * prompt_prefix_env = std::getenv("LLAMA_PROMPT_PREFIX");
static const char * prompt_prefix = prompt_prefix_env ? prompt_prefix_env : "> ";
#ifdef WIN32 #ifdef WIN32
printf("\r" LOG_CLR_TO_EOL LOG_COL_DEFAULT "%s", prompt_prefix); printf("\r" LOG_CLR_TO_EOL LOG_COL_DEFAULT "%s", prompt_prefix);