torch==2.3.1
torchvision==0.18.1
diffusers==0.31.0
transformers==4.42.0
accelerate==0.30.1
einops==0.7.0
tqdm==4.66.4
Pillow==10.4.0
opencv-python==4.10.0.84
PyYAML==6.0.2
wandb==0.17.5
numpy==1.26.4
huggingface-hub==0.23.5
git+https://github.com/tgxs002/HPSv2.git@main
