From 164c65c8f940f092503903ac28c868d26c7e9c51 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Wed, 1 Mar 2023 02:03:20 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E9=9B=85=E9=80=80=E5=87=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/server.py b/server.py
index 7c5ef4e..13e578e 100644
--- a/server.py
+++ b/server.py
@@ -234,7 +234,16 @@ def get_tasks(tasks:list=[]):
             time.sleep(2)
     return tasks
 
+import sys                                                                  
+import signal
+
+def quit(signum, frame):
+    print('stop fusion')
+    sys.exit()
+
 def main_dev(opt):
+    signal.signal(signal.SIGINT, quit)
+    signal.signal(signal.SIGTERM, quit)
     model_name = '' # 默认模型
     model = None    # 默认模型
     config = None   # 默认配置