baohuynhbk14 commited on
Commit
d7daddf
·
verified ·
1 Parent(s): b5f4e7b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -13
requirements.txt CHANGED
@@ -1,17 +1,11 @@
1
- torch==2.9.1
 
2
  transformers==4.46.3
3
  tokenizers==0.20.3
 
4
  einops
5
- addict
6
  easydict
7
- gradio>=4.0.0
8
- spaces>=0.20.0
9
- Pillow>=10.0.0
10
- safetensors>=0.4.0
11
- accelerate>=0.24.0
12
- sentencepiece>=0.1.99
13
- protobuf>=3.20.0
14
- torchvision
15
- 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
16
- PyMuPDF>=1.24.0
17
- numpy
 
1
+ torch==2.8.0
2
+ torchvision==0.23.0
3
  transformers==4.46.3
4
  tokenizers==0.20.3
5
+ accelerate
6
  einops
7
+ addict
8
  easydict
9
+ flash-attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.8cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
10
+ PyMuPDF
11
+ hf_transfer