ncut-pytorch / requirements.txt
huzey's picture
update gpu
b26090c
raw
history blame
377 Bytes
ncut-pytorch
transformers
segment-anything @ git+https://github.com/facebookresearch/segment-anything.git
mobile-sam @ git+https://github.com/ChaoningZhang/MobileSAM.git
--extra-index-url https://download.pytorch.org/whl/cu118
torch
--extra-index-url https://download.pytorch.org/whl/cu118
torchvision
--extra-index-url https://download.pytorch.org/whl/cu118
torchaudio
timm