lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-04 23:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 16:31:45 -08:00
parent eaf80f71f8
commit 8e97e3ceed
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -389,7 +389,12 @@ class ControlNetForForgeOfficial(scripts.Script):
return h, w, hr_y, hr_x
def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs):
def process_unit_after_click_generate(self,
p: StableDiffusionProcessing,
unit: external_code.ControlNetUnit,
params: ControlNetCachedParameters,
*args, **kwargs):
h, w, hr_y, hr_x = self.get_target_dimensions(p)
has_high_res_fix = (
@@ -452,9 +457,18 @@ class ControlNetForForgeOfficial(scripts.Script):
return
def process_unit_before_every_sampling(self, p, unit, params, *args, **kwargs):
def process_unit_before_every_sampling(self,
p: StableDiffusionProcessing,
unit: external_code.ControlNetUnit,
params: ControlNetCachedParameters,
*args, **kwargs):
h, w, hr_y, hr_x = self.get_target_dimensions(p)
is_hr_pass = p.is_hr_pass
is_hr_pass = p.get('is_hr_pass', False)
cond = params.control_cond_for_hr_fix if is_hr_pass else params.control_cond
a = 0
return
def process(self, p, *args, **kwargs):