Prevent potential int overflow in SaveImage

1f717903 · comfyanonymous · 44294a44 · 1f717903
Commit 1f717903 authored Mar 11, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nodes.py nodes.py +1 -1

No files found.
--- a/nodes.py
+++ b/nodes.py
@@ -790,7 +790,7 @@ class SaveImage:
        paths = list()
        for image in images:
            i = 255. * image.cpu().numpy()
-            img = Image.fromarray(i.astype(np.uint8))
+            img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
            metadata = PngInfo()
            if prompt is not None:
                metadata.add_text("prompt", json.dumps(prompt))