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 1ebbad54..2975380c 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 @@ -88,7 +88,8 @@ class ControlNetExampleForge(scripts.Script): unet = apply_controlnet_advanced(unet=unet, controlnet=self.model, image_bhwc=control_image, strength=0.6, start_percent=0.0, end_percent=0.8, - positive_advanced_weighting=None, negative_advanced_weighting=None) + positive_advanced_weighting=None, negative_advanced_weighting=None, + only_apply_to_one_frame_in_a_batch=None) p.sd_model.forge_objects.unet = unet diff --git a/modules_forge/controlnet.py b/modules_forge/controlnet.py index 9754e901..84da62ce 100644 --- a/modules_forge/controlnet.py +++ b/modules_forge/controlnet.py @@ -6,11 +6,14 @@ def apply_controlnet_advanced( start_percent, end_percent, positive_advanced_weighting=None, - negative_advanced_weighting=None): + negative_advanced_weighting=None, + only_apply_to_one_frame_in_a_batch=None, +): cnet = controlnet.copy().set_cond_hint(image_bhwc.movedim(-1, 1), strength, (start_percent, end_percent)) cnet.positive_advanced_weighting = positive_advanced_weighting cnet.negative_advanced_weighting = negative_advanced_weighting + cnet.only_apply_to_one_frame_in_a_batch = only_apply_to_one_frame_in_a_batch m = unet.clone() m.add_patched_controlnet(cnet)