Fix controlnet regression.

This commit is contained in:
comfyanonymous 2024-08-09 12:08:58 -04:00
parent 5acdadc9f3
commit 86a97e91fc
1 changed files with 1 additions and 1 deletions

View File

@ -173,7 +173,7 @@ class ControlBase:
class ControlNet(ControlBase):
def __init__(self, control_model=None, global_average_pooling=False, compression_ratio=8, latent_format=None, device=None, load_device=None, manual_cast_dtype=None, extra_conds=[], strength_type=StrengthType.CONSTANT):
def __init__(self, control_model=None, global_average_pooling=False, compression_ratio=8, latent_format=None, device=None, load_device=None, manual_cast_dtype=None, extra_conds=["y"], strength_type=StrengthType.CONSTANT):
super().__init__(device)
self.control_model = control_model
self.load_device = load_device