How to correctly train your Stable Diffusion model