Skip to content

Commit 772de7c

Browse files
PerpNeg Guider optimizations. (#8753)
1 parent b22e97d commit 772de7c

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

comfy_extras/nodes_perpneg.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import comfy.samplers
55
import comfy.utils
66
import node_helpers
7+
import math
78

89
def perp_neg(x, noise_pred_pos, noise_pred_neg, noise_pred_nocond, neg_scale, cond_scale):
910
pos = noise_pred_pos - noise_pred_nocond
@@ -69,6 +70,12 @@ def predict_noise(self, x, timestep, model_options={}, seed=None):
6970
negative_cond = self.conds.get("negative", None)
7071
empty_cond = self.conds.get("empty_negative_prompt", None)
7172

73+
if model_options.get("disable_cfg1_optimization", False) == False:
74+
if math.isclose(self.neg_scale, 0.0):
75+
negative_cond = None
76+
if math.isclose(self.cfg, 1.0):
77+
empty_cond = None
78+
7279
conds = [positive_cond, negative_cond, empty_cond]
7380

7481
out = comfy.samplers.calc_cond_batch(self.inner_model, conds, x, timestep, model_options)

0 commit comments

Comments
 (0)