From 6f138ebd4bb8e52e3e78ed86a67ed4aebee0713b Mon Sep 17 00:00:00 2001
From: aria1th <35677394+aria1th@users.noreply.github.com>
Date: Sun, 2 Mar 2025 16:00:06 +0900
Subject: [PATCH] Update io_node.py

---
 io_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/io_node.py b/io_node.py
index 9ef7038..6a4ec07 100644
--- a/io_node.py
+++ b/io_node.py
@@ -289,7 +289,7 @@ class SaveImageWebpCustomNode:
     def save_images(self, images, filename_prefix="ComfyUI",subfolder_dir="", prompt=None, extra_pnginfo=None, quality=100, lossless=False, compression=4, optimize=False, metadata_string=""):
         if images is None: # sometimes images is empty
             images = []
-        if not isinstance(images, (list, tuple)):
+        if not isinstance(images, (list, tuple, torch.Tensor)):
             images = [images]
         throw_if_parent_or_root_access(filename_prefix)
         throw_if_parent_or_root_access(subfolder_dir)