[diffusion] hardware: support FA3 attention backend on MUSA (attn backend, 14/N) (#18648)

Signed-off-by: Xiaodong Ye <xiaodong.ye@mthreads.com>
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
R0CKSTAR
2026-04-01 10:49:34 -07:00
committed by GitHub
co-authored by Mick
parent 6098c51bc2
commit ca3286d2d5
4 changed files with 58 additions and 7 deletions
+1 -1
View File
@@ -114,7 +114,7 @@ srt_musa = [
"sglang[runtime_common]",
"torch",
"torch_musa",
"torchada>=0.1.25",
"torchada>=0.1.45",
"mthreads-ml-py",
"numpy<2.0",
]