用Colab免费部署自己的AI绘画云平台-- Stable Diffusion
看到chen_zhe的博客才了解到最近AI绘画又热了起来。于是有了这一篇博客。
其实我电脑还是能带AI绘画的毕竟是GTX1660ti有9G显存~,不过电脑存储空间不愿花掉10+GB跑,才选择网上亲爱的Google在线笔记本电脑跑。
AI绘画门槛又又又降低了,从最开始需要花半天时间折腾的 Disco-Diffusion ,紧接着 Stable Diffusion 在 github 上开源,各家平台都推出了云平台,让用户通过轻松的点击、选择、输入就能生成一张张AI图。
再到现在!!!使用 webui 在 github 上开源了,不仅有手动教程、docker 教程,还有 Colab 的集成方案。
webui github 地址:webui
注意事项
博主一直用的是特殊网络环境,部分人可能会有问题(包括但不仅限于网页加载不了)
平台搭建
打开链接
连接
等连接上后我们运行第一段脚本,就是查看当前使用的机器。一般是从 K80、T4、P100、V100 中随机分配一个。
设置 Token
点击这个 1 - Setup stage 左边的小箭头进行展开,在1.4 Connect to Google Drive按要求获取Token并设置,记住要勾选✔
勾选 download_if_missing ,然后到 https://huggingface.co/settings/tokens 复制你的 toekn 并填入
设置密码
划到 2 - Run the Stable Diffusion webui 部分,因为要部署一个服务,当然得设置一个密码,不然人人都可以随意使用你的服务了。
运行服务
将他们收起来,依次运行三个步骤
打开 Web 服务
账号为 webui,密码如果设置了就是你设置的密码。
地址在运行中输出的Log中会有。
写在最后
这里需要注意的是,不知道是不是我电脑的分辨率问题,默认宽高为 512 和 256,这个时候会报以下错误,但是将宽度调整成适应自己电脑的就没有问题。
1 | RuntimeError: The size of tensor a (128) must match the size of tensor b (32) at non-singleton dimension 3 |