myniu commited on
Commit
76136e0
1 Parent(s): 095ec88
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -3,6 +3,7 @@ gradio==4.5.0
3
  opencv-python
4
  opencv-python-headless
5
  scikit-image
 
6
  torch==1.13.1
7
  torchvision==0.14.1
8
  einops
 
3
  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