298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 10:53:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix an OOM issue.

Browse Source
This commit is contained in:
comfyanonymous 2023-02-17 16:21:01 -05:00
parent d66415c021
commit 09f1d76ed8
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ldm/modules/diffusionmodules/openaimodel.py
View File

@ -786,6 +786,7 @@ class UNetModel(nn.Module):
if control is not None: if control is not None:
hsp += control.pop() hsp += control.pop()
h = th.cat([h, hsp], dim=1) h = th.cat([h, hsp], dim=1)
del hsp
h = module(h, emb, context) h = module(h, emb, context)
h = h.type(x.dtype) h = h.type(x.dtype)
if self.predict_codebook_ids: if self.predict_codebook_ids: