Skip to content

GitLab

Commit 779033b7 authored by zhuwenwen's avatar zhuwenwen
Browse files

update deps of flashmla

parent f91443e2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
requirements/rocm.txt
View file @ 779033b7
......@@ -20,6 +20,7 @@ setuptools_scm>=8
torch == 2.4.1
triton == 3.0.0
flash_attn == 2.6.1
flash_mla == 1.0.0
lmslim == 0.2.1
numa
python-multipart
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment