1、简述
AI-Writer 是一个开源的中文「AI 写小说 / 文本生成」工具,由 BlinkDL 发布。它使用一个基于 RNN/语言模型 (具体为 RWKV‑LM) 的预训练模型 — 类似 GPT-2,但结构有所不同 — 专注于生成中文网文 (如玄幻、言情、网络小说) 。
与传统手写小说不同,AI-Writer 将“根据开头 + 一些设定 → 自动续写/生成后续内容”的流程自动化,使你能够快速生成连贯 (或半连贯) 的小说段落 / 章节。对于学习创作、娱乐写作、灵感草稿、尝试AI写作的人来说,它是一个方便、有趣、开源、可本地部署的候选。
不过项目本身也声明,它训练的数据主要来自网络小说 (“网文”),缺乏现实生活常识 — 所以生成内容更适合娱乐与灵感,而不一定严谨或现实。

2、安装 与 配置 — 快速上手
下面是一个典型的「从零开始」安装 + 运行流程 (以 Python 运行版本为主):
克隆项目
git clone https://github.com/BlinkDL/AI-Writer.git
cd AI-Writer
准备模型
🔥 在 GitHub Releases 中下载合适的模型 (如 N卡 GPU 版 / CPU 版)。
🔥 解压模型,并将模型文件放入 model/ 目录 (或 README/说明中指定的目录)
安装依赖
🔥 建议使用 Python 3.8.x。官方 README 提到对 Python 版本有要求。
🔥 安装依赖库 (如 torch 等 — 如果你打算用 GPU,则需要安装带 CUDA 的版本)
🔥 若仅用 CPU,也可以安装普通版本 (但速度较慢)
运行程序
在命令行里运行 (以文本生成模式):
python run.py
或者如果你希望通过网页界面 (web UI) 使用,运行:
python server.py
然后使用浏览器打开项目提供的客户端 (通常 README 中会说明)

输入开头 / 提示 + 参数
🔥 打开后,在界面 (或命令行) 中输入文章 / 小说的开头 (几句话或者一个段落)
🔥 设置续写长度、采样参数 (如 temperature / 随机度 /生成字数)
🔥 执行续写 → 等待 AI 输出生成内容
检查 / 编辑 / 导出
🔥 AI 自动生成后,你可以手动编辑、删改、组合输出内容
🔥 如果满意,可将其保存 / 导出为纯文本 / Markdown / 其他格式 (你也可以自己写脚本做进一步处理)
注意:如果你机器没有 GPU、或模型较大,生成速度可能比较慢;另外,生成质量与模型版本 (大小)、采样参数 (temperature / 随机度) 等密切相关。
3、实践样例
假设你想用 AI-Writer 来生成一个短篇小说 (科幻 / 玄幻 / 言情都可以)。下面给出一个“从输入到输出”的简化流程示例 (伪代码 + 操作步骤):
场景:生成一个“玄幻”开头 + 自动续写 ~ 500 字
🔥 在项目根目录,确保模型已经放在 model/ 下,依赖已安装。
🔥 运行 python run.py。
🔥 当提示输入文本时,输入例如:
“夜色如墨,寒风撕裂森林深处的静谧。血月高挂天际,幽蓝的月光洒在古老的祭坛上,四周雕刻的符文微微发光。林萧抬头,目光中透出坚定——”
🔥 设置生成参数,比如 “生成 500 字”,“随机度 0.8 (或 0.9)” — (randomness / temperature 较高时,更具创意,但逻辑可能松散)
🔥 等待生成 — 你将得到 AI 自动续写的一段 ~500 字故事 (新的描述 /剧情 /对白)
🔥 查看输出,手动挑选满意部分,做必要修改 /润色 /重组。
🔥 如果需要多段连贯章节,可以把上一步的输出作为下一次输入 (“续写”),继续生成 — 直到得到满意的篇幅 /节奏 /剧情。
输出示例 (伪):
“林萧脚步微颤,却未退缩。他攥紧拳头,幽蓝符文映出他苍白的脸庞。突然——一阵惨烈的嘶吼划破夜空,古树裂开,一只…(以下省略)…”
通过几次“续写 + 编辑 + 组合”,你可能拼出一个完整短篇大纲 /若干章节雏形。
4、应用场景与适合人群
🔥 快速生成小说 / 网文草稿 — 如果你只是想快速搭建故事骨架、获得灵感、生成开头或片段,AI-Writer 足够。
🔥 写作练习 / 灵感来源 — 对于缺乏灵感或想尝试不一样风格的人,它能提供多种可能。
🔥 娱乐 / 玩梗 /轻松创作 — 因为其训练数据主要是网文,带有一定 “网文味” + 娱乐性 (玄幻、言情、夸张…) 非常适合喜欢轻松、夸张文学风格的人。
🔥 学习语言模型 /文本生成 /机器学习 — 如果你对 NLP、语言模型、生成模型原理感兴趣,可以通过研究 AI-Writer 的代码 +模型 (RWKV-LM) 来学习基础原理与实践。
限制 / 注意事项 / 风险
🔥 内容质量参差 — 由于模型训练语料主要是网络小说 (网文),逻辑可能混乱、背景、角色、情节会出现“不合常识 /荒诞 /跳跃”,特别是当生成很长文本时。项目 README 中也有类似提醒。
🔥 语义/常识不可靠 — 模型欠缺现实世界的常识与约束,生成内容不适合严肃 /现实主义写作 (尤其是当你希望内容逻辑严谨、人物行为合理时)
🔥 可能重复 / 填充 /无意义文字 — 如果采样随机度太高或续写太长,有可能出现重复、无意义或格式混乱的段落。多数用户建议生成后手工修订。
🔥 仅供娱乐 /灵感参考 — 官方及社区多处强调:生成内容仅供娱乐 /灵感 /实践参考,不适合当作严肃创作或发表用途。
🔥 硬件 /性能限制 — 如果没有 GPU,使用 CPU 生成会比较慢;大模型、长文本、频繁续写,可能需要较多显存 /计算资源。
5、总结
AI-Writer 是一个有趣、门槛低、开源 + 可自托管 (self-hosted) 的中文 AI 写作 /小说生成工具。对于想快速“玩文字”、找灵感、练习写作、体验 AI 创作的人来说,非常值得一试。
但也要有清醒认知:它的输出不等同于“高质量文学作品”,更像是“草稿 /灵感 /雏形”。如果你希望写出结构严谨、情节逻辑清晰、人物丰满、语言精炼的小说/文章 —— 你可能仍需要人工审校 / 编辑 /润色。