diffusers==0.26.3
datasets==2.14.6
accelerate==0.27.2
deepspeed==0.13.2
transformers==4.34.1
torch
numpy
peft
tqdm
torchvision