Grounded-Segment-Anything / requirements.txt
yizhangliu's picture
Update requirements.txt
b674517 verified
accelerate
addict
diffusers
gradio==3.36.1
huggingface_hub
matplotlib
numpy
onnxruntime
opencv_python
Pillow
pycocotools
PyYAML
requests
setuptools
supervision
termcolor
timm
# torch==2.0.0
# torchvision==0.15.1
torch==2.2.1
torchvision==0.17.1
gevent
yapf
numba
scipy
safetensors
pynvml
sentencepiece
loguru
transformers==4.27.4
# transformers@git+https://github.com/huggingface/transformers.git@main
# lama-cleaner==1.2.4
lama-cleaner@git+https://github.com/yizhangliu/lama-cleaner.git@main
# mmcv==2.1.0
mmengine
openmim==0.3.9
# tiktoken
# einops
# transformers_stream_generator==0.0.4
# tensorboard
# bitsandbytes
# auto-gptq
tencentcloud-sdk-python