huzey commited on
Commit
b26090c
1 Parent(s): 54e6424

update gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -4,7 +4,9 @@ segment-anything @ git+https://github.com/facebookresearch/segment-anything.git
4
  mobile-sam @ git+https://github.com/ChaoningZhang/MobileSAM.git
5
  --extra-index-url https://download.pytorch.org/whl/cu118
6
  torch
 
7
  torchvision
 
8
  torchaudio
9
 
10
  timm
 
4
  mobile-sam @ git+https://github.com/ChaoningZhang/MobileSAM.git
5
  --extra-index-url https://download.pytorch.org/whl/cu118
6
  torch
7
+ --extra-index-url https://download.pytorch.org/whl/cu118
8
  torchvision
9
+ --extra-index-url https://download.pytorch.org/whl/cu118
10
  torchaudio
11
 
12
  timm