prithivMLmods commited on
Commit
14d580f
·
verified ·
1 Parent(s): 8ff9ecc

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
@@ -8,6 +9,7 @@ pyvips-binary
8
  sentencepiece
9
  opencv-python
10
  docling-core
 
11
  python-docx
12
  torchvision
13
  supervision
@@ -24,7 +26,6 @@ pillow
24
  gradio
25
  einops
26
  httpx
27
- torch
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/huggingface/peft.git
4
  transformers-stream-generator
 
9
  sentencepiece
10
  opencv-python
11
  docling-core
12
+ torch==2.6.0
13
  python-docx
14
  torchvision
15
  supervision
 
26
  gradio
27
  einops
28
  httpx
 
29
  fpdf
30
  timm
31
  av