开源一个基于 stable-diffusion 的 AI 文字生成图片的 QQ 机器人 - V2EX
V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
listnodeptr
V2EX    Stable Diffusion

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

  •  1
     
  •   listnodeptr 2022-08-23 21:21:57 +08:00 4901 次点击
    这是一个创建于 1145 天前的主题,其中的信息可能已经有所发展或是发生改变。

    img

    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 哈

    15 条回复    2022-09-01 15:13:00 +08:00
    israinbow
        1
    israinbow  
       2022-08-24 01:00:37 +08:00
    有没有能发 NSFW 内容的机器人?
    listnodeptr
        2
    listnodeptr  
    OP
       2022-08-24 02:12:46 +08:00   6
    @israinbow 这位同学你不要担心,我们一起努力做好 NSFW 检测共建河蟹社会,你看代码里的 x_checked_image, has_nsfw_cOncept= safety_checker(images=x_image, clip_input=safety_checker_input.pixel_values) ,之前没有这个检测的话,各种不和谐内容 AI 模型什么都敢乱画,多亏了这一行代码它巩固了我们人民的纯洁心灵,我们要保卫这个检测不被别有用心的人删掉,有这一行在,非常安全
    c0xt30a
        3
    c0xt30a  
       2022-08-24 03:27:19 +08:00
    这么好的东西,我居然是第二个 star 的……
    listnodeptr
        4
    listnodeptr  
    OP
       2022-08-24 11:47:05 +08:00
    @c0xt30a 感谢老铁的关注,双击 666 ,等我找到允许放奇怪图片的图床后 用马云、川普给大家表演杂技,奥利给
    loadingimg
        5
    loadingimg  
       2022-08-25 10:59:16 +08:00
    考虑做个 telegram bot 吗
    listnodeptr
        6
    listnodeptr  
    OP
       2022-08-25 11:33:01 +08:00
    @loadingimg 如果有条件能够上 TG 的人,直接加入他们官方的 discord 用就可以了,另外如果你指的是你懂得那种 bot ,没有计划做那种 bot 。。。
    listnodeptr
        7
    listnodeptr  
    OP
       2022-08-25 12:48:45 +08:00
    老妈玩差不多了,各种马云扎克伯格英国女王 灵感快耗尽了,我更新 github 加上自己部署的 QQ 群地址了,有兴趣但是没有 GPU 的 V 友们可以进群玩耍了,免费分享不易,个人 GPU 资源紧张,大家稍微悠着点玩哈,本周都可以玩,计划下周一关闭 bot
    suanren
        8
    suanren  
       2022-08-26 08:15:55 +08:00
    有意思,跑这个显存要 10GB 以上吗?
    listnodeptr
        9
    listnodeptr  
    OP
       2022-08-26 12:28:58 +08:00
    @suanren 最低配置可以优化到 6GB 以下(前提删除安全检测模型,所有 batchsize 调成 1, 然后以 512 分辨率不能加),带着安全检测模型需要不到 10GB
    wmsljsgd
        10
    wmsljsgd  
       2022-08-26 16:42:57 +08:00
    最近也喜欢玩这个 他的官方网站只有 200 个点数 用完需要付费 lz 可以移植到百度飞浆上 白嫖算力 速度快的很 现在上面也有已经移植了的 可以参考下 https://aistudio.baidu.com/aistudio/projectdetail/4459390
    joApioVVx4M4X6Rf
    nbsp;   11
    joApioVVx4M4X6Rf  
       2022-08-26 19:06:34 +08:00
    @listnodeptr 懂了,这就 clone 注释掉自己玩
    listnodeptr
        12
    listnodeptr  
    OP
       2022-08-30 14:29:21 +08:00
    因为电脑还有其他学习任务,后面不定期 GPU 开放分享,一般在周末白天分享,工作日随缘,睡觉时必定关机
    neofacenew
        13
    neofacenew  
       2022-08-30 15:07:43 +08:00   1
    我写了一个 TG bot 的 RScipt ,有兴趣可以自取
    https://blog.yhong.wang/article/sd_tg_bot
    listnodeptr
        14
    listnodeptr  
    OP
       2022-08-31 20:28:52 +08:00
    @neofacenew 赞一个,不过友情提示:每次新加载 txt2img 会导致系统反复加载 4GB 的模型从硬盘进内存再进显存,这个过程远超直接 inference 的时间(生成 10 秒钟,硬盘加载 45 秒),所以还是走 redis 香
    neofacenew
        15
    neofacenew  
       2022-09-01 15:13:00 +08:00
    @listnodeptr 我测试了一下,的确不算上和 telegram api 沟通的时间,我本地加载 txt2img 要 28s 左右,不过我不懂得怎么走 redis ,这个是我想到最简单的解决方案了...
    关于     帮助文档     自助推广系统     博客     API     FAQ     Solana     881 人在线   最高记录 6679       Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 26ms UTC 21:48 PVG 05:48 LAX 14:48 JFK 17:48
    Do have faith in what you're doing.
    ubao snddm index pchome yahoo rakuten mypaper meadowduck bidyahoo youbao zxmzxm asda bnvcg cvbfg dfscv mmhjk xxddc yybgb zznbn ccubao uaitu acv GXCV ET GDG YH FG BCVB FJFH CBRE CBC GDG ET54 WRWR RWER WREW WRWER RWER SDG EW SF DSFSF fbbs ubao fhd dfg ewr dg df ewwr ewwr et ruyut utut dfg fgd gdfgt etg dfgt dfgd ert4 gd fgg wr 235 wer3 we vsdf sdf gdf ert xcv sdf rwer hfd dfg cvb rwf afb dfh jgh bmn lgh rty gfds cxv xcv xcs vdas fdf fgd cv sdf tert sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf sdf shasha9178 shasha9178 shasha9178 shasha9178 shasha9178 liflif2 liflif2 liflif2 liflif2 liflif2 liblib3 liblib3 liblib3 liblib3 liblib3 zhazha444 zhazha444 zhazha444 zhazha444 zhazha444 dende5 dende denden denden2 denden21 fenfen9 fenf619 fen619 fenfe9 fe619 sdf sdf sdf sdf sdf zhazh90 zhazh0 zhaa50 zha90 zh590 zho zhoz zhozh zhozho zhozho2 lislis lls95 lili95 lils5 liss9 sdf0ty987 sdft876 sdft9876 sdf09876 sd0t9876 sdf0ty98 sdf0976 sdf0ty986 sdf0ty96 sdf0t76 sdf0876 df0ty98 sf0t876 sd0ty76 sdy76 sdf76 sdf0t76 sdf0ty9 sdf0ty98 sdf0ty987 sdf0ty98 sdf6676 sdf876 sd876 sd876 sdf6 sdf6 sdf9876 sdf0t sdf06 sdf0ty9776 sdf0ty9776 sdf0ty76 sdf8876 sdf0t sd6 sdf06 s688876 sd688 sdf86