Stable Diffusion XL training and inference as a cog model - View it on GitHub