From 01e79d606eca21910bd0731641476054af20d6fa Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 20 Sep 2023 04:08:44 -0500 Subject: [PATCH] Fixed ControlNet check - looks for exacty type match instead of isinstance to not mistake ControlLora for ControlNet --- control.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/control.py b/control.py index 22e14d4..18227bf 100644 --- a/control.py +++ b/control.py @@ -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):