xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 04:47:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

More ruff fixes

Browse Source
This commit is contained in:
Panchovix 2025-07-25 21:22:34 -04:00 committed by GitHub
parent bb7393565b
commit 9a70423973
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/ldm/modules/attention.py
View File

@ -510,9 +510,9 @@ def attention_sage(q, k, v, heads, mask=None, attn_precision=None, skip_reshape=
q_sa3, k_sa3, v_sa3 = map(lambda t: t.transpose(1, 2), (q, k, v)) q_sa3, k_sa3, v_sa3 = map(lambda t: t.transpose(1, 2), (q, k, v))
else: else:
q_sa3, k_sa3, v_sa3 = q, k, v q_sa3, k_sa3, v_sa3 = q, k, v
out = sageattn_blackwell(q_sa3, k_sa3, v_sa3, attn_mask=mask, is_causal=False, per_block_mean=False) out = sageattn_blackwell(q_sa3, k_sa3, v_sa3, attn_mask=mask, is_causal=False, per_block_mean=False)
# Convert back to expected layout # Convert back to expected layout
if tensor_layout == "HND": if tensor_layout == "HND":
if not skip_output_reshape: if not skip_output_reshape: