xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2025-12-10 14:34:25 +08: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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats 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):