Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add inapint mask field class

Browse Source
This commit is contained in:
Sergey Borisov 2023-08-17 19:19:07 +03:00
parent 5aa7bfebd4
commit bf0dfcac2f
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/invocations/primitives.py
View File

@ -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