diff --git a/invokeai/app/models/image.py b/invokeai/app/models/image.py index 1d493e65c2..544951ea34 100644 --- a/invokeai/app/models/image.py +++ b/invokeai/app/models/image.py @@ -28,6 +28,7 @@ class ImageCategory(str, Enum, metaclass=MetaEnum): GENERAL = "general" CONTROL = "control" + MASK = "mask" OTHER = "other"