accelerate==1.8.1
aesthetic-predictor-v2-5==2024.12.18.1
deepspeed==0.14.4
diffusers @ git+https://github.com/huggingface/diffusers.git@3f3f0c16a6418c7c5505c0a33088fddb5bc90317
image-reward==1.5
lightning==2.5.0.post0
opencv-python==4.11.0.86
opencv-python-headless==4.11.0.86
peft==0.15.0
safetensors==0.5.2
sentencepiece==0.2.0
torch==2.5.1
torchvision==0.20.1
transformers==4.48.3
wandb==0.19.6
xformers==0.0.28.post3