diff --git a/comfy/controlnet.py b/comfy/controlnet.py index 3d17d2f3070..354b0036707 100644 --- a/comfy/controlnet.py +++ b/comfy/controlnet.py @@ -322,6 +322,7 @@ def __init__(self, control_weights, global_average_pooling=False, device=None): ControlBase.__init__(self, device) self.control_weights = control_weights self.global_average_pooling = global_average_pooling + self.extra_conds += ["y"] def pre_run(self, model, percent_to_timestep_function): super().pre_run(model, percent_to_timestep_function)