haryoaw
/

id-recigen-bart

Text2Text Generation

Model card Files Files and versions Community

haryoaw commited on Apr 16, 2022

Commit

c671738

·

1 Parent(s): 9a4b043

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -46,9 +46,9 @@ tokenizer = IndoNLGTokenizer.from_pretrained("haryoaw/id-recigen-bart")
 The model can be loaded by using AutoModel.
 ```python
-from transformers import AutoModelWithLMHead
-model = AutoModelWithLMHead.from_pretrained("haryoaw/id-recigen-bart")
 ```

 The model can be loaded by using AutoModel.
 ```python
+from transformers import AutoModelForSeq2SeqLM
+model = AutoModelForSeq2SeqLM.from_pretrained("haryoaw/id-recigen-bart")
 ```