accelerate==0.23.0
av2
clip
datasets
diffusers==0.28.0
einops
ftfy
gradio==4.38.1
hpsv2
image-reward
imageio-ffmpeg
imgaug
numpy==1.26.2
omegaconf
open-clip-torch
opencv-python
peft
Pillow==9.5.0
scikit-image
scikit-learn
scipy
tensorboard
torch
torchaudio
torchmetrics
torchvision
tqdm
transformers==4.32.1