Diffusion Models Class | Awesome Repository