diff --git a/main.py b/main.py index 4780a9c69..065acb285 100644 --- a/main.py +++ b/main.py @@ -105,6 +105,9 @@ import threading import gc +import faulthandler, signal +faulthandler.register(signal.SIGUSR1) + if os.name == "nt": logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())