Purple11 commited on
Commit
679bfc5
·
1 Parent(s): 2b4fef0

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,4 +1,6 @@
1
  opencv-python==4.1.2.30
 
 
2
  omegaconf==2.1.1
3
  einops==0.3.0
4
  pytorch-lightning==1.4.2
 
1
  opencv-python==4.1.2.30
2
+ torch
3
+ torchvision
4
  omegaconf==2.1.1
5
  einops==0.3.0
6
  pytorch-lightning==1.4.2