prithivMLmods commited on
Commit
73acf00
·
verified ·
1 Parent(s): b696989

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -3
requirements.txt CHANGED
@@ -1,4 +1,4 @@
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
@@ -11,8 +11,6 @@ pyvips-binary
11
  sentencepiece
12
  opencv-python
13
  docling-core
14
- transformers
15
- torch==2.6.0
16
  python-docx
17
  torchvision
18
  matplotlib
@@ -36,6 +34,7 @@ gradio
36
  einops
37
  httpx
38
  click
 
39
  oss2
40
  fpdf
41
  timm
 
1
+ git+https://github.com/huggingface/transformers.git@v4.57.1
2
  git+https://github.com/huggingface/accelerate.git
3
  git+https://github.com/huggingface/peft.git
4
  transformers-stream-generator
 
11
  sentencepiece
12
  opencv-python
13
  docling-core
 
 
14
  python-docx
15
  torchvision
16
  matplotlib
 
34
  einops
35
  httpx
36
  click
37
+ torch
38
  oss2
39
  fpdf
40
  timm