Skip to content

Commit

Permalink
Fix regression.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Aug 10, 2024
1 parent 6678d5c commit 1b5b8ca
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions comfy/controlnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 1b5b8ca

Please sign in to comment.