diffusers[torch]==0.17.1
transformers==4.30.2
timm==0.4.12 
fairscale==0.4.4
tensorboardX
imgaug
torch==1.11
omegaconf
pytorch-lightning==1.5.0
datasets==2.12.0