KB Teo commited on
Commit
1e25e54
·
1 Parent(s): 2dc6482

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,3 +1,5 @@
 
 
1
  OmegaConf>=2.1.1
2
  albumentations>=1.1.0,<2.0.0
3
  opencv-python>=4.5.3.56
 
1
+ torch==2.1.2
2
+ torchvision==0.16.2
3
  OmegaConf>=2.1.1
4
  albumentations>=1.1.0,<2.0.0
5
  opencv-python>=4.5.3.56