Windows: reactive sigint handler after each Ctrl-C (ggerganov#736)

sw · Apr 3, 2023 · 53dbba7 · 53dbba7
1 parent 437e778
commit 53dbba7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/main/main.cpp b/examples/main/main.cpp
@@ -368,6 +368,11 @@ int main(int argc, char ** argv) {
                 // potentially set color to indicate we are taking user input
                 set_console_color(con_st, CONSOLE_COLOR_USER_INPUT);
 
+#if defined (_WIN32)
+                // Windows: must reactivate sigint handler after each signal
+                signal(SIGINT, sigint_handler);
+#endif
+
                 if (params.instruct) {
                     printf("\n> ");
                 }