Fix regression.

This commit is contained in:
comfyanonymous 2024-08-09 21:45:21 -04:00
parent 6678d5cf65
commit 1b5b8ca81a
1 changed files with 1 additions and 0 deletions

View File

@ -322,6 +322,7 @@ class ControlLora(ControlNet):
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)