From ff5c72558605a70ff59b13075aa02e6147e7552d Mon Sep 17 00:00:00 2001
From: Sergey Borisov <stalkek7779@yandex.ru>
Date: Thu, 17 Aug 2023 19:35:03 +0300
Subject: [PATCH] Update mask field type

---
 invokeai/app/invocations/latent.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/invokeai/app/invocations/latent.py b/invokeai/app/invocations/latent.py
index 40f7af8703..e75637bd58 100644
--- a/invokeai/app/invocations/latent.py
+++ b/invokeai/app/invocations/latent.py
@@ -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,
     )