McGill-NLP
/

bart-qg-nq-checkpoint

Text2Text Generation

Inference Endpoints

Model card Files Files and versions Community

geekydevu commited on Apr 1, 2022

Commit

26b6354

•

1 Parent(s): 578b52c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ from transformers import AutoModel, BartTokenizer
 #Load the tokenizer
 tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
 #Load the model
-model = AutoModel.from_pretrained("McGill-NLP/bart-qg-nq-checkpoint")
 ```
 ## Citation

 #Load the tokenizer
 tokenizer = BartTokenizer.from_pretrained('facebook/bart-base')
 #Load the model
+model = AutoModelForSeq2SeqLM.from_pretrained("McGill-NLP/bart-qg-nq-checkpoint")
 ```
 ## Citation