From fdd8e5b1fd92de8d0ee20e8166b88c8356d13c52 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Thu, 12 Mar 2026 15:48:50 -0700
Subject: [PATCH] Make repeated Ctrl+C force a shutdown

---
 server.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/server.py b/server.py
index 73f190b6..340f7126 100644
--- a/server.py
+++ b/server.py
@@ -33,6 +33,10 @@ warnings.filterwarnings('ignore', category=UserWarning, message='Field "model_na
 
 
 def signal_handler(sig, frame):
+    # On second Ctrl+C, force an immediate exit
+    signal.signal(signal.SIGINT, signal.SIG_DFL)
+    signal.signal(signal.SIGTERM, signal.SIG_DFL)
+
     logger.info("Received Ctrl+C. Shutting down Text Generation Web UI gracefully.")
 
     # Explicitly stop LlamaServer to avoid __del__ cleanup issues during shutdown