Hugging Face's logo

Spaces:
huzey
/
ncut-pytorch
Running on Zero

App Files Community
1
huzey commited on
Commit
54e6424
•
1 Parent(s): 3f7fee9

update gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,7 +2,7 @@ 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/cpu
6
  torch
7
  torchvision
8
  torchaudio
 
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
  torchvision
8
  torchaudio