[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:
@@ -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",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user