Update controlnet.py

2024-01-29 15:57:50 -08:00
parent 20ccfef8ed
commit 70a831a087
1 changed files with 6 additions and 0 deletions
@@ -822,6 +822,12 @@ class ControlNetForForgeOfficial(scripts.Script):

    def process_unit_after_click_generate(self, p, unit, params, *args, **kwargs):
        h, w, hr_y, hr_x = self.get_target_dimensions(p)
+        
+        has_high_res_fix = (
+            isinstance(p, StableDiffusionProcessingTxt2Img)
+            and getattr(p, 'enable_hr', False)
+        )
+
        input_image, resize_mode = self.choose_input_image(p, unit)
        assert isinstance(input_image, np.ndarray), 'Invalid input image!'