Skip to content

Commit e600520

Browse files
Fix hunyuan refiner blownout colors at noise aug less than 0.25 (#9832)
1 parent fd2b820 commit e600520

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

comfy/model_base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1449,7 +1449,9 @@ def concat_cond(self, **kwargs):
14491449
image = utils.resize_to_batch_size(image, noise.shape[0])
14501450
if noise_augmentation > 0:
14511451
noise = torch.randn(image.shape, generator=torch.manual_seed(kwargs.get("seed", 0) - 10), dtype=image.dtype, device="cpu").to(image.device)
1452-
image = noise_augmentation * noise + (1.0 - noise_augmentation) * image
1452+
image = noise_augmentation * noise + min(1.0 - noise_augmentation, 0.75) * image
1453+
else:
1454+
image = 0.75 * image
14531455
return image
14541456

14551457
def extra_conds(self, **kwargs):

0 commit comments

Comments
 (0)