From 09a7095e67e367f8f1096eec64e87d6a3c0a3246 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=B0=91=E9=B9=8F?= Date: Tue, 28 Oct 2025 21:58:18 +0800 Subject: [PATCH] fix sd attention api call --- .../Flux.1-DEV/FLUX1dev/layers/attention_processor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/MindIE/MultiModal/Flux.1-DEV/FLUX1dev/layers/attention_processor.py b/MindIE/MultiModal/Flux.1-DEV/FLUX1dev/layers/attention_processor.py index ac02729309..e53573ce93 100644 --- a/MindIE/MultiModal/Flux.1-DEV/FLUX1dev/layers/attention_processor.py +++ b/MindIE/MultiModal/Flux.1-DEV/FLUX1dev/layers/attention_processor.py @@ -44,7 +44,8 @@ def apply_fa(query, key, value, attention_mask): heads = query.shape[-2] head_dim = query.shape[-1] - hidden_states = attention_forward(query, key, value, attn_mask=attention_mask) + hidden_states = attention_forward(query, key, value, opt_mode="manual", attn_mask=attention_mask, + op_type="fused_attn_score", layout="BSND") return hidden_states.reshape(batch_size, -1, head_dim * heads) -- Gitee