Update torchscript.rst

Import fixed to pytorch_transformers else torchscript flag can't be used.

Update torchscript.rst
Import fixed to pytorch_transformers else torchscript flag can't be used.
35c52f2f · Sukuya · GitHub · 067923d3 · 35c52f2f
Unverified Commit 35c52f2f authored Jul 25, 2019 by Sukuya Committed by GitHub Jul 25, 2019
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

docs/source/torchscript.rst docs/source/torchscript.rst +2 -2

No files found.
--- a/docs/source/torchscript.rst
+++ b/docs/source/torchscript.rst
@@ -74,7 +74,7 @@ according to a ``BertConfig`` class and then saved to disk under the filename ``

 .. code-block:: python

-    from pytorch_pretrained_bert import BertModel, BertTokenizer, BertConfig
+    from pytorch_transformers import BertModel, BertTokenizer, BertConfig
    import torch

    enc = BertTokenizer.from_pretrained("bert-base-uncased")