Revert "Uncomment OOM warning from attention"

This reverts commit 8e276b6.
jn-jairo · Nov 30, 2023 · 94ed8ee · 94ed8ee
1 parent 8e276b6
commit 94ed8ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py
@@ -237,12 +237,12 @@ def attention_split(q, k, v, heads, mask=None):
                 model_management.soft_empty_cache(True)
                 if cleared_cache == False:
                     cleared_cache = True
-                    print("out of memory error, emptying cache and trying again")
+                    # print("out of memory error, emptying cache and trying again")
                     continue
                 steps *= 2
                 if steps > 64:
                     raise e
-                print("out of memory error, increasing steps and trying again", steps)
+                # print("out of memory error, increasing steps and trying again", steps)
             else:
                 raise e
 

diff --git a/comfy/ldm/modules/diffusionmodules/model.py b/comfy/ldm/modules/diffusionmodules/model.py
@@ -189,7 +189,7 @@ def slice_attention(q, k, v):
             steps *= 2
             if steps > 128:
                 raise e
-            print("out of memory error, increasing steps and trying again", steps)
+            # print("out of memory error, increasing steps and trying again", steps)
 
     return r1