Stable Diffusion code implementation