CDial-GPT2_LCCC-base.zip
PyTorch GPT
在线加载模型地址

确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程

在代码中实现加载预训练模型地址

调用模型的实现方法


# 必须使用该方法下载模型,然后加载 

from flyai.utils import remote_helper 

path = remote_helper.get_remote_data('https://www.flyai.com/m/CDial-GPT2_LCCC-base.zip')

import os 

from transformers import BertTokenizer, GPT2LMHeadModel, GPT2Config 

model = GPT2LMHeadModel.from_pretrained(path) 

tokenizer = BertTokenizer.from_pretrained(path)