bokesyo commited on
Commit
b7fd459
·
verified ·
1 Parent(s): fac2ded

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -4
requirements.txt CHANGED
@@ -2,10 +2,12 @@ PyMuPDF
2
  tqdm
3
  gradio
4
  Pillow==10.1.0
5
- torch==2.2.0
6
- torchvision
7
  sentencepiece==0.1.99
8
  numpy==1.26.0
9
  transformers==4.40.2
10
- timm
11
- flash_attn
 
 
 
 
 
2
  tqdm
3
  gradio
4
  Pillow==10.1.0
 
 
5
  sentencepiece==0.1.99
6
  numpy==1.26.0
7
  transformers==4.40.2
8
+
9
+ torch==2.1.2
10
+ torchvision==0.16.2
11
+ https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.2/flash_attn-2.6.2+cu123torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
12
+
13
+ timm