Add warning to catch torch import mistakes.

2026-09-02 11:57:05 +08:00 · 2025-07-10 00:59:04 -04:00 · 2025-07-10 00:59:04 -04:00 · 233fb5455f
commit 233fb5455f
parent 1205afc708
1 changed files with 3 additions and 0 deletions
--- a/main.py
+++ b/main.py
@ -127,6 +127,9 @@ if __name__ == "__main__":

    import cuda_malloc

+if 'torch' in sys.modules:
+    logging.warning("WARNING: Potential Error in code: Torch already imported, torch should never be imported before this point.")
+
 import comfy.utils

 import execution