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

Update mask field type

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

3
invokeai/app/invocations/latent.py
View File

@ -24,6 +24,7 @@ from invokeai.app.invocations.primitives import (
ImageOutput,
LatentsField,
LatentsOutput,
InpaintMaskField,
build_latents_output,
)
from invokeai.app.util.controlnet_utils import prepare_control_image
@ -126,7 +127,7 @@ class DenoiseLatentsInvocation(BaseInvocation):
default=None, description=FieldDescriptions.control, input=Input.Connection
)
latents: Optional[LatentsField] = InputField(description=FieldDescriptions.latents, input=Input.Connection)
mask: Optional[ImageField] = InputField(
mask: Optional[InpaintMaskField] = InputField(
default=None,
description=FieldDescriptions.mask,
)