开源项目:AI-Writer 小说 AI 生成器

admin
2
2025-12-22

1、简述

AI-Writer 是一个开源的中文「AI 写小说 / 文本生成」工具,由 BlinkDL 发布。它使用一个基于 RNN/语言模型 (具体为 RWKV‑LM) 的预训练模型 — 类似 GPT-2,但结构有所不同 — 专注于生成中文网文 (如玄幻、言情、网络小说) 。

与传统手写小说不同,AI-Writer 将“根据开头 + 一些设定 → 自动续写/生成后续内容”的流程自动化,使你能够快速生成连贯 (或半连贯) 的小说段落 / 章节。对于学习创作、娱乐写作、灵感草稿、尝试AI写作的人来说,它是一个方便、有趣、开源、可本地部署的候选。

不过项目本身也声明,它训练的数据主要来自网络小说 (“网文”),缺乏现实生活常识 — 所以生成内容更适合娱乐与灵感,而不一定严谨或现实。

image-cv5l.png


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 中会说明)

image-w3xc.png

输入开头 / 提示 + 参数

🔥 打开后,在界面 (或命令行) 中输入文章 / 小说的开头 (几句话或者一个段落)

🔥 设置续写长度、采样参数 (如 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 创作的人来说,非常值得一试。

但也要有清醒认知:它的输出不等同于“高质量文学作品”,更像是“草稿 /灵感 /雏形”。如果你希望写出结构严谨、情节逻辑清晰、人物丰满、语言精炼的小说/文章 —— 你可能仍需要人工审校 / 编辑 /润色。

动物装饰