298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-13 19:07:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix a bug in the attn_masked redux code when using weight=1.0

Browse Source
This commit is contained in:
Raphael Walker 2025-02-06 14:22:05 +01:00
parent 14880e6dba
commit d7e8d0af01
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nodes.py
View File

@ -1064,7 +1064,8 @@ class StyleModelApply:
for t in conditioning: for t in conditioning:
(txt, keys) = t (txt, keys) = t
keys = keys.copy() keys = keys.copy()
if strength_type == "attn_bias" and strength != 1.0: # even if the strength is 1.0 (i.e, no change), if there's already a mask, we have to add to it
if strength_type == "attn_bias" and strength != 1.0 and "attention_mask" not in keys:
# math.log raises an error if the argument is zero # math.log raises an error if the argument is zero
# torch.log returns -inf, which is what we want # torch.log returns -inf, which is what we want
attn_bias = torch.log(torch.Tensor([strength])) attn_bias = torch.log(torch.Tensor([strength]))