在发布 RWKV-6-ChnNovel 中文小说模型的同时,我们也发布了与小说模型搭配使用的小说扩写 State。
小说扩写 State 下载链接:https://huggingface.co/BlinkDL/temp-latest-training-models/tree/main ✨ RWKV State 文件是什么?
RWKV 的 State 文件可以强化模型在特定使命的表现,类似于 RWKV 模型的增强插件。详细用法请查察: RWKV state 介绍和用法
小说扩写 State 需要搭配 RWKV-6-ChnNovel 小说模型使用,主要有以下两种功能:
通过插入大纲,控制小说续写方向
从一条大纲扩写一个小说段落
通过大纲控制小说续写方向
搭载 State 文件后,依然可以按照前文的方法,使用 RWKV-6-ChnNovel 小说模型进行小说续写。
唯一不同的是,搭载 State 之后,我们可以通过插入一段“###下文情节”格式的细纲来控制小说剧情的走向。
下图红框圈出来的是用户手动输入的小说片断和大纲,剩余内容由模型生成:
可以看到,搭载 State 之后,我们在续写之前先为模型指定了一条大纲:
###下文情节:苏月舒报复性的买了一大堆薯片,苏陌劝告苏月舒不要吃这些垃圾食品
插入大纲可以更好地控制模型续写的内容走向,制止出现脱轨题目。 假如你没有预备好后文的大纲,可以通过留白大纲内容,让模型为你生成一条大纲。
你可以抽卡出合适的大纲,或修改模型生成的大纲,然后再继承让模型续写符合大纲内容的后文。
参考下图中的格式:
从给定的大纲扩写小说段落
搭载小说扩写 State 后,RWKV-6-ChnNovel 小说模型也可以实现从一条大纲扩写出小说段落。
下图是一个从大纲扩写小说段落的示例: