Update usage.

This commit is contained in:
Mu Yang 2021-01-06 13:47:42 +08:00
parent c550aa39f3
commit 9a0a13b5c8
1 changed files with 2 additions and 2 deletions

View File

@ -35,11 +35,11 @@ Please use BertTokenizerFast as tokenizer instead of AutoTokenizer.
```
from transformers import (
BertTokenizerFast,
AutoModelForTokenClassification,
AutoModel,
)
tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
model = AutoModelForTokenClassification.from_pretrained('ckiplab/bert-base-chinese-ner')
model = AutoModel.from_pretrained('ckiplab/bert-base-chinese-ner')
```
For full usage and more information, please refer to https://github.com/ckiplab/ckip-transformers.