nodes/Kosinkadink-ComfyUI-Advanced-ControlNet
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added fix for certain ControlNets after most recent ComfyUI update

Browse Source
This commit is contained in:
Jedrzej Kosinski
2023-10-25 08:16:11 -05:00
parent d1ad745d63
commit 1b8800d783
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
control/control.py
+4 -1
View File
@@ -229,7 +229,10 @@ class ControlNetAdvanced(ControlNet):
self.mask_cond_hint = self.mask_cond_hint.to(self.control_model.dtype).to(self.device)
context = cond['c_crossattn']
y = cond.get('c_adm', None)
# uses 'y' in new ComfyUI update
y = cond.get('y', None)
if y is None: # TODO: remove this in the future since no longer used by newest ComfyUI
y = cond.get('c_adm', None)
if y is not None:
y = y.to(self.control_model.dtype)
control = self.control_model(x=x_noisy.to(self.control_model.dtype), hint=self.cond_hint, timesteps=t, context=context.to(self.control_model.dtype), y=y)