root commited on
Commit
8dbe2ab
1 Parent(s): fe1b81a
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- -f --extra-index-url https://download.pytorch.org/whl/cu116
2
- torch==1.11.0
3
  torchvision
4
  numpy
5
  transformers==4.19.2
@@ -10,3 +10,4 @@ regex==2022.9.13
10
  timm==0.6.12
11
  imageio
12
  gradio
 
 
1
+ --find-links https://download.pytorch.org/whl/torch_stable.html
2
+ torch==1.11.0+cu113
3
  torchvision
4
  numpy
5
  transformers==4.19.2
 
10
  timm==0.6.12
11
  imageio
12
  gradio
13
+ # cudatoolkit==11.3.1