classifier-free-guidance-pytorch>=0.0.15
einops>=0.6
torch>=1.6
