From 09f1d76ed88e2bab7d6938f7f4b6a2c6ac378feb Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 17 Feb 2023 16:21:01 -0500
Subject: [PATCH] Fix an OOM issue.

---
 comfy/ldm/modules/diffusionmodules/openaimodel.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/comfy/ldm/modules/diffusionmodules/openaimodel.py b/comfy/ldm/modules/diffusionmodules/openaimodel.py
index 1769cc0..9054a1c 100644
--- a/comfy/ldm/modules/diffusionmodules/openaimodel.py
+++ b/comfy/ldm/modules/diffusionmodules/openaimodel.py
@@ -786,6 +786,7 @@ class UNetModel(nn.Module):
             if control is not None:
                 hsp += control.pop()
             h = th.cat([h, hsp], dim=1)
+            del hsp
             h = module(h, emb, context)
         h = h.type(x.dtype)
         if self.predict_codebook_ids: