Skip to content

Commit 50c77d9

Browse files
dunkeronipsychedelicious
authored andcommitted
error message for incorrect mask size
1 parent 358cc03 commit 50c77d9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

invokeai/app/invocations/image.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -734,7 +734,10 @@ def invoke(self, context: InvocationContext) -> ImageOutput:
734734
mask_image = context.images.get_pil(self.mask.image_name, "L")
735735
# Start with corrected image, paste base image where mask is white
736736
result = corrected_image.copy()
737-
result.paste(base_image.convert("RGB"), mask=mask_image)
737+
if mask_image.size != result.size:
738+
raise ValueError("Mask size must match base image size.")
739+
else:
740+
result.paste(base_image.convert("RGB"), mask=mask_image)
738741
else:
739742
result = corrected_image
740743

0 commit comments

Comments
 (0)