This commit is contained in:
Homfen 2026-01-12 05:25:39 +00:00 committed by GitHub
commit 681ae19dc2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1609,6 +1609,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: