From 0ebc9699fad3889a6e882dce1e4c465232d73438 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Mon, 24 Aug 2020 15:54:57 -0400 Subject: [PATCH] [fixdoc] Add import to pegasus usage doc (#6698) --- docs/source/model_doc/pegasus.rst | 1 + 1 file changed, 1 insertion(+) diff --git a/docs/source/model_doc/pegasus.rst b/docs/source/model_doc/pegasus.rst index 9275cf2152..852c03380c 100644 --- a/docs/source/model_doc/pegasus.rst +++ b/docs/source/model_doc/pegasus.rst @@ -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.""" ]