Bill Psomas commited on
Commit
ada65de
·
1 Parent(s): 4f0b9a4

update demo

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  numpy==1.24.3
2
  matplotlib==3.7.1
3
  pillow==9.5.0
4
- torch==1.8.0
5
- torchvision==0.9.0
 
1
  numpy==1.24.3
2
  matplotlib==3.7.1
3
  pillow==9.5.0
4
+ torch==1.8.0+cu111
5
+ torchvision==0.9.0+cu111