fix(SaveImage Node)

This commit is contained in:
homfen 2025-08-12 12:20:49 +08:00
parent f4231a80b1
commit 6d9f2737d5

View File

@ -1587,6 +1587,7 @@ class SaveImage:
results = list()
for (batch_number, image) in enumerate(images):
i = 255. * image.cpu().numpy()
i = np.nan_to_num(i, nan=0.0, posinf=255.0, neginf=0.0)
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
metadata = None
if not args.disable_metadata: