模型:
nghuyong/ernie-3.0-base-zh
ERNIE 3.0:用于语言理解和生成的大规模知识增强预训练模型详情请见 https://arxiv.org/abs/2107.02137 。
此发布的pytorch模型是由官方发布的PaddlePaddle ERNIE模型转换而来,并进行了一系列实验来检查转换的准确性。
然后您可以像以前一样加载ERNIE-3.0模型:
from transformers import BertTokenizer, ErnieForMaskedLM tokenizer = BertTokenizer.from_pretrained("nghuyong/ernie-3.0-base-zh") model = ErnieForMaskedLM.from_pretrained("nghuyong/ernie-3.0-base-zh")
@article{sun2021ernie, title={Ernie 3.0: Large-scale knowledge enhanced pre-training for language understanding and generation}, author={Sun, Yu and Wang, Shuohuan and Feng, Shikun and Ding, Siyu and Pang, Chao and Shang, Junyuan and Liu, Jiaxiang and Chen, Xuyi and Zhao, Yanbin and Lu, Yuxiang and others}, journal={arXiv preprint arXiv:2107.02137}, year={2021} }