[diffusion] multi-platform: support diffusion on amd and fix encoder loading on MI325 (#13760)

Co-authored-by: Sabre Shao <sabre.shao@amd.com>
Co-authored-by: Yusheng (Ethan) Su <yushengsu.thu@gmail.com>
Co-authored-by: Hubert Lu <Hubert.Lu@amd.com>
Co-authored-by: xsun <sunxiao04@gmail.com>
This commit is contained in:
Yuzhen Zhou
2025-12-19 15:38:46 +08:00
committed by GitHub
co-authored by Sabre Shao Yusheng Su Hubert Lu xsun
parent f2d64e6782
commit 4bf06635fc
34 changed files with 823 additions and 72 deletions
@@ -87,6 +87,7 @@ class CausalWanSelfAttention(nn.Module):
causal=False,
supported_attention_backends=(
AttentionBackendEnum.FA,
AttentionBackendEnum.AITER,
AttentionBackendEnum.TORCH_SDPA,
),
)