Skip to content

Commit

Permalink
closes #334; prefer newer version of flash-attn (#394)
Browse files Browse the repository at this point in the history
  • Loading branch information
chengzeyi authored Dec 16, 2024
1 parent 6707840 commit 0cbdf4a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def get_cuda_version():
"imageio",
"imageio-ffmpeg",
"optimum-quanto",
"flash_attn>=2.6.3"
"flash_attn>=2.7.0" # flash_attn>=2.7.0 with torch>=2.4.0 wraps ops with torch.ops
],
extras_require={
"diffusers": [
Expand Down

0 comments on commit 0cbdf4a

Please sign in to comment.