matplotlib
diffusers
hydra-core
omegaconf