fffiloni commited on
Commit
56d974d
1 Parent(s): ab9dc1e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  gradio
2
  tqdm
3
- torch
 
4
  torchvision
5
  numpy
6
  ninja
1
  gradio
2
  tqdm
3
+ torch==2.0.1+cu118
4
+ --extra-index-url https://download.pytorch.org/whl/cu118
5
  torchvision
6
  numpy
7
  ninja