lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-03 22:58:58 +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

7
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@@ -463,10 +463,13 @@ class ControlNetForForgeOfficial(scripts.Script):
params: ControlNetCachedParameters,
*args, **kwargs):
h, w, hr_y, hr_x = self.get_target_dimensions(p)
is_hr_pass = p.get('is_hr_pass', False)
is_hr_pass = getattr(p, 'is_hr_pass', False)
cond = params.control_cond_for_hr_fix if is_hr_pass else params.control_cond
kwargs.update(dict(unit=unit, params=params))
params.preprocessor.process_before_every_sampling(process=p, cond=cond, **kwargs)
params.model.process_before_every_sampling(process=p, cond=cond, **kwargs)
a = 0
return

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

2
modules_forge/supported_preprocessor.py
View File

@@ -61,7 +61,7 @@ class Preprocessor:
model_management.free_memory(required_memory, device=model_management.get_torch_device())
return
def process_before_every_sampling(self, process, cnet):
def process_before_every_sampling(self, process, cond, *args, **kwargs):
return
def __call__(self, input_image, resolution, slider_1=None, slider_2=None, slider_3=None, **kwargs):