Update nodes.py

2024-08-15 01:44:09 +02:00
parent 784c1e4f38
commit 0d92ee6c7c
1 changed files with 2 additions and 2 deletions
@@ -1221,7 +1221,7 @@ class gradient_scaling_pre_cfg_node:
                return conds_out
            else:
                target_scales = maximum_scale * mask_as_weight * strength + torch.full_like(conds_out[1], cond_scale) * (1 - mask_as_weight * strength)
-                conds_out[1]  = make_new_uncond_at_scale(conds_out[0],conds_out[1],cond_scale,maximum_scale)
+                conds_out[1]  = make_new_uncond_at_scale(conds_out[0],conds_out[1],cond_scale,target_scales)
                return conds_out

        m = model.clone()
@@ -1323,4 +1323,4 @@ class GradientRGBImage:

        mask_gradient = gradient.expand(1, height, width, 1).squeeze(-1)

-        return (rgb_image,mask_gradient,)
+        return (rgb_image,mask_gradient,)