Skip to content

pass-lin/misaka-writer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

image

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镇楼

image

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages