From 05f05534b95d8afb5eb0867f3bb425e7e88a6cf5 Mon Sep 17 00:00:00 2001
From: aria1th <35677394+aria1th@users.noreply.github.com>
Date: Sun, 2 Mar 2025 15:01:48 +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 078734e..9ef7038 100644
--- a/io_node.py
+++ b/io_node.py
@@ -328,7 +328,7 @@ class SaveImageWebpCustomNode:
                 })
             
             with filelock.FileLock(filelock_path, timeout=10): # timeout 10 seconds should be enough for most cases
-                full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, output_dir, images[0].shape[1], images[0].shape[0])
+                full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, output_dir, img.size[1], img.size[0])
                 counter_len = len(str(len(images))) # for padding
                 #file = f"{filename}_{counter:05}_.webp"
                 file = f"{filename}_{str(counter).zfill(max(5, counter_len))}_.webp"