From 16b2c61a22f828ea77d9f084ca871c63bc5cc283 Mon Sep 17 00:00:00 2001
From: Justin Suess <justin.suess@westpoint.edu>
Date: Wed, 15 Mar 2023 15:39:38 -0400
Subject: [PATCH] fixed color reset on exit (#149)
* fixed color reset on exit
* added sigint handler for ansi_color_reset
* Update main.cpp
---------
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
---
main.cpp | 5 +++++
1 file changed, 5 insertions(+)
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;
}