xfbai
/

AMRBART-large

Text2Text Generation Transformers PyTorch English bart AMRBART Inference Endpoints

Model card Files Files and versions Community

Xuefeng Bai commited on Apr 26, 2022

Commit

17f7c83

•

1 Parent(s): 3fc0ffd

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -36,8 +36,8 @@ be fine-tuned on a downstream task.
 Here is how to initialize this model in PyTorch:
 ```python
-from transformers import BartModel
-model = BartModel.from_pretrained("xfbai/AMRBART-large")
 ```
 Please refer to [this repository](https://github.com/muyeby/AMRBART) for tokenizer initialization and data preprocessing.

 Here is how to initialize this model in PyTorch:
 ```python
+from transformers import BartForConditionalGeneration
+model = BartForConditionalGeneration.from_pretrained("xfbai/AMRBART-large")
 ```
 Please refer to [this repository](https://github.com/muyeby/AMRBART) for tokenizer initialization and data preprocessing.