OmegaConf
pytorch_lightning
accelerate
diffusers[torch]
transformers
scipy
tensorboard