diff --git a/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py b/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py index 89bb4a20..aa99fa98 100644 --- a/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py +++ b/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py @@ -80,7 +80,7 @@ class ControlNetExampleForge(scripts.Script): unet = apply_controlnet_advanced(unet=unet, controlnet=self.model, image=canny_image, strength=1.0, start_percent=0.0, end_percent=1.0, - advanced_weighting=None) + positive_advanced_weighting=None, negative_advanced_weighting=None) p.sd_model.forge_objects.unet = unet diff --git a/modules_forge/controlnet.py b/modules_forge/controlnet.py index 4a5d90d6..316747a0 100644 --- a/modules_forge/controlnet.py +++ b/modules_forge/controlnet.py @@ -1,2 +1,10 @@ -def apply_controlnet_advanced(unet, controlnet, image, strength, start_percent, end_percent, advanced_weighting=None): +def apply_controlnet_advanced( + unet, + controlnet, + image, + strength, + start_percent, + end_percent, + positive_advanced_weighting=None, + negative_advanced_weighting=None): return unet