开源一个基于 stable-diffusion 的 AI 文字生成图片的 QQ 机器人

[listnodeptr]

https://github.com/lxj616/stable-diffusion-qqbot

使用了北京时间今天凌晨 2:00 开源的 stable-diffusion 模型，封装了 docker 运行环境 https://github.com/lxj616/stable-diffusion-docker-service ，走 redis 搞了一个 QQ 机器人然后使用 google 翻译支持中文

当然，直接拿走 docker 环境然后按照原 repo 说明本地跑也是极好的

特地说明一下，里面封装使用了 CompVis/stable-diffusion-safety-checker 来做 NSFW 屏蔽，非常的安全，请大家放心给幼儿园小朋友们使用，千万不要一不小心删除这个检测，不然真的能够生成足够封 QQ 号的邪恶东西

如果大家有足够兴趣，我可以考虑后续建个 QQ 群拉大家直接用嘴画图，不过我 gtx 1080ti 大概 15 秒才能画出一张图，我老妈一个人就能占满一半的算力……我先让亲戚朋友们玩玩再借出来给大家共享 GPU 哈

[israinbow]

有没有能发 NSFW 内容的机器人?

[listnodeptr]

@israinbow 这位同学你不要担心，我们一起努力做好 NSFW 检测共建河蟹社会，你看代码里的 x_checked_image, has_nsfw_concept = safety_checker(images=x_image, clip_input=safety_checker_input.pixel_values) ，之前没有这个检测的话，各种不和谐内容 AI 模型什么都敢乱画，多亏了这一行代码它巩固了我们人民的纯洁心灵，我们要保卫这个检测不被别有用心的人删掉，有这一行在，非常安全

[c0xt30a]

这么好的东西，我居然是第二个 star 的……

[listnodeptr]

@c0xt30a 感谢老铁的关注，双击 666 ，等我找到允许放奇怪图片的图床后 用马云、川普给大家表演杂技，奥利给

[loadingimg]

考虑做个 telegram bot 吗

[listnodeptr]

@loadingimg 如果有条件能够上 TG 的人，直接加入他们官方的 discord 用就可以了，另外如果你指的是你懂得那种 bot ，没有计划做那种 bot 。。。囧

[listnodeptr]

老妈玩差不多了，各种马云扎克伯格英国女王 灵感快耗尽了，我更新 github 加上自己部署的 QQ 群地址了，有兴趣但是没有 GPU 的 V 友们可以进群玩耍了，免费分享不易，个人 GPU 资源紧张，大家稍微悠着点玩哈，本周都可以玩，计划下周一关闭 bot

[suanren]

有意思，跑这个显存要 10GB 以上吗？

[listnodeptr]

@suanren 最低配置可以优化到 6GB 以下（前提删除安全检测模型，所有 batchsize 调成 1, 然后以 512 分辨率不能加），带着安全检测模型需要不到 10GB

[wmsljsgd]

最近也喜欢玩这个 他的官方网站只有 200 个点数 用完需要付费 lz 可以移植到百度飞浆上 白嫖算力 速度快的很 现在上面也有已经移植了的 可以参考下 https://aistudio.baidu.com/aistudio/projectdetail/4459390

[joApioVVx4M4X6Rf]

@listnodeptr 懂了，这就 clone 注释掉自己玩

[listnodeptr]

因为电脑还有其他学习任务，后面不定期 GPU 开放分享，一般在周末白天分享，工作日随缘，睡觉时必定关机

[neofacenew]

我写了一个 TG bot 的 RScipt ，有兴趣可以自取
https://blog.yhong.wang/article/sd_tg_bot

[listnodeptr]

@neofacenew 赞一个，不过友情提示：每次新加载 txt2img 会导致系统反复加载 4GB 的模型从硬盘进内存再进显存，这个过程远超直接 inference 的时间（生成 10 秒钟，硬盘加载 45 秒），所以还是走 redis 香

[neofacenew]

@listnodeptr 我测试了一下，的确不算上和 telegram api 沟通的时间，我本地加载 txt2img 要 28s 左右，不过我不懂得怎么走 redis ，这个是我想到最简单的解决方案了...