Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "3fd31eef518b73ee592f82435f3d370a716ead4f"
Commit b12b48e1 authored by comfyanonymous's avatar comfyanonymous
Browse files

cleanup.

parent 329c5719
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
comfy_extras/nodes_sag.py
View file @ b12b48e1
...@@ -111,7 +111,6 @@ class SelfAttentionGuidance: ...@@ -111,7 +111,6 @@ class SelfAttentionGuidance:
m = model.clone() m = model.clone()
attn_scores = None attn_scores = None
mid_block_shape = None
# TODO: make this work properly with chunked batches # TODO: make this work properly with chunked batches
# currently, we can only save the attn from one UNet call # currently, we can only save the attn from one UNet call
...@@ -134,7 +133,6 @@ class SelfAttentionGuidance: ...@@ -134,7 +133,6 @@ class SelfAttentionGuidance:
def post_cfg_function(args): def post_cfg_function(args):
nonlocal attn_scores nonlocal attn_scores
nonlocal mid_block_shape
uncond_attn = attn_scores uncond_attn = attn_scores
sag_scale = scale sag_scale = scale
......
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