Small changes in perplexity.rstto make the notebook executable on google collaboratory (#13541)

* add imports * Update docs/source/perplexity.rst Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2021-09-13 13:32:32 +02:00
parent f1c22dae7d
commit 149c833b75
1 changed files with 3 additions and 0 deletions
--- a/docs/source/perplexity.rst
+++ b/docs/source/perplexity.rst
@@ -110,6 +110,9 @@ available to condition on).

 .. code-block:: python

+    import torch
+    from tqdm import tqdm
+
    max_length = model.config.n_positions
    stride = 512