mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Clean-up Ctrl+C Shutdown (#4802)

Browse source
This commit is contained in:
erew123 2023-12-05 05:16:16 +00:00 • committed by GitHub
parent 0f828ea441
commit f786aa3caa
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 20 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
one_click.py
View file

@ -4,6 +4,7 @@ import hashlib
import os
import platform
import re
import signal
import site
import subprocess
import sys
@ -27,6 +28,13 @@ else:
flags = f"{' '.join([flag for flag in sys.argv[1:] if flag != '--update'])} {CMD_FLAGS}"
def signal_handler(sig, frame):
sys.exit(0)
signal.signal(signal.SIGINT, signal_handler)
def is_linux():
return sys.platform.startswith("linux")