Skip to content

Commit

Permalink
[fixdoc] Add import to pegasus usage doc (huggingface#6698)
Browse files Browse the repository at this point in the history
  • Loading branch information
sshleifer authored and fabiocapsouza committed Nov 15, 2020
1 parent 031d1bd commit fe0965b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions docs/source/model_doc/pegasus.rst
Original file line number Diff line number Diff line change
Expand Up @@ -48,6 +48,7 @@ Usage Example
.. code-block:: python
from transformers import PegasusForConditionalGeneration, PegasusTokenizer
import torch
src_text = [
""" PG&E stated it scheduled the blackouts in response to forecasts for high winds amid dry conditions. The aim is to reduce the risk of wildfires. Nearly 800 thousand customers were scheduled to be affected by the shutoffs which were expected to last through at least midday tomorrow."""
]
Expand Down

0 comments on commit fe0965b

Please sign in to comment.