requirements.txt

torch==2.3
torchvision==0.18.0
pytorch-lightning==2.2
datasets==2.18
transformers==4.38.2
diffusers
hpsv2==1.2
hpsv2x==1.2.0
image-reward==1.5
open-clip-torch==2.24
blobfile
openai-clip
setuptools==60.2
optimum
xformers