mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-18 08:35:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix hunyuan refiner blownout colors at noise aug less than 0.25 (#9832)

Browse Source
This commit is contained in:
comfyanonymous
2025-09-12 13:35:34 -07:00
committed by GitHub
parent fd2b820ec2
commit e600520f8a
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/model_base.py
View File

@@ -1449,7 +1449,9 @@ class HunyuanImage21Refiner(HunyuanImage21):
image = utils.resize_to_batch_size(image, noise.shape[0]) image = utils.resize_to_batch_size(image, noise.shape[0])
if noise_augmentation > 0: if noise_augmentation > 0:
noise = torch.randn(image.shape, generator=torch.manual_seed(kwargs.get("seed", 0) - 10), dtype=image.dtype, device="cpu").to(image.device) noise = torch.randn(image.shape, generator=torch.manual_seed(kwargs.get("seed", 0) - 10), dtype=image.dtype, device="cpu").to(image.device)
image = noise_augmentation * noise + (1.0 - noise_augmentation) * image image = noise_augmentation * noise + min(1.0 - noise_augmentation, 0.75) * image
else:
image = 0.75 * image
return image return image
def extra_conds(self, **kwargs): def extra_conds(self, **kwargs):