alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-26 07:35: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

4
py/nodes/save_image.py
View File

@@ -418,6 +418,10 @@ 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)
# If images is already a list or array of images, do nothing; otherwise, convert to list
if isinstance(images, (list, np.ndarray)):
pass
else:
# Ensure images is always a list of images # Ensure images is always a list of images
if len(images.shape) == 3: # Single image (height, width, channels) if len(images.shape) == 3: # Single image (height, width, channels)
images = [images] images = [images]