bokesyo commited on
Commit
63e9c5e
1 Parent(s): 42fd4b0

fix dependency

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -4,8 +4,8 @@ tqdm
4
  gradio
5
  Pillow==10.1.0
6
  timm==0.9.10
7
- torch==2.2.0
8
- torchvision
9
  transformers==4.40.2
10
  sentencepiece==0.1.99
11
  numpy==1.26.0
 
4
  gradio
5
  Pillow==10.1.0
6
  timm==0.9.10
7
+ torch==2.1.2+cu118
8
+ torchvision==0.16.2+cu118
9
  transformers==4.40.2
10
  sentencepiece==0.1.99
11
  numpy==1.26.0