Fix an OOM issue.

2023-02-17 16:21:01 -05:00 · 2023-02-17 16:21:01 -05:00 · 09f1d76ed8
parent d66415c021
commit 09f1d76ed8
1 changed files with 1 additions and 0 deletions
--- 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: