myniu commited on
Commit
995bdf7
1 Parent(s): 76136e0
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -4,8 +4,8 @@ opencv-python
4
  opencv-python-headless
5
  scikit-image
6
  --extra-index-url https://download.pytorch.org/whl/cu113
7
- torch==1.13.1
8
- torchvision==0.14.1
9
  einops
10
  accelerate==0.30.1
11
  transformers==4.41.1
 
4
  opencv-python-headless
5
  scikit-image
6
  --extra-index-url https://download.pytorch.org/whl/cu113
7
+ torch
8
+ torchvision
9
  einops
10
  accelerate==0.30.1
11
  transformers==4.41.1