classifier-free-guidance-pytorch>=0.4.0
einops>=0.7
torch>=2.0
