Uncomment OOM warning from attention

jn-jairo · Nov 21, 2023 · 8e276b6 · 8e276b6
1 parent f53e211
commit 8e276b6
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