nodes/banodoco-steerable-motion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed ControlNet check - looks for exacty type match instead of isinstance to not mistake ControlLora for ControlNet

Browse Source
This commit is contained in:
Jedrzej Kosinski
2023-09-20 04:08:44 -05:00
parent 2b144c4599
commit 01e79d606e
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
control.py
+2 -2
View File
@@ -319,8 +319,8 @@ class T2IAdapterAdvanced(T2IAdapter):
def load_controlnet(ckpt_path, timestep_keyframe: TimestepKeyframeGroup=None, model=None):
control = comfy_cn.load_controlnet(ckpt_path, model=model)
# if ControlNet returned, transform it into ControlNetAdvanced
if isinstance(control, ControlNet):
# if exactly ControlNet returned, transform it into ControlNetAdvanced
if type(control) == ControlNet:
return ControlNetAdvanced(control.control_model, timestep_keyframe, global_average_pooling=control.global_average_pooling)
# if T2IAdapter returned, transform it into T2IAdapterAdvanced
elif isinstance(control, T2IAdapter):