torch==2.5.1 torchvision==0.20.1 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 hf-xet==1.1.8