From 1fd5f330a834eb6c945e43a0c065a35ebba5e84e Mon Sep 17 00:00:00 2001 From: drhead <1313496+drhead@users.noreply.github.com> Date: Sun, 25 May 2025 10:56:00 -0400 Subject: [PATCH] div(2).mul(255) -> mul(127.5) --- latent_preview.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/latent_preview.py b/latent_preview.py index 453bb210b..e59da1165 100644 --- a/latent_preview.py +++ b/latent_preview.py @@ -23,9 +23,8 @@ def preview_to_image(preview_image: torch.Tensor): preview_image = preview_image.cpu() preview_image.add_(1.0) - preview_image.div_(2.0) + preview_image.mul_(127.5) preview_image.clamp_(0, 1) # change scale from -1..1 to 0..1 and clamp - preview_image.mul_(255.) # change to uint8 range preview_image.round_() # default behavior when casting is truncate which is wrong for image processing return Image.fromarray(preview_image.to(dtype=torch.uint8).numpy())