
如何 5 分钟跑通 AI-WriterAI 写作完整上手指南【免费下载链接】AI-WriterAI 写小说生成玄幻和言情网文等等。中文预训练生成模型。采用我的 RWKV 模型类似 GPT-2 。AI写作。RWKV for Chinese novel generation.项目地址: https://gitcode.com/gh_mirrors/ai/AI-WriterAI-Writer 是一个中文 AI 写作项目用 RWKV 模型比 GPT 更省算力的语言模型给你写的开头自动续写玄幻、言情网文。适合第一次接触文本生成、想立刻看到效果的开发者。跑起来之前环境与依赖先拉代码。项目里没有 requirements.txt依赖要手动装Python 3.8 PyTorch 1.9 以上 numpygit clone https://gitcode.com/gh_mirrors/ai/AI-Writer cd AI-Writer pip install torch numpy按显卡情况补一条N 卡还需要 CUDA 11.1 cuDNN 和 CUDA 版 torch1.9.1cu111 即可AMD/Intel 卡和网页版另装pip install onnxruntime-directml # AMD/Intel 卡才需要 pip install websockets # 想开网页界面才需要还差最后一块模型权重不在仓库里去项目 Release 下载对应模型包解压到model/目录齐活。项目结构导航装完依赖打开项目根目录先花 30 秒认一遍每个文件夹路径一句话说明run.py命令行主入口设备、模型、采样参数、开头全在这一堆变量里server.py网页界面后端起一个监听 8266 端口的 WebSocket常驻连接服务web-client/网页版前端Chrome 打开 index.html 就能写src/model.py 是 RWKV 结构utils.py 是 top-p 采样逻辑model/模型权重和词表文件放这里new/更新一版的代码和词表有空再研究模型训练数据全部来自网文、缺乏生活常识生成的内容仅供娱乐别拿它当事实依据。文件夹认全了接着打开run.py按执行顺序把生成一个字这件事拆开看。启动流程拆解双击run.bat或执行python run.py内部依次发生这 6 件事读RUN_DEVICE决定用 GPU、DML 还是 CPU 跑读词表.json建立字 ↔ 编号的双向映射加载.pth权重顺手把时序衰减矩阵预计算好把context开头切成编号只保留最后 512 个字模型前向一次得到下一个字的概率分布按top_p采样出 1 个字接回开头继续写直到写满 512 字整条链路就是改run.py→ 读词表 → 载权重 → 逐字采样 → 控制台滚动输出。这就是 AI-Writer 快速上手要记的全部流程跑完一遍你基本就能读懂剩下 200 行代码。配置项逐个讲AI-Writer 怎么配置没有 ini 文件就是run.py头部那一排变量改完重跑一次即生效参数改什么改完会怎样RUN_DEVICEgpu/dml/cpu没显卡改cpu照样跑只是慢A/I 卡用dml并换对应模型MODEL_NAME、WORD_NAME换成言情模型的路径两个必须一致且模型要解压在model/下NUM_OF_RUNS/LENGTH_OF_EACH默认 999 遍 × 512 字控制写多少遍、每遍多少字一次写太长容易死循环top_p/top_p_newline默认 0.75 / 0.9调到 1.0 文风放飞调到 0.5 更规矩后者专管换行后那个字context换成你自己的开头开头质量决定续写质量乱写开头就乱写续文模型只看最后 512 字改完记得重跑不然不生效。下一步你可以怎么玩参数玩得熟了可以往这三个方向走开网页版执行python server.py或双击server.bat再用 Chrome 打开 web-client/index.html终端续写就变成浏览器里写小说。固定随机数取消run.py里src.utils.set_seed(42)那行的注释每次运行输出完全一致方便对比参数效果。换口味把MODEL_NAME换成言情模型context写个带情绪的开头同一套代码两种文风效果对比一目了然。【免费下载链接】AI-WriterAI 写小说生成玄幻和言情网文等等。中文预训练生成模型。采用我的 RWKV 模型类似 GPT-2 。AI写作。RWKV for Chinese novel generation.项目地址: https://gitcode.com/gh_mirrors/ai/AI-Writer创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考