[Share Experiences] 体验RWKV的写小说功能
Tofloor
poster avatar
川顺页
deepin
2023-04-22 20:36
Author

开源主页:https://github.com/BlinkDL/ChatRWKV

模型下载:https://huggingface.co/BlinkDL/rwkv-4-novel/tree/main

选择的模型:``RWKV-4-Novel-7B-v1-Chn-20230409-ctx4096.pth`

简单使用方法:装好cuda 11.4/11.8, pytorch 2.0, 改chat.py, CHAT_LANG = 'Chinese'args.MODEL_NAME = "model/RWKV-4-Novel-7B-v1-Chn-20230409-ctx4096" 运行chat.py即可

下面是生成效果:基本文笔还行,但是逻辑和我的设定是有冲突的,后面可能要加入UIE识别实体,关系,抽取知识图谱,然后通过实体对齐核查语言逻辑,让chatGLM或者chatGPT来修改润色,剩下的继续给RWKV续写,循环下来就可以实现AI教AI写作。

体验RWKV-Novel.png

Reply Favorite View the author
All Replies
川顺页
deepin
2023-04-22 20:38
#1

补充 args.ctx_len = 4096

Reply View the author