misaka-writer

ai-续写小说

基于encoder-decoder结构的续写小说模型，模型比较小只有8kw，模型结构是魔改的transformer模型。

依赖环境

本项目的依赖有：tensorflow bert4keras jieba pandas。

如果使用GPU请安装 cuda 和 cudnn。

推荐的配置为 tensorflow 2.3.0，cuda 10.1，cudnn 7.6。

对于不支持 cuda 10 的 30 系显卡，建议使用 tensorflow 2.5.0，cuda 11.2，cudnn 8。

使用 conda 配置

对于 tensorflow 2.3.0：

conda create -n misaka-writer python=3.8
conda activate misaka-writer
conda install -c conda-forge pandas cudatoolkit=10.1 cudnn
pip install tensorflow==2.3.0 bert4keras jieba

对于 tensorflow 2.5.0：

conda create -n misaka-writer python=3.9
conda activate misaka-writer
conda install -c conda-forge pandas cudatoolkit=11.2 cudnn
pip install tensorflow==2.5.0 bert4keras jieba

使用方法

见 main.py。

model_path 是模型的权重路径，建议使用相对路径。

support_english 表示英文/中文模式，False 表示中文模式。

num 代表生成的下文的数量。 text 为输入，建议输入在20到250字之间。

训练语料

训练语料有100G中文和50G英文。

链接：https://pan.baidu.com/s/1WCiPA_tplI0AhdpDEuQ5ig
提取码：rlse

预训练权重

base/综合模型

链接：https://pan.baidu.com/s/1SdvL6W70np2qp9jDWbsGVQ
提取码：9sno

玄幻

链接：https://pan.baidu.com/s/1vGBJr6NOsWQAvJvxjqld-w
提取码：hszv

日轻

链接：https://pan.baidu.com/s/1n7vXu-1uLF6XKtizoJmQZg
提取码：miw0

如果受不了百度云的网速，建议加QQ群在群文件下载，更多类型目前现在Q群测试。 目前Q群总共有"穿越重生" "都市" "二次元" "古言" "洪荒" "科幻" "虐文" "女言" "日轻" "网游电竞" "武侠" "西方名著" "玄幻" "玄幻v2" "娱乐圈" "中国古典文学" "中国现代文学" "综合" "综合言情" 版本供使用

社区

如有问题可加Q群-143626394(大群，除了本项目还有 https://github.com/BlinkDL/AI-Writer 项目群）、905398734（本项目小群），本人qq 935499957

感谢由北大美少女提供的进度条和ui，ui详解：https://github.com/Wybxc/misaka-writer-ui

最后，misaka镇楼