可在线运行的notebook示例:在kaggle网站搜索 stable-diffusion-xl-base-1.0
首先需要安装依赖包:
接着加载模型:(注意这里设置16精度加载模型可以节省显存,同时注意设置模型为gpu的cuda加速加快模型推理)
然后设置文本指令词并生成你想要的图片,最后保存图片:
若你使用notebook编程环境,可以在notebook中查看图片: