这是一个使用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)
基于BART的长文本生成demo
最新推荐文章于 2025-03-31 10:26:39 发布