demo_space1 / requirements.txt
zen-vton's picture
Update requirements.txt
e3ecaaa verified
raw
history blame contribute delete
686 Bytes
# Core dependencies for IDM-VTON Virtual Try-On
gradio==4.44.1
spaces>=0.18.0
av
onnxruntime-gpu
# AI/ML Core
torch>=2.0.0
torchvision
transformers==4.36.2
diffusers==0.25.0
accelerate==0.26.1
safetensors==0.4.1
# Hugging Face
huggingface_hub==0.20.3
tokenizers==0.15.0
# Image Processing
pillow==9.4.0
opencv-python-headless==4.7.0.72
scikit-image==0.21.0
numpy==1.24.4
# Utilities
tqdm==4.64.1
einops==0.7.0
omegaconf==2.3.0
regex==2023.12.25
# Detectron2 + DensePose Required
fvcore==0.1.5.post20221221
pycocotools==2.0.7
cloudpickle==2.2.1
# Optional if DensePose fails with image loading
scipy==1.10.1
matplotlib==3.7.4
git+https://github.com/facebookresearch/detectron2.git