这是一个使用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条)