mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-26 06:10:15 +08:00
fix(SaveImage Node)
This commit is contained in:
parent
f4231a80b1
commit
6d9f2737d5
1
nodes.py
1
nodes.py
@ -1587,6 +1587,7 @@ class SaveImage:
|
|||||||
results = list()
|
results = list()
|
||||||
for (batch_number, image) in enumerate(images):
|
for (batch_number, image) in enumerate(images):
|
||||||
i = 255. * image.cpu().numpy()
|
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))
|
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
|
||||||
metadata = None
|
metadata = None
|
||||||
if not args.disable_metadata:
|
if not args.disable_metadata:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user