From f84a6a7bb8b9d7a1c31cd6cc6e5d61d15c6c2573 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gr=C3=A9gory=20Howard?= <gregory.f.howard@gmail.com>
Date: Fri, 15 Aug 2025 20:08:02 +0200
Subject: [PATCH] Update nodes.py

added numpy array possibility in method save_images.
---
 nodes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 860a236aa..2cb9b0bc2 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1586,7 +1586,10 @@ class SaveImage:
         full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0])
         results = list()
         for (batch_number, image) in enumerate(images):
-            i = 255. * image.cpu().numpy()
+            if hasattr(image, "cpu"):  # torch tensor
+                i = 255. * image.cpu().numpy()
+            else:
+                i = 255. * image
             img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
             metadata = None
             if not args.disable_metadata: