prithivMLmods commited on
Commit
f85950b
·
verified ·
1 Parent(s): c155180

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,7 +1,5 @@
1
  #git+https://github.com/Dao-AILab/flash-attention.git
2
  https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.8/flash_attn-2.7.4.post1+cu126torch2.7-cp310-cp310-linux_x86_64.whl
3
- git+https://github.com/huggingface/accelerate.git
4
- git+https://github.com/huggingface/peft.git
5
  transformers-stream-generator
6
  transformers==4.51.3
7
  xformers==0.0.30
@@ -16,6 +14,7 @@ python-docx
16
  torchvision
17
  supervision
18
  matplotlib
 
19
  pdf2image
20
  num2words
21
  reportlab
@@ -34,5 +33,6 @@ httpx
34
  click
35
  torch
36
  fpdf
 
37
  timm
38
  av
 
1
  #git+https://github.com/Dao-AILab/flash-attention.git
2
  https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.0.8/flash_attn-2.7.4.post1+cu126torch2.7-cp310-cp310-linux_x86_64.whl
 
 
3
  transformers-stream-generator
4
  transformers==4.51.3
5
  xformers==0.0.30
 
14
  torchvision
15
  supervision
16
  matplotlib
17
+ accelerate
18
  pdf2image
19
  num2words
20
  reportlab
 
33
  click
34
  torch
35
  fpdf
36
+ peft
37
  timm
38
  av