Update nodes.py

2024-06-20 04:26:11 +02:00
parent bf6e30e65a
commit 470e01bcff
1 changed files with 2 additions and 2 deletions
@@ -19,7 +19,7 @@ class uncondZeroNode:
            x_orig = args["input"]
            x_orig -= x_orig.mean()
            cond   -= cond.mean()
-            return x_orig - cond / cond.std() ** .5 * scale
+            return x_orig - cond / cond.std() ** .5 * scale # the square root of the std is simply an ever changing scale that fits the bill.

        def uncond_zero_v2(args):
            cond   = args["cond_denoised"]
@@ -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, )