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

feat: Add InpaintMask Field type

Browse Source
This commit is contained in:
blessedcoolant
2023-08-18 04:54:23 +12:00
parent ff5c725586
commit b213335316
8 changed files with 103 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,10 +1,10 @@
# Copyright (c) 2023 Kyle Schouviller (https://github.com/kyle0654)
from typing import Literal, Optional, Tuple, Union
from anyio import Condition
from pydantic import BaseModel, Field
import torch
from anyio import Condition
from pydantic import BaseModel, Field
from .baseinvocation import (
BaseInvocation,
@ -311,7 +311,7 @@ class ImageCollectionInvocation(BaseInvocation):
class InpaintMaskField(BaseModel):
"""An inapint mask field"""
"""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")