最近 stable diffusion model 大杀四方,而且参数较小给了用户自己玩的可能性 基于 huggingface 的 web app 体验很差,不开放参数选项,不如自己部署一个 web 网页玩一玩
huggingface 的 web app 的源码 bug 多多,不修改的情况下是没办法直接进行部署的,缺少文档。对于想要 5 分钟快速就跑起来是比较困难的,我修改了一下源码使其容易部署,写了一个 5 分钟左右的指南,可以一步步快速部署一个网页,包含所有可调的参数可以交互。
Gihub 源码 https://github.com/Raibows/WebPlayground-of-StableDiffusion
提示,大约需要 11G 显存,正比例于图片的大小。
欢迎大家使用,谢谢大家!
1
davinci21s 2022-10-16 17:41:39 +08:00
11G 显存。。。很多电脑都达不到
|
2
winglight2016 2022-10-16 17:57:30 +08:00
之前我在 colab 上简单试了一下,发现生成的图片很粗糙,lz 的 demo 图片是用了什么参数吗?看起来比较精细一些
|
3
iluolSNS 2022-10-16 18:30:53 +08:00 1
带不动 这就买个 4090[手动狗头]
|
4
Tanix2 2022-10-16 18:56:54 +08:00
已经有很成熟的 webui 了: https://github.com/AUTOMATIC1111/stable-diffusion-webui
|
5
chizuo OP @winglight2016 试试增大 steps ?一般 40 步以上比较好,还有就是图像尺寸要大一点,默认 512x512 往上。scale 在 6-15 会好一点。再有就是检查 stable diffusion model 的 weights version ,我用的是最新的 1.4
|
6
chizuo OP @Tanix2 哈哈哈哈是的,我改完 huggingface 的才发现,已经有很成熟了。我这个比较简单,细节很清楚,适合玩具随便玩一下
|
7
chizuo OP @davinci21s 期待以后的剪枝或者蒸馏版本吧。但是 diffusion 相对 dalle 这些来说已经算是很节省显存了哈哈哈
|
8
winglight2016 2022-10-17 11:25:09 +08:00
|
9
haiyan 2022-10-17 14:45:58 +08:00
4G 都可以跑啊,我的 3070 也是 8G ,为啥 11G
|
10
neptuno 2022-10-17 17:55:13 +08:00
还想在我的辣鸡腾讯云上部署一下,哈哈哈太天真
|
12
chizuo OP @haiyan emmm ,我没有准确衡量过。我这里推理 batch 6 ,step 50 ,512x512 ,显存会稳定在 11-13G ,有时会冲到 30G 几秒...
|
14
ltkun 2022-10-19 08:12:25 +08:00
感谢楼主 在 win10 的 wsl2 下也成功部署了 不过是笔记本 4g 入门级别的显卡 256 分辨率的哈哈 好玩
|
15
qiandongdong 2022-10-27 21:44:04 +08:00
如果要部署在云端什么机器可以支持?一般的 VPS 也不行啊???
|