pytorch_lightning
loralib
fairscale
diffusers
transformers
torchvision
accelerate
einops
scipy