Merge pull request #899 from sukuya/master

Fixed import to use torchscript flag.

Merge pull request #899 from sukuya/master
Fixed import to use torchscript flag.
46cc9dd2 · Thomas Wolf · GitHub · 6219ad72 · e1e2ab34 · 46cc9dd2
Unverified Commit 46cc9dd2 authored Jul 25, 2019 by Thomas Wolf 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")