bokesyo commited on
Commit
1d9b3d3
·
verified ·
1 Parent(s): 9c67aa9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -4
requirements.txt CHANGED
@@ -5,9 +5,7 @@ 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
 
5
  sentencepiece==0.1.99
6
  numpy==1.26.0
7
  transformers==4.40.2
8
+ timm
9
 
10
  torch==2.1.2
11
+ torchvision==0.16.2