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

Update controlnet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 15:43:43 -08:00
parent d0e3124d62
commit 929ee203c2
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -855,6 +855,8 @@ class ControlNetForForgeOfficial(scripts.Script):
def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs): def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs):
h, w, hr_y, hr_x = self.get_target_dimensions(p) h, w, hr_y, hr_x = self.get_target_dimensions(p)
input_image, resize_mode = self.choose_input_image(p, unit) input_image, resize_mode = self.choose_input_image(p, unit)
assert isinstance(input_image, np.ndarray), 'Invalid input image!'
return return
def process_unit_before_every_sampling(self, p, unit, params, *args, **kwargs): def process_unit_before_every_sampling(self, p, unit, params, *args, **kwargs):