Skip to content

GitLab

"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "9af6ab8f892b970bfe493098b50f461369012cf6"
Commit 09f1d76e authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix an OOM issue.

parent d66415c0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/ldm/modules/diffusionmodules/openaimodel.py
View file @ 09f1d76e
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment