From 6d9f2737d5db248a0c09b58ea253fa857ae931a7 Mon Sep 17 00:00:00 2001
From: homfen <homfen@gmail.com>
Date: Tue, 12 Aug 2025 12:20:49 +0800
Subject: [PATCH] fix(SaveImage Node)

---
 nodes.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes.py b/nodes.py
index 9448f9c1b..03a489dd1 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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: