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 ...@@ -20,6 +20,7 @@ setuptools_scm>=8
torch == 2.4.1 torch == 2.4.1
triton == 3.0.0 triton == 3.0.0
flash_attn == 2.6.1 flash_attn == 2.6.1
flash_mla == 1.0.0
lmslim == 0.2.1 lmslim == 0.2.1
numa numa
python-multipart 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