update deps of flashmla

779033b7 · zhuwenwen · f91443e2 · 779033b7
Commit 779033b7 authored Apr 24, 2025 by zhuwenwen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

requirements/rocm.txt requirements/rocm.txt +1 -0

No files found.
--- a/requirements/rocm.txt
+++ b/requirements/rocm.txt
@@ -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