print stack traces for all threads on SIGUSR1

comfyanonymous · Sep 16, 2024 · 4174ddb · 4174ddb
1 parent 56e8f5e
commit 4174ddb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -65,6 +65,9 @@ def execute_script(script_path):
 import logging
 import utils.extra_config
 
+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())