Ming Li commited on
Commit
43363f2
1 Parent(s): af5481e

update torch version

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -8,8 +8,8 @@ huggingface-hub==0.21.2
8
  mediapipe==0.10.1
9
  opencv-python-headless==4.8.0.74
10
  safetensors==0.4.1
11
- torch==2.0.1
12
- torchvision==0.15.2
13
  transformers==4.38.1
14
  xformers==0.0.20
15
  kornia==0.7.0
 
8
  mediapipe==0.10.1
9
  opencv-python-headless==4.8.0.74
10
  safetensors==0.4.1
11
+ torch==2.2.2
12
+ torchvision==0.17.2
13
  transformers==4.38.1
14
  xformers==0.0.20
15
  kornia==0.7.0