BiRefNet-legacy / requirements.txt
ZhengPeng7's picture
Add inference endpoint feature.
ca16a0e
raw
history blame contribute delete
271 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1
--extra-index-url https://download.pytorch.org/whl/cu118
torchvision==0.15.2
numpy<2
opencv-python
timm
scipy
scikit-image
kornia
einops
tqdm
prettytable
transformers
huggingface-hub>0.25
accelerate