wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-20 19:30:20 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into master

Browse Source
This commit is contained in:
patientx 2024-11-10 10:44:29 +03:00 committed by GitHub
commit 918d3f1ff1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
comfy_extras/nodes_sag.py
View File

@ -58,16 +58,23 @@ def create_blur_map(x0, attn, sigma=3.0, threshold=1.0):
# Global Average Pool # Global Average Pool
mask = attn.mean(1, keepdim=False).sum(1, keepdim=False) > threshold mask = attn.mean(1, keepdim=False).sum(1, keepdim=False) > threshold
f = float(lh) / float(lw) total = mask.shape[-1]
fh = f ** 0.5 x = round(math.sqrt((lh / lw) * total))
fw = (1/f) ** 0.5 xx = None
S = mask.size(1) ** 0.5 for i in range(0, math.floor(math.sqrt(total) / 2)):
w = int(0.5 + S * fw) for j in [(x + i), max(1, x - i)]:
h = int(0.5 + S * fh) if total % j == 0:
xx = j
break
if xx is not None:
break
x = xx
y = total // x
# Reshape # Reshape
mask = ( mask = (
mask.reshape(b, h, w) mask.reshape(b, x, y)
.unsqueeze(1) .unsqueeze(1)
.type(attn.dtype) .type(attn.dtype)
) )