google
/

bigbird-pegasus-large-arxiv

bigbird_pegasus

text2text-generation

Model card Files Files and versions

vasudevgupta commited on May 4, 2021

Commit

8237f42

·

1 Parent(s): 2b48be8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -38,7 +38,7 @@ model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-p
 text = "Replace me by any text you'd like."
 inputs = tokenizer(text, return_tensors='pt')
-prediction = model.generate(input_ids=input_ids)
 prediction = tokenizer.batch_decode(prediction)
 ```

 text = "Replace me by any text you'd like."
 inputs = tokenizer(text, return_tensors='pt')
+prediction = model.generate(**inputs)
 prediction = tokenizer.batch_decode(prediction)
 ```