From 8e276b67b41e771567ba4c24f5c015b125cf8fd1 Mon Sep 17 00:00:00 2001 From: Jairo Correa Date: Tue, 21 Nov 2023 19:29:11 -0300 Subject: [PATCH] Uncomment OOM warning from attention --- 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 f9d90cf454d..ac0d9c8c5df 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 33ba64f344a..6576df4b68e 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