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

2020-08-24 15:54:57 -04:00
parent 6b4c617666
commit 0ebc9699fa
1 changed files with 1 additions and 0 deletions
--- 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."""
    ]