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

Add noise augmentation to hunyuan image refiner. (#9831)

Browse Source 
This was missing and should help with colors being blown out.
This commit is contained in:
comfyanonymous
2025-09-12 13:03:08 -07:00
committed by GitHub
parent d6b977b2e6
commit fd2b820ec2
2 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
comfy_extras/nodes_hunyuan.py
View File

@@ -134,6 +134,7 @@ class HunyuanRefinerLatent:
return {"required": {"positive": ("CONDITIONING", ),
"negative": ("CONDITIONING", ),
"latent": ("LATENT", ),
"noise_augmentation": ("FLOAT", {"default": 0.10, "min": 0.0, "max": 1.0, "step": 0.01}),
}}
RETURN_TYPES = ("CONDITIONING", "CONDITIONING", "LATENT")
@@ -141,11 +142,10 @@ class HunyuanRefinerLatent:
FUNCTION = "execute"
def execute(self, positive, negative, latent):
def execute(self, positive, negative, latent, noise_augmentation):
latent = latent["samples"]
positive = node_helpers.conditioning_set_values(positive, {"concat_latent_image": latent})
negative = node_helpers.conditioning_set_values(negative, {"concat_latent_image": latent})
positive = node_helpers.conditioning_set_values(positive, {"concat_latent_image": latent, "noise_augmentation": noise_augmentation})
negative = node_helpers.conditioning_set_values(negative, {"concat_latent_image": latent, "noise_augmentation": noise_augmentation})
out_latent = {}
out_latent["samples"] = torch.zeros([latent.shape[0], 32, latent.shape[-3], latent.shape[-2], latent.shape[-1]], device=comfy.model_management.intermediate_device())
return (positive, negative, out_latent)