google
/

bigbird-pegasus-large-pubmed

Summarization Transformers PyTorch

English bigbird_pegasus text2text-generation Eval Results Inference Endpoints

Model card Files Files and versions Community

vasudevgupta commited on May 10, 2021

Commit

4565b53

•

1 Parent(s): 51784e7

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -25,7 +25,9 @@ BigBird relies on **block sparse attention** instead of normal attention (i.e. B
 Here is how to use this model to get the features of a given text in PyTorch:
 ```python
-from transformers import BigBirdPegasusForConditionalGeneration, BigBirdPegasusTokenizer
 # by default encoder-attention is `block_sparse` with num_random_blocks=3, block_size=64
 model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-pegasus-large-pubmed")

 Here is how to use this model to get the features of a given text in PyTorch:
 ```python
+from transformers import BigBirdPegasusForConditionalGeneration, AutoTokenizer
+tokenizer = AutoTokenizer.from_pretrained("google/bigbird-pegasus-large-pubmed")
 # by default encoder-attention is `block_sparse` with num_random_blocks=3, block_size=64
 model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-pegasus-large-pubmed")