mfidabel commited on
Commit
dad18ba
1 Parent(s): 6447c95

Fixed Torch Installation

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -5,6 +5,7 @@ jax[cuda11_pip]
5
  jaxlib
6
  git+https://github.com/huggingface/diffusers@main
7
  opencv-python
 
8
  torch
9
  torchvision
10
  git+https://github.com/facebookresearch/segment-anything.git
 
5
  jaxlib
6
  git+https://github.com/huggingface/diffusers@main
7
  opencv-python
8
+ --extra-index-url https://download.pytorch.org/whl/cu113
9
  torch
10
  torchvision
11
  git+https://github.com/facebookresearch/segment-anything.git