Now tweak it for stable diffusion 1.5