Add warning to catch torch import mistakes. (#8852)

2025-08-02 15:04:50 +08:00 · 2025-07-09 22:03:27 -07:00
parent 1205afc708
commit 1fd306824d
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