Update WAS_Node_Suite.py

This commit is contained in:
Jordan Thompson
2023-08-01 09:33:26 -07:00
parent 1dd8ae2960
commit 009675f44f
+4 -4
View File
@@ -4805,7 +4805,7 @@ class WAS_Load_Image_Batch:
"label": ("STRING", {"default": 'Batch 001', "multiline": False}),
"path": ("STRING", {"default": '', "multiline": False}),
"pattern": ("STRING", {"default": '*', "multiline": False}),
"RGBA": (["false","true"],),
"RGBA_output": (["false","true"],),
},
}
@@ -4815,9 +4815,9 @@ class WAS_Load_Image_Batch:
CATEGORY = "WAS Suite/IO"
def load_batch_images(self, path, pattern='*', index=0, mode="single_image", label='Batch 001', RGBA='false'):
def load_batch_images(self, path, pattern='*', index=0, mode="single_image", label='Batch 001', RGBA_output='false'):
RGBA = (RGBA == 'true')
RGBA_output = (RGBA_output == 'true')
if not os.path.exists(path):
return (None, )
@@ -4837,7 +4837,7 @@ class WAS_Load_Image_Batch:
# Update history
update_history_images(new_paths)
if not RGBA:
if not RGBA_output:
image = image.convert("RGB")
return (pil2tensor(image), filename)