numpy==1.25.0
xformers==0.0.24
pytorch_lightning
loralib
fairscale
diffusers
transformers
torchvision
accelerate
einops
scipy
timm