fix: control image size

Bing-su · Oct 11, 2023 · b3ac0bd · b3ac0bd
1 parent 376591b
commit b3ac0bd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/asdff/base.py b/asdff/base.py
@@ -166,4 +166,9 @@ def process_inpainting(
         inpaint_args = self._get_inpaint_args(common, inpaint_only)
         inpaint_args["image"] = crop_image
         inpaint_args["mask_image"] = crop_mask
+
+        if "control_image" in inpaint_args:
+            inpaint_args["control_image"] = inpaint_args["control_image"].resize(
+                crop_image.size
+            )
         return self.inpaint_pipeline(**inpaint_args)