Xuefeng Bai commited on
Commit ·
17f7c83
1
Parent(s): 3fc0ffd
Update README.md
Browse files
README.md
CHANGED
|
@@ -36,8 +36,8 @@ be fine-tuned on a downstream task.
|
|
| 36 |
Here is how to initialize this model in PyTorch:
|
| 37 |
|
| 38 |
```python
|
| 39 |
-
from transformers import
|
| 40 |
-
model =
|
| 41 |
```
|
| 42 |
Please refer to [this repository](https://github.com/muyeby/AMRBART) for tokenizer initialization and data preprocessing.
|
| 43 |
|
|
|
|
| 36 |
Here is how to initialize this model in PyTorch:
|
| 37 |
|
| 38 |
```python
|
| 39 |
+
from transformers import BartForConditionalGeneration
|
| 40 |
+
model = BartForConditionalGeneration.from_pretrained("xfbai/AMRBART-large")
|
| 41 |
```
|
| 42 |
Please refer to [this repository](https://github.com/muyeby/AMRBART) for tokenizer initialization and data preprocessing.
|
| 43 |
|