diff --git a/nodes.py b/nodes.py index 21c956f..b5323c8 100644 --- a/nodes.py +++ b/nodes.py @@ -47,7 +47,7 @@ class uncondZeroNode: cond = args["cond_denoised"] cond -= cond.mean() - cond = x_orig - cond / cond.std() ** .5 + cond = x_orig - cond / cond.std() ** .5 * min(scale, 1) sigma = args["sigma"] sigma = sigma.view(sigma.shape[:1] + (1,) * (cond.ndim - 1)) @@ -58,7 +58,7 @@ class uncondZeroNode: cond = ((x - (x_orig - cond)) * (sigma ** 2 + 1.0) ** 0.5) / (sigma) #rescalecfg - x_cfg = uncond + new_scale * scale * (cond - uncond) + x_cfg = uncond + new_scale * max(scale, 1) * (cond - uncond) ro_pos = torch.std(cond, dim=(1,2,3), keepdim=True) ro_cfg = torch.std(x_cfg, dim=(1,2,3), keepdim=True) @@ -68,4 +68,4 @@ class uncondZeroNode: m = model.clone() m.set_model_sampler_cfg_function({"uncond_zero_v1":uncond_zero,"uncond_zero_v2":uncond_zero_v2,"uncond_zero_v3":rescale_cfg}[method]) - return (m, ) + return (m, ) \ No newline at end of file