yhzhai commited on
Commit
976fdd7
1 Parent(s): 38ba8a5

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- torch==1.12.1+cu116
2
- torchvision==0.13.1+cu116
3
  albumentations==1.0.0
4
  einops==0.4.1
5
  fast_pytorch_kmeans==0.1.6
 
1
+ torch==1.12.1
2
+ torchvision==0.13.1
3
  albumentations==1.0.0
4
  einops==0.4.1
5
  fast_pytorch_kmeans==0.1.6