ASOUL-Generator-Backend
本项目为 https://asoul.infedg.xyz/ 的后端。 模型为基于 CPM-Distill 的 transformers 转化版本 CPM-Generate-distill 训练而成。 训练数据集:
运行方式
下载模型
需要将文件夹内的两个文件(pytorch_model.bin
和 config.json
) 放入 asoul_cpm
文件夹下。
模型会不定期更新。
安装依赖
pip install -r requirements.txt
运行后端
python3 api.py
此时后端运行在 5089
端口。