diff --git a/src/nodes/helpers.py b/src/nodes/helpers.py index 796c11b..36da74c 100644 --- a/src/nodes/helpers.py +++ b/src/nodes/helpers.py @@ -352,6 +352,8 @@ def save_image(images, filenames, prompt=None, extra_pnginfo=None, compress_leve all_preview_imgs = [] for index, (image, filename) in enumerate(zip(imgs, filenames)): + if filename is None: + continue i = 255. * image.cpu().numpy() img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8)) metadata = PngInfo()