Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "62f5ae68ec2ebee34f2a0bbbf195343165e4ce3a"
Commit 7ea6bb03 authored by comfyanonymous's avatar comfyanonymous
Browse files

Print warning when controlnet can't be applied instead of crashing.

parent dcec1047
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
comfy/ldm/modules/diffusionmodules/openaimodel.py
View file @ 7ea6bb03
...@@ -255,7 +255,10 @@ def apply_control(h, control, name): ...@@ -255,7 +255,10 @@ def apply_control(h, control, name):
if control is not None and name in control and len(control[name]) > 0: if control is not None and name in control and len(control[name]) > 0:
ctrl = control[name].pop() ctrl = control[name].pop()
if ctrl is not None: if ctrl is not None:
h += ctrl try:
h += ctrl
except:
print("warning control could not be applied", h.shape, ctrl.shape)
return h return h
class UNetModel(nn.Module): class UNetModel(nn.Module):
......
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