Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "577b8a2783c2ba9cf4ebbb3c36ea024a6175ef95"
Commit 172984db authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix SAG not working on certain resolutions.

parent 6596654d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy_extras/nodes_sag.py
View file @ 172984db
...@@ -58,7 +58,7 @@ def create_blur_map(x0, attn, sigma=3.0, threshold=1.0): ...@@ -58,7 +58,7 @@ def create_blur_map(x0, attn, sigma=3.0, threshold=1.0):
attn = attn.reshape(b, -1, hw1, hw2) attn = attn.reshape(b, -1, hw1, hw2)
# 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
ratio = round(math.sqrt(lh * lw / hw1)) ratio = math.ceil(math.sqrt(lh * lw / hw1))
mid_shape = [math.ceil(lh / ratio), math.ceil(lw / ratio)] mid_shape = [math.ceil(lh / ratio), math.ceil(lw / ratio)]
# Reshape # Reshape
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment