分享下三四个近期搞的基于 GPT2 的应用
資深大佬 : mymusise 14
最近整理了这半年弄的一些东西,主要围绕 OPENAI 的 GPT2 模型做了一些应用。
1. AIXX:一款基于 AI 的文字冒险游戏
- 在线体验: https://aixx.mymusise.com
- 项目地址: https://github.com/mymusise/AI-XIUXIAN
游戏原型参考了国外 19 年发布的 AIDungeon,20 年 9 月才发现,当时就被惊艳到了,感觉这才是以后游戏 NPC 发展趋势,而不是每次跟 NPC 对话都只会跟你抱怨他膝盖中了一箭的故事。
然后买了几块显卡回来,用中文语料重头训练了一个中文武侠类型的文字游戏。效果虽然跟设想有点出入,但是作为游戏应该还是可以给大家带来欢乐的。
2. vscode 上的故事续写插件
- 项目地址: https://github.com/mymusise/vscode-writer-assistant
- 下载地址: https://marketplace.visualstudio.com/items?itemName=mymusise.writer-assistant
基于上面训练出来的模型,写了个 vscode 的插件,有一说一,虽然 vscode 的文档非常友好,但限于本人 JS 太菜,插件写得比较作急,Js 大佬轻拍。Ps:欢迎有兴趣的 Js 大佬一起来完善。
3. 整理了一个超容易上手用来训练 GPT2 项目代码
- 项目地址: https://github.com/mymusise/gpt2-quickly
没啥好说的,就很好上手,如果对训练 GPT2 等预训练模型有兴趣的童鞋可以试试。
4. 在线体验号称最大的中文与训练模型:CPM
- 项目地址: https://github.com/mymusise/CPM-TF2Transformer
- 在线体验: https://colab.research.google.com/github/mymusise/CPM-TF2Transformer/blob/main/demo-fp16.ipynb
在弄 AIXX 的过程突然发现清华的大佬开源了个最大中文与训练模型,因为模型很大,对机器要求会比较高。 于是借用 colab 提供的计算能力和 huggingface 提供的存储,转了个在线版本,一起薅资本主义的羊毛呀~
今天元宵节,祝大家团团圆圆~
大佬有話說 (2)