Stable Diffusion Training Code