wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 23:00:51 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

div(2).mul(255) -> mul(127.5)

Browse Source
This commit is contained in:
drhead 2025-05-25 10:56:00 -04:00 committed by GitHub
parent 7dd4265cc0
commit 1fd5f330a8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
latent_preview.py
View File

@ -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())