From 94ed8ee6aff2aec034d32d088c079210ea679bf9 Mon Sep 17 00:00:00 2001 From: Jairo Correa Date: Thu, 30 Nov 2023 09:16:16 -0300 Subject: [PATCH] Revert "Uncomment OOM warning from attention" This reverts commit 8e276b67b41e771567ba4c24f5c015b125cf8fd1. --- comfy/ldm/modules/attention.py | 4 ++-- comfy/ldm/modules/diffusionmodules/model.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py index ac0d9c8c5df..f9d90cf454d 100644 --- 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 index 6576df4b68e..33ba64f344a 100644 --- 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