最近ChatGPT大火,但是很多人还不知道什么是GPT,GPT全称是generative pre-trained transformer(可生成的,预训练的transformer),实际上就是文本生成(text generation),只不过ChatGPT加入了对话,使得它可以通过理解人类的描述生成文本。
今天我们来用一下微软开源的BioGPT,BioGPT是微软开发的,针对生物医学领域的GPT模型,可用于关系提取、问答、文档分类和文本生成任务。据官方介绍,BioGPT在大多数任务上的表现均优于之前的模型,所以我们来尝试一下。
这篇教程只使用了文本生成模型,仅支持英文,因为我的显存只有4G,太小,加载模型会导致显存溢出,所以只使用到了CPU,没有使用GPU。
首先来安装环境。
我们要分别安装pytorch(参考之前的安装教程)、transformers和sacremoses(后面这两个直接使用pip安装即可)。
接下来输入如下代码:

然后把“COVID 19 is"换成自己想使用的文本开头即可。
需要注意,生成的结果都属于生物医学领域,比如我之前尝试的结果就是各种疾病之类的。
代码写完后直接执行,首次使用会下载模型到C盘的缓存目录,大概需要6G左右的空间,模型下载完成之后就会开始生成文本,比如我生成的结果:

祝吧友玩的开心~
今天我们来用一下微软开源的BioGPT,BioGPT是微软开发的,针对生物医学领域的GPT模型,可用于关系提取、问答、文档分类和文本生成任务。据官方介绍,BioGPT在大多数任务上的表现均优于之前的模型,所以我们来尝试一下。
这篇教程只使用了文本生成模型,仅支持英文,因为我的显存只有4G,太小,加载模型会导致显存溢出,所以只使用到了CPU,没有使用GPU。
首先来安装环境。
我们要分别安装pytorch(参考之前的安装教程)、transformers和sacremoses(后面这两个直接使用pip安装即可)。
接下来输入如下代码:

然后把“COVID 19 is"换成自己想使用的文本开头即可。
需要注意,生成的结果都属于生物医学领域,比如我之前尝试的结果就是各种疾病之类的。
代码写完后直接执行,首次使用会下载模型到C盘的缓存目录,大概需要6G左右的空间,模型下载完成之后就会开始生成文本,比如我生成的结果:

祝吧友玩的开心~