Make inpaint workflow compatible with requests from older plugin version

Acly · Jul 7, 2024 · 885c2b0 · 885c2b0
1 parent 7377f86
commit 885c2b0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/ai_diffusion/api.py b/ai_diffusion/api.py
@@ -32,6 +32,7 @@ class ExtentInput:
 class ImageInput:
     extent: ExtentInput
     initial_image: Image | None = None
+    initial_mask: Image | None = None  # deprecated (1.20.0) - hires_mask is scaled during workflow
     hires_image: Image | None = None
     hires_mask: Image | None = None
 

diff --git a/ai_diffusion/workflow.py b/ai_diffusion/workflow.py
@@ -619,8 +619,14 @@ def inpaint(
     in_image = scale_to_initial(extent, w, in_image, models)
     in_mask = w.load_mask(ensure(images.hires_mask))
     in_mask = apply_grow_feather(w, in_mask, params)
-    initial_mask = scale_to_initial(extent, w, in_mask, models, is_mask=True)
-    cropped_mask = w.crop_mask(in_mask, target_bounds)
+    if images.initial_mask:
+        # Deprecated in 1.20.0 - initial_mask is no longer used
+        initial_mask = w.load_mask(images.initial_mask)
+        initial_mask = scale_to_initial(extent, w, initial_mask, models, is_mask=True)
+        cropped_mask = in_mask
+    else:
+        initial_mask = scale_to_initial(extent, w, in_mask, models, is_mask=True)
+        cropped_mask = w.crop_mask(in_mask, target_bounds)
 
     cond_base = cond.copy()
     cond_base.downscale(extent.input, extent.initial)