myniu commited on
Commit
799c206
1 Parent(s): 9dc30ee
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -3,8 +3,8 @@ gradio==4.5.0
3
  opencv-python
4
  opencv-python-headless
5
  scikit-image
6
- torch==1.13.0
7
- torchvision==0.14.0
8
  einops
9
  accelerate==0.30.1
10
  transformers==4.41.1
 
3
  opencv-python
4
  opencv-python-headless
5
  scikit-image
6
+ torch==2.0.1
7
+ torchvision==0.15.2
8
  einops
9
  accelerate==0.30.1
10
  transformers==4.41.1