From 9a7042397347671ce905a5de5d4ba4068a534699 Mon Sep 17 00:00:00 2001 From: Panchovix Date: Fri, 25 Jul 2025 21:22:34 -0400 Subject: [PATCH] More ruff fixes --- comfy/ldm/modules/attention.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py index 6a2290629..cc6a84f91 100644 --- a/comfy/ldm/modules/attention.py +++ b/comfy/ldm/modules/attention.py @@ -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)) else: 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) - + # Convert back to expected layout if tensor_layout == "HND": if not skip_output_reshape: