diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index 85e44e32..e714577f 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -942,9 +942,12 @@ class ControlNetForForgeOfficial(scripts.Script): self.post_processors = post_processors def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs): + h, w, hr_y, hr_x = self.get_target_dimensions(p) return def process_unit_before_every_sampling(self, p, unit, params, *args, **kwargs): + h, w, hr_y, hr_x = self.get_target_dimensions(p) + is_hr_pass = p.is_hr_pass return def process(self, p, *args, **kwargs):