From 833da0541eaa33061998faab4a970b09eb78ec44 Mon Sep 17 00:00:00 2001 From: CMonk Date: Wed, 24 Apr 2024 02:00:14 +0800 Subject: [PATCH] Add mask out the same as load image --- comfy/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/__init__.py b/comfy/__init__.py index dde929e..0016ffc 100644 --- a/comfy/__init__.py +++ b/comfy/__init__.py @@ -329,8 +329,8 @@ class GetImageFromPhotoshopLayerNode: bounds_id = int(bounds_layer_name_and_id_split.pop().strip()[:-1]) return id, bounds_id - RETURN_TYPES = ("IMAGE",) - RETURN_NAMES = ("image_out",) + RETURN_TYPES = ("IMAGE", "MASK") + RETURN_NAMES = ("image_out", "mask_out") FUNCTION = "get_image" CATEGORY = "Photoshop" @@ -343,7 +343,7 @@ class GetImageFromPhotoshopLayerNode: loadImage = LoadImage() (output_image, output_mask) = loadImage.load_image(image_id) - return (output_image,) + return (output_image, output_mask) def cache_images(images): ret = []