From 2c7bb101b41948ed4e4476ae144550ff8c2e8180 Mon Sep 17 00:00:00 2001 From: Panchovix Date: Fri, 25 Jul 2025 13:01:40 -0400 Subject: [PATCH] Change per_block_mean to False Not sure what are the implications of this, but it seems to make sage3 actually run, --- comfy/ldm/modules/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/modules/attention.py b/comfy/ldm/modules/attention.py index e49679e07..200c07678 100644 --- a/comfy/ldm/modules/attention.py +++ b/comfy/ldm/modules/attention.py @@ -514,7 +514,7 @@ def attention_sage(q, k, v, heads, mask=None, attn_precision=None, skip_reshape= 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=True) + 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":