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