Update mask field type

This commit is contained in:
Sergey Borisov
2023-08-17 19:35:03 +03:00
parent bf0dfcac2f
commit ff5c725586

View File

@ -24,6 +24,7 @@ from invokeai.app.invocations.primitives import (
ImageOutput, ImageOutput,
LatentsField, LatentsField,
LatentsOutput, LatentsOutput,
InpaintMaskField,
build_latents_output, build_latents_output,
) )
from invokeai.app.util.controlnet_utils import prepare_control_image 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 default=None, description=FieldDescriptions.control, input=Input.Connection
) )
latents: Optional[LatentsField] = InputField(description=FieldDescriptions.latents, input=Input.Connection) latents: Optional[LatentsField] = InputField(description=FieldDescriptions.latents, input=Input.Connection)
mask: Optional[ImageField] = InputField( mask: Optional[InpaintMaskField] = InputField(
default=None, default=None,
description=FieldDescriptions.mask, description=FieldDescriptions.mask,
) )