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 # 默认配置