YixuanWeng
commited on
Commit
·
996025f
1
Parent(s):
569aea9
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,10 +13,10 @@ CMPT (Chinese Minority Pre-Trained Language Model) 是在BART的基础上,加
|
|
| 13 |
|
| 14 |
|
| 15 |
```python
|
| 16 |
-
>>> from modeling_cmpt import
|
| 17 |
>>> from transformers import AutoTokenizer
|
| 18 |
>>> tokenizer = AutoTokenizer.from_pretrained('./CMTP')
|
| 19 |
-
>>> model =
|
| 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]))
|