From bf0dfcac2fb1b820b441fd0819ffe9371bc6b58e Mon Sep 17 00:00:00 2001 From: Sergey Borisov Date: Thu, 17 Aug 2023 19:19:07 +0300 Subject: [PATCH] Add inapint mask field class --- invokeai/app/invocations/primitives.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) 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