本文主要介绍如何使用 Amazon SageMaker机器学习服务轻松托管 Stable Diffusion WebUI,一键部署开箱即用的AIGC图片方向轻量级应用。通过Amazon CloudFormation基础设施即代码的服务,实现底层环境、AI 模型和前端 Stable Diffusion WebUI 的快速部署,帮助用户在15-20分钟内部署一套文生图、图生图的AI应用。此方案适合企业级客户对AIGC图片方向做前期调研和快速验证、小型团队快速搭建轻量级 AI 应用的业务场景。
一、方案优势
本方案使用CloudFormation一键部署,仅需点击几个步骤,即可快速在Amaozn账户中成功部署一套AIGC图片方向的web应用环境,交互界面简洁友好,帮助用户快速体验文本生成图片、图片生成图片等AI服务。
底层使用Amaozn SageMaker Notebook Instance托管的Jupyter笔记本实例,用户无需关心基础环境(如存储、网络等)的构建及底层基础设施运维。
方案采用全托管的Stable Diffusion AI模型服务(1.5轻量版本,包含Stable Diffusio WedUI与InvokeAI两个界面),具备良好的开源项目使用体验,并支持根据用户的需要安装插件以拓展使用场景(如ControlNet)。
用户可使用自带的图片数据微调模型,产出的模型可基于自动化流水线工具快速部署上线至推理节点,方便规模化的图片推理调用。
本方案完全开源,用户可以在SageMaker Notebook中对模型和拓展定制开发,满足自身业务需求。
二、方案架构
三、方案部署
1.登录控制台,在控制台页面右上角切换区域至us-east-1。
2.在Service搜索框中输入Cloudformation,点击进入。
3.创建堆栈、使用新资源(标准)、上传模板文件,模板在以下链接下载:
https://xiekl.s3.cn-northwest-1.amazonaws.com.cn/sd-webui.yml
4.填入堆栈名称,Instance Type建议选择g5.2xlarge,磁盘Volume大小默认为100G,选择WedUI Type,默认是auto代表Stable Diffusion WedUI,WedUI Version选择0316,点击“下一步”。
5.设置保持默认,点击“下一步”。
6.确认配置(维持默认即可),勾选“我确定……”,点击“提交”,等待15-20分钟左右,即可部署完成。