numpy
einops
omegaconf
matplotlib
accelerate>=0.22.0
diffusers>=0.20.1
torch>=2.1.0
torchdiffeq>=0.2.3
diffusers==0.26.3
xformers