lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 17:09:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 16:40:01 -08:00
parent 8e97e3ceed
commit 7e8ff9ceb5
3 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/supported_controlnet.py
View File

@@ -136,13 +136,13 @@ class ControlNetPatcher(ControlModelPatcher):
self.advanced_frame_weighting = None
self.advanced_sigma_weighting = None
def patch_to_process(self, p, control_image):
unet = p.sd_model.forge_objects.unet
def process_before_every_sampling(self, process, cond, *args, **kwargs):
unet = process.sd_model.forge_objects.unet
unet = apply_controlnet_advanced(
unet=unet,
controlnet=self.model_patcher,
image_bchw=control_image,
image_bchw=cond,
strength=self.strength,
start_percent=self.start_percent,
end_percent=self.end_percent,
@@ -151,7 +151,7 @@ class ControlNetPatcher(ControlModelPatcher):
advanced_frame_weighting=self.advanced_frame_weighting,
advanced_sigma_weighting=self.advanced_sigma_weighting)
p.sd_model.forge_objects.unet = unet
process.sd_model.forge_objects.unet = unet
return