开源一个基于 stable-diffusion 的 AI 文字生成图片的 QQ 机器人
listnodeptr · 2022-08-23 21:21:57 +08:00 · 4333 次点击这是一个创建于 798 天前的主题，其中的信息可能已经有所发展或是发生改变。
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 哈
 |
1
israinbow 2022-08-24 01:00:37 +08:00
有没有能发 NSFW 内容的机器人?
|
 |
2
listnodeptr OP  6
@israinbow 这位同学你不要担心，我们一起努力做好 NSFW 检测共建河蟹社会，你看代码里的 x_checked_image, has_nsfw_concept = safety_checker(images=x_image, clip_input=safety_checker_input.pixel_values) ，之前没有这个检测的话，各种不和谐内容 AI 模型什么都敢乱画，多亏了这一行代码它巩固了我们人民的纯洁心灵，我们要保卫这个检测不被别有用心的人删掉，有这一行在，非常安全
|
 |
3
c0xt30a 2022-08-24 03:27:19 +08:00
这么好的东西，我居然是第二个 star 的……
|
|
4
listnodeptr OP @c0xt30a 感谢老铁的关注，双击 666 ，等我找到允许放奇怪图片的图床后 用马云、川普给大家表演杂技，奥利给
|
 |
5
loadingimg 2022-08-25 10:59:16 +08:00
考虑做个 telegram bot 吗
|
|
6
listnodeptr OP @loadingimg 如果有条件能够上 TG 的人，直接加入他们官方的 discord 用就可以了，另外如果你指的是你懂得那种 bot ，没有计划做那种 bot 。。。囧
|
|
7
listnodeptr OP 老妈玩差不多了，各种马云扎克伯格英国女王 灵感快耗尽了，我更新 github 加上自己部署的 QQ 群地址了，有兴趣但是没有 GPU 的 V 友们可以进群玩耍了，免费分享不易，个人 GPU 资源紧张，大家稍微悠着点玩哈，本周都可以玩，计划下周一关闭 bot
|
 |
8
suanren 2022-08-26 08:15:55 +08:00
有意思，跑这个显存要 10GB 以上吗？
|
|
9
listnodeptr OP @suanren 最低配置可以优化到 6GB 以下（前提删除安全检测模型，所有 batchsize 调成 1, 然后以 512 分辨率不能加），带着安全检测模型需要不到 10GB
|
 |
10
wmsljsgd 2022-08-26 16:42:57 +08:00
最近也喜欢玩这个 他的官方网站只有 200 个点数 用完需要付费 lz 可以移植到百度飞浆上 白嫖算力 速度快的很 现在上面也有已经移植了的 可以参考下 https://aistudio.baidu.com/aistudio/projectdetail/4459390
|
 |
11
joApioVVx4M4X6Rf 2022-08-26 19:06:34 +08:00
@listnodeptr 懂了，这就 clone 注释掉自己玩
|
|
12
listnodeptr OP 因为电脑还有其他学习任务，后面不定期 GPU 开放分享，一般在周末白天分享，工作日随缘，睡觉时必定关机
|
 |
13
neofacenew 2022-08-30 15:07:43 +08:00 1
我写了一个 TG bot 的 RScipt ，有兴趣可以自取
https://blog.yhong.wang/article/sd_tg_bot |
|
14
listnodeptr OP @neofacenew 赞一个，不过友情提示：每次新加载 txt2img 会导致系统反复加载 4GB 的模型从硬盘进内存再进显存，这个过程远超直接 inference 的时间（生成 10 秒钟，硬盘加载 45 秒），所以还是走 redis 香
|
|
15
neofacenew 2022-09-01 15:13:00 +08:00
@listnodeptr 我测试了一下，的确不算上和 telegram api 沟通的时间，我本地加载 txt2img 要 28s 左右，不过我不懂得怎么走 redis ，这个是我想到最简单的解决方案了...
|
Select Language