Skip to content

Commit

Permalink
[misc] fix flash attn requirement
Browse files Browse the repository at this point in the history
  • Loading branch information
@ver217
ver217 committed Aug 16, 2023
1 parent aafb974 commit 1e2cd05
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion requirements/requirements-test.txt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,4 +16,4 @@ triton==2.0.0.dev20221202
requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611
SentencePiece
ninja
flash-attn
flash-attn==2.0.5

0 comments on commit 1e2cd05

Please sign in to comment.