Fix controlnet bug.
This commit is contained in:
parent
cfe1c54de8
commit
5c363a9d86
|
@ -143,7 +143,7 @@ class ControlNet(ControlBase):
|
|||
if control_prev is not None:
|
||||
return control_prev
|
||||
else:
|
||||
return {}
|
||||
return None
|
||||
|
||||
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]:
|
||||
|
|
Loading…
Reference in New Issue