Skip to content

GitLab

Commit 1f717903 authored by comfyanonymous's avatar comfyanonymous
Browse files

Prevent potential int overflow in SaveImage

parent 44294a44
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nodes.py
View file @ 1f717903
...@@ -790,7 +790,7 @@ class SaveImage: ...@@ -790,7 +790,7 @@ class SaveImage:
paths = list() paths = list()
for image in images: for image in images:
i = 255. * image.cpu().numpy() 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() metadata = PngInfo()
if prompt is not None: if prompt is not None:
metadata.add_text("prompt", json.dumps(prompt)) metadata.add_text("prompt", json.dumps(prompt))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment