andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make tonemap latent work on any dim latents. (#12363)

Browse Source
This commit is contained in:
comfyanonymous
2026-02-08 18:16:40 -08:00
committed by GitHub
parent f350a84261
commit a0302cc6a8
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy_extras/nodes_latent.py
+3 -2
View File
@@ -391,8 +391,9 @@ class LatentOperationTonemapReinhard(io.ComfyNode):
latent_vector_magnitude = (torch.linalg.vector_norm(latent, dim=(1)) + 0.0000000001)[:,None] latent_vector_magnitude = (torch.linalg.vector_norm(latent, dim=(1)) + 0.0000000001)[:,None]
normalized_latent = latent / latent_vector_magnitude normalized_latent = latent / latent_vector_magnitude
mean = torch.mean(latent_vector_magnitude, dim=(1,2,3), keepdim=True) dims = list(range(1, latent_vector_magnitude.ndim))
std = torch.std(latent_vector_magnitude, dim=(1,2,3), keepdim=True) mean = torch.mean(latent_vector_magnitude, dim=dims, keepdim=True)
std = torch.std(latent_vector_magnitude, dim=dims, keepdim=True)
top = (std * 5 + mean) * multiplier top = (std * 5 + mean) * multiplier