利用GPT-2制作聊天机器人
-
GPT-2
是一种深度学习领域的语言模型,可以用于完成自然语言相关的生成任务。最近在GitHub上找到了一个项目,该项目使用
GPT-2
实现了在聊天过程中的短对话生成,并提供了预训练模型https://github.com/yangjianxin1/GPT2-chitchat
效果
我将其集成到了聊天机器人内,使用仓库内的预训练模型,群内使用效果如下
虽然看起来聊天还算比较连贯,
但其实很多时候不是嘴臭就是智障,也算是比较有趣了可以来我的群内实际体验一下
206073050
适配
实际上并不需要很复杂的修改代码,只需要把他的历史记录单独按照每一个群成员记录即可。然后再使用
flask
启动一个http服务器
为机器人进程服务即可性能
系统:Windows server 2012r2 64位
CPU:E5-2670v2 8核虚拟机
内存:8GB
在Python3.9下使用CPU推理,单次推理速度约700ms
,可以满足负载不高的群内聊天推理,实际内存占用约为1.6GB
-
坐等小白化,话说我这小水管带不动啊
-
@未来のミライ 已经是很小白化了,就改几行Python代码就行,模型已经有大佬训练好了
-
@rainchan 不会python怎么办QAQ
-
请问能不能分享下修改后的文件?
-
坐等小白花
-
大佬能不能出一篇教程呢?
-
呜呜呜,浇浇我(蹲教程)
-
这个我有做,不过是用的gocqhttp
当时还不知道有mirai
项目的地址https://gitee.com/october_Wr/qq-chat-robot
有打包好的exe版 -
linux能用吗?
-
-