Has anyone tried training a diffusion model for speed ?