Просмотр исходного кода

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>
Justin Suess 2 лет назад
Родитель
Сommit
16b2c61a22
1 измененных файлов с 5 добавлено и 0 удалено
  1. 5 0
      main.cpp

+ 5 - 0
main.cpp

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