Watch for SIGUSR1 and print stack traces for all threads if signal re…

…ceived
comfyanonymous · Aug 11, 2024 · 6f5b3f1 · 6f5b3f1
1 parent 75b9b55
commit 6f5b3f1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -60,6 +60,9 @@ def execute_script(script_path):
 
 import logging
 
+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())
 

diff --git a/models/checkpoints/put_checkpoints_here b/models/checkpoints/put_checkpoints_here