mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-01-30 21:09:46 +00:00
21 lines
679 B
Python
21 lines
679 B
Python
def apply_controlnet_advanced(
|
|
unet,
|
|
controlnet,
|
|
image_bhwc,
|
|
strength,
|
|
start_percent,
|
|
end_percent,
|
|
positive_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)
|
|
return m
|