"docs/source/ko/vscode:/vscode.git/clone" did not exist on "5802c2e3f27c6b45290773691bbece4091b69ddc"
Commit cfe1c54d authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix controlnet issue.

parent 57beace3
...@@ -155,7 +155,7 @@ class ControlNet(ControlBase): ...@@ -155,7 +155,7 @@ class ControlNet(ControlBase):
self.cond_hint = broadcast_image_to(self.cond_hint, x_noisy.shape[0], batched_number) self.cond_hint = broadcast_image_to(self.cond_hint, x_noisy.shape[0], batched_number)
context = torch.cat(cond['c_crossattn'], 1) context = cond['c_crossattn']
y = cond.get('c_adm', None) y = cond.get('c_adm', None)
if y is not None: if y is not None:
y = y.to(self.control_model.dtype) y = y.to(self.control_model.dtype)
......
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