欢迎光临散文网 会员登陆 & 注册

stable-diffusion-xl-base-1.0 模型推理

2023-08-12 15:02 作者:AI日日新  | 我要投稿

可在线运行的notebook示例:在kaggle网站搜索 stable-diffusion-xl-base-1.0


首先需要安装依赖包:

接着加载模型:(注意这里设置16精度加载模型可以节省显存,同时注意设置模型为gpu的cuda加速加快模型推理)

然后设置文本指令词并生成你想要的图片,最后保存图片:

若你使用notebook编程环境,可以在notebook中查看图片:


stable-diffusion-xl-base-1.0 模型推理的评论 (共 条)

分享到微博请遵守国家法律