wd-tagger-heatmap / requirements.txt
neggles's picture
add pyproject and reqs
83c4fa5
raw
history blame contribute delete
No virus
305 Bytes
colorcet
diffusers
gradio >=4.25.0, < 5.0.0
hf-transfer
huggingface-hub >= 0.14.0
matplotlib
numpy >= 1.23.5
opencv-contrib-python-headless
pandas >= 2.0.0
Pillow >= 9.5.0
rich
timm @ git+https://github.com/huggingface/pytorch-image-models@main#egg=timm
tokenizers
torch >= 2.1.0
torchvision
transformers