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: