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

Added node for creating mask inpaint

Browse Source
This commit is contained in:
Sergey Borisov
2023-08-18 04:07:40 +03:00
parent e9a294f733
commit cfd827cfad
3 changed files with 122 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -314,7 +314,14 @@ class InpaintMaskField(BaseModel):
"""An inpaint 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")
masked_latents_name: Optional[str] = Field(description="The name of the masked image latents")
class InpaintMaskOutput(BaseInvocationOutput):
"""Base class for nodes that output a single image"""
type: Literal["inpaint_mask_output"] = "inpaint_mask_output"
inpaint_mask: InpaintMaskField = OutputField(description="Mask for inpaint model run")
# endregion