prithivMLmods commited on
Commit
5775c50
·
verified ·
1 Parent(s): 4356de9

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/huggingface/peft.git
3
  transformers-stream-generator
@@ -11,6 +12,7 @@ sentencepiece
11
  opencv-python
12
  transformers
13
  docling-core
 
14
  python-docx
15
  torchvision
16
  matplotlib
@@ -30,7 +32,6 @@ gradio
30
  einops
31
  httpx
32
  click
33
- torch
34
  oss2
35
  fpdf
36
  timm
 
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/huggingface/peft.git
4
  transformers-stream-generator
 
12
  opencv-python
13
  transformers
14
  docling-core
15
+ torch==2.6.0
16
  python-docx
17
  torchvision
18
  matplotlib
 
32
  einops
33
  httpx
34
  click
 
35
  oss2
36
  fpdf
37
  timm