基于BART的长文本生成demo

陬者

关注

阅读 99

2022-04-22

这是一个使用pytorch,pytorch_lightning框架简单基于NLG领域预训练模型BART进行故事生成的demo。BART预训练模型来源于Hugginface: bart-base.

代码可见:
github: story-generation-demohttps://github.com/tangg555/story-generation-demo或者
gitee: story-generation-demohttps://gitee.com/travis-tang/story-generation-demo

运行程序的前置条件:
- Python 3 or Anaconda (mine is 3.8)
- Pytorch 
- transformers (a package for huggingface)
- pytorch-lightning (a package)


精彩评论(0)

0 0 举报