torch~=2.2.2
torchvision~=0.17.1
diffusers~=0.28.0
datasets
transformers
tqdm
einops
numpy
wandb
pillow
peft
matplotlib
accelerate
packaging
requests
openai
safetensors
ftfy
regex