vasudevgupta commited on
Commit
ba85d08
1 Parent(s): 26cbf4b

fix tokenizer in readme

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -25,7 +25,9 @@ BigBird relies on **block sparse attention** instead of normal attention (i.e. B
25
  Here is how to use this model to get the features of a given text in PyTorch:
26
 
27
  ```python
28
- from transformers import BigBirdPegasusForConditionalGeneration, BigBirdPegasusTokenizer
 
 
29
 
30
  # by default encoder-attention is `block_sparse` with num_random_blocks=3, block_size=64
31
  model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-pegasus-large-arxiv")
 
25
  Here is how to use this model to get the features of a given text in PyTorch:
26
 
27
  ```python
28
+ from transformers import BigBirdPegasusForConditionalGeneration, AutoTokenizer
29
+
30
+ tokenizer = AutoTokenizer.from_pretrained("google/bigbird-pegasus-large-arxiv")
31
 
32
  # by default encoder-attention is `block_sparse` with num_random_blocks=3, block_size=64
33
  model = BigBirdPegasusForConditionalGeneration.from_pretrained("google/bigbird-pegasus-large-arxiv")