一个简单部署的 Stable Diffusion Model，体验自由图像生成

[chizuo]

大家好

最近 stable diffusion model 大杀四方，而且参数较小给了用户自己玩的可能性 基于 huggingface 的 web app 体验很差，不开放参数选项，不如自己部署一个 web 网页玩一玩

为什么要修改源码

huggingface 的 web app 的源码 bug 多多，不修改的情况下是没办法直接进行部署的，缺少文档。对于想要 5 分钟快速就跑起来是比较困难的，我修改了一下源码使其容易部署，写了一个 5 分钟左右的指南，可以一步步快速部署一个网页，包含所有可调的参数可以交互。

示例

Gihub 源码 https://github.com/Raibows/WebPlayground-of-StableDiffusion

提示，大约需要 11G 显存，正比例于图片的大小。

欢迎大家使用，谢谢大家！

[davinci21s]

11G 显存。。。很多电脑都达不到

[winglight2016]

之前我在 colab 上简单试了一下，发现生成的图片很粗糙，lz 的 demo 图片是用了什么参数吗？看起来比较精细一些

[iluolSNS]

带不动 这就买个 4090[手动狗头]

[Tanix2]

已经有很成熟的 webui 了: https://github.com/AUTOMATIC1111/stable-diffusion-webui

[chizuo]

@winglight2016 试试增大 steps ？一般 40 步以上比较好，还有就是图像尺寸要大一点，默认 512x512 往上。scale 在 6-15 会好一点。再有就是检查 stable diffusion model 的 weights version ，我用的是最新的 1.4

[chizuo]

@Tanix2 哈哈哈哈是的，我改完 huggingface 的才发现，已经有很成熟了。我这个比较简单，细节很清楚，适合玩具随便玩一下

[chizuo]

@davinci21s 期待以后的剪枝或者蒸馏版本吧。但是 diffusion 相对 dalle 这些来说已经算是很节省显存了哈哈哈

[winglight2016]

@davinci21s
@iluolSNS
colab 上可以免费跑，显存肯定够用了
@chizuo
好的，谢谢，我抽空再试试

[haiyan]

4G 都可以跑啊，我的 3070 也是 8G ，为啥 11G

[neptuno]

还想在我的辣鸡腾讯云上部署一下，哈哈哈太天真

[chizuo]

@neptuno 哈哈哈，可以试一试 cpu mode ，也许 8G 内存可以？

[chizuo]

@haiyan emmm ，我没有准确衡量过。我这里推理 batch 6 ，step 50 ，512x512 ，显存会稳定在 11-13G ，有时会冲到 30G 几秒...

[neptuno]

@chizuo #11 256m 请求一战 hahah ，后面自己台式上面试一下

[ltkun]

感谢楼主 在 win10 的 wsl2 下也成功部署了 不过是笔记本 4g 入门级别的显卡 256 分辨率的哈哈 好玩

[qiandongdong]

如果要部署在云端什么机器可以支持？一般的 VPS 也不行啊？？？