Views
No views yet
1from transformers import AutoModelForMaskedLM, AutoTokenizer, FillMaskPipeline
2import torch
3
4tokenizer=AutoTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese', use_fast=False)
5model=AutoModelForMaskedLM.from_pretrained('IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-Chinese')
6text = '生活的真谛是[MASK]。'
7fillmask_pipe = FillMaskPipeline(model, tokenizer, device=7)
8print(fillmask_pipe(text, top_k=10))| Model | OCNLI | CMNLI |
|---|---|---|
| RoBERTa-base | 0.743 | 0.7973 |
| Erlangshen-Deberta-97M-Chinese | 0.752 | 0.807 |
@misc{Fengshenbang-LM,
title={Fengshenbang-LM},
author={IDEA-CCNL},
year={2022},
howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
}