huzey commited on
Commit
e1001b9
1 Parent(s): caaf478

update gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -6
requirements.txt CHANGED
@@ -1,12 +1,8 @@
 
 
1
  ncut-pytorch
2
  transformers
3
  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
- --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
 
1
+ torch
2
+ torchvision
3
  ncut-pytorch
4
  transformers
5
  segment-anything @ git+https://github.com/facebookresearch/segment-anything.git
6
  mobile-sam @ git+https://github.com/ChaoningZhang/MobileSAM.git
 
 
 
 
 
 
7
 
8
  timm