diff --git a/main.cpp b/main.cpp index 6dc9ae9..6227db6 100644 --- a/main.cpp +++ b/main.cpp @@ -755,6 +755,7 @@ static bool is_interacting = false; #if defined (__unix__) || (defined (__APPLE__) && defined (__MACH__)) void sigint_handler(int signo) { + printf(ANSI_COLOR_RESET); if (signo == SIGINT) { if (!is_interacting) { is_interacting=true; @@ -1052,5 +1053,9 @@ int main(int argc, char ** argv) { ggml_free(model.ctx); + if (params.use_color) { + printf(ANSI_COLOR_RESET); + } + return 0; }