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

2、安装 与 配置 --- 快速上手
下面是一个典型的「从零开始」安装 + 运行流程 (以 Python 运行版本为主):
克隆项目
bash
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,也可以安装普通版本 (但速度较慢)
运行程序
-
在命令行里运行 (以文本生成模式):
bashpython run.py或者如果你希望通过网页界面 (web UI) 使用,运行:
bashpython 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 创作的人来说,非常值得一试。
但也要有清醒认知:它的输出不等同于"高质量文学作品",更像是"草稿 /灵感 /雏形"。如果你希望写出结构严谨、情节逻辑清晰、人物丰满、语言精炼的小说/文章 ------ 你可能仍需要人工审校 / 编辑 /润色。