Diffusion Model Arxiv Api Python