From c55eb784cb5b7eb0a26a029e5e7f3a98eee03a30 Mon Sep 17 00:00:00 2001 From: wbpxre150 Date: Wed, 12 Apr 2023 15:29:59 +0800 Subject: [PATCH] add exit call to interactive mode. --- examples/main/main.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/examples/main/main.cpp b/examples/main/main.cpp index ba153cb82..b53d04713 100644 --- a/examples/main/main.cpp +++ b/examples/main/main.cpp @@ -437,6 +437,15 @@ int main(int argc, char ** argv) { n_remain -= line_inp.size(); } + // exit gracefully when the word "exit" is typed. + if (buffer.length() == 5 && strncmp(buffer.c_str(), "exit", 4) == 0) { + printf("Exit call given, printing stats and exiting.\n"); + llama_print_timings(ctx); + llama_free(ctx); + set_console_color(con_st, CONSOLE_COLOR_DEFAULT); + return 0; + } + input_noecho = true; // do not echo this again }