diff --git a/invokeai/app/invocations/primitives.py b/invokeai/app/invocations/primitives.py index 61ef9fa27c..7444e826e9 100644 --- a/invokeai/app/invocations/primitives.py +++ b/invokeai/app/invocations/primitives.py @@ -305,6 +305,18 @@ class ImageCollectionInvocation(BaseInvocation): return ImageCollectionOutput(collection=self.collection) +# endregion + +# region InpaintMask + + +class InpaintMaskField(BaseModel): + """An inapint mask field""" + + mask_name: str = Field(description="The name of the mask image") + masked_latens_name: Optional[str] = Field(description="The name of the masked image latents") + + # endregion # region Latents