prithivMLmods commited on
Commit
4c777be
·
verified ·
1 Parent(s): 9632ea7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  git+https://github.com/huggingface/accelerate.git
2
  git+https://github.com/WePOINTS/WePOINTS.git
3
  git+https://github.com/huggingface/peft.git
@@ -24,7 +25,7 @@ pyvips
24
  pillow
25
  gradio
26
  einops
27
- torch==2.5.1
28
  fpdf
29
  timm
30
  av
 
1
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.3/flash_attn-2.7.3+cu12torch2.6cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
2
  git+https://github.com/huggingface/accelerate.git
3
  git+https://github.com/WePOINTS/WePOINTS.git
4
  git+https://github.com/huggingface/peft.git
 
25
  pillow
26
  gradio
27
  einops
28
+ torch==2.6.0
29
  fpdf
30
  timm
31
  av