YixuanWeng commited on
Commit
996025f
·
1 Parent(s): 569aea9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -13,10 +13,10 @@ CMPT (Chinese Minority Pre-Trained Language Model) 是在BART的基础上,加
13
 
14
 
15
  ```python
16
- >>> from modeling_cmpt import CMPTForCir
17
  >>> from transformers import AutoTokenizer
18
  >>> tokenizer = AutoTokenizer.from_pretrained('./CMTP')
19
- >>> model = CMPTForCir.from_pretrained('./CMTP')
20
  >>> inputs = tokenizer.encode("Hello world, 你好 世界", return_tensors='pt')
21
  >>> pred_ids = model.generate(input_ids, num_beams=4, max_length=20)
22
  >>> print(tokenizer.convert_ids_to_tokens(pred_ids[i]))
 
13
 
14
 
15
  ```python
16
+ >>> from modeling_cmpt import BartForConditionalGeneration
17
  >>> from transformers import AutoTokenizer
18
  >>> tokenizer = AutoTokenizer.from_pretrained('./CMTP')
19
+ >>> model = BartForConditionalGeneration.from_pretrained('./CMTP')
20
  >>> inputs = tokenizer.encode("Hello world, 你好 世界", return_tensors='pt')
21
  >>> pred_ids = model.generate(input_ids, num_beams=4, max_length=20)
22
  >>> print(tokenizer.convert_ids_to_tokens(pred_ids[i]))