Unverified Commit 0ebc9699 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[fixdoc] Add import to pegasus usage doc (#6698)

parent 6b4c6176
......@@ -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."""
]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment