diff --git a/comfy-nodes/input_image.py b/comfy-nodes/input_image.py index 4c41847..101512e 100755 --- a/comfy-nodes/input_image.py +++ b/comfy-nodes/input_image.py @@ -196,7 +196,7 @@ class ShellAgentPluginInputImage: if encrypt: decoded_image = xor_decrypt_bytes(decoded_image, ENCRYPTION_KEY) - image = Image.open(BytesIO(decoded_image)) + image = safe_open_image(decoded_image) else: if not os.path.isfile(image_path): # abs path # local path