emfomy commited on
Commit
e271be2
•
1 Parent(s): 4214020

Update usage.

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -35,11 +35,11 @@ Please use BertTokenizerFast as tokenizer instead of AutoTokenizer.
35
  ```
36
  from transformers import (
37
  BertTokenizerFast,
38
- AutoModelForMaskedLM,
39
  )
40
 
41
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
42
- model = AutoModelForMaskedLM.from_pretrained('ckiplab/albert-tiny-chinese')
43
  ```
44
 
45
  For full usage and more information, please refer to https://github.com/ckiplab/ckip-transformers.
 
35
  ```
36
  from transformers import (
37
  BertTokenizerFast,
38
+ AutoModel,
39
  )
40
 
41
  tokenizer = BertTokenizerFast.from_pretrained('bert-base-chinese')
42
+ model = AutoModel.from_pretrained('ckiplab/albert-tiny-chinese')
43
  ```
44
 
45
  For full usage and more information, please refer to https://github.com/ckiplab/ckip-transformers.