Skip to content

GitLab

Commit 5c363a9d authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix controlnet bug.

parent cfe1c54d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/controlnet.py
View file @ 5c363a9d
...@@ -143,7 +143,7 @@ class ControlNet(ControlBase): ...@@ -143,7 +143,7 @@ class ControlNet(ControlBase):
if control_prev is not None: if control_prev is not None:
return control_prev return control_prev
else: else:
return {} return None
output_dtype = x_noisy.dtype output_dtype = x_noisy.dtype
if self.cond_hint is None or x_noisy.shape[2] * 8 != self.cond_hint.shape[2] or x_noisy.shape[3] * 8 != self.cond_hint.shape[3]: if self.cond_hint is None or x_noisy.shape[2] * 8 != self.cond_hint.shape[2] or x_noisy.shape[3] * 8 != self.cond_hint.shape[3]:
......
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