alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 15:15:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix: improve image handling logic to ensure input is always a list or array, see #346

Browse Source
This commit is contained in:
Will Miao
2025-08-09 07:20:28 +08:00
parent e1dda2795a
commit 3050bbe260
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
py/nodes/save_image.py
View File

@@ -418,11 +418,15 @@ class SaveImage:
# Make sure the output directory exists # Make sure the output directory exists
os.makedirs(self.output_dir, exist_ok=True) os.makedirs(self.output_dir, exist_ok=True)
# Ensure images is always a list of images # If images is already a list or array of images, do nothing; otherwise, convert to list
if len(images.shape) == 3: # Single image (height, width, channels) if isinstance(images, (list, np.ndarray)):
images = [images] pass
else: # Multiple images (batch, height, width, channels) else:
images = [img for img in images] # Ensure images is always a list of images
if len(images.shape) == 3: # Single image (height, width, channels)
images = [images]
else: # Multiple images (batch, height, width, channels)
images = [img for img in images]
# Save all images # Save all images
results = self.save_images( results = self.save_images(