EasyNLP: A Comprehensive and Easy-to-use Toolkit for Natural Language Processing
Paper • 2205.00258 • Published
How to use alibaba-pai/pai-bert-tiny-zh with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="alibaba-pai/pai-bert-tiny-zh") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("alibaba-pai/pai-bert-tiny-zh", dtype="auto")This project provides Chinese pre-trained language models and various types of NLP tools. The models are pre-trained on the large-scale corpora hosted by the Alibaba PAI team. It is developed based on the EasyNLP framework (https://github.com/alibaba/EasyNLP).
If you find the resource is useful, please cite the following paper in your work:
@article{easynlp,
title = {EasyNLP: A Comprehensive and Easy-to-use Toolkit for Natural Language Processing},
publisher = {arXiv},
author = {Wang, Chengyu and Qiu, Minghui and Zhang, Taolin and Liu, Tingting and Li, Lei and Wang, Jianing and Wang, Ming and Huang, Jun and Lin, Wei},
url = {https://arxiv.org/abs/2205.00258},
year = {2022}
}