ZhengChong commited on
Commit
5ff3003
1 Parent(s): 498fb67

chore: Update torch and torchvision versions in requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -11,8 +11,8 @@ PyYAML==6.0.1
11
  scipy==1.13.1
12
  setuptools==51.0.0
13
  scikit-image==0.24.0
14
- pytorch==2.0.0
15
- torchvision==0.15.0
16
  tqdm==4.66.4
17
  transformers==4.27.3
18
  xformers==0.0.23.post1
 
11
  scipy==1.13.1
12
  setuptools==51.0.0
13
  scikit-image==0.24.0
14
+ torch==2.0.1
15
+ torchvision==0.15.2
16
  tqdm==4.66.4
17
  transformers==4.27.3
18
  xformers==0.0.23.post1