Skip to content

Commit

Permalink
fix: Add explicit flash_attn requirement for ROCm
Browse files Browse the repository at this point in the history
Signed-off-by: Fabien Dupont <[email protected]>
(cherry picked from commit d74105a)
  • Loading branch information
fabiendupont authored and mergify[bot] committed Oct 8, 2024
1 parent ed833b9 commit 4c05199
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions requirements-rocm.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
flash-attn>=2.6.2,<2.7.0

0 comments on commit 4c05199

Please sign in to comment.