Merge pull request #1 from sukuya/sukuya-patch-1

Update torchscript.rst
huggingface · Jul 25, 2019 · e1e2ab3 · e1e2ab3
2 parents 067923d + 35c52f2
commit e1e2ab3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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")
@@ -129,4 +129,4 @@ Using the traced model for inference is as simple as using its ``__call__`` dund
 
 .. code-block:: python
 
-    traced_model(tokens_tensor, segments_tensors)
+    traced_model(tokens_tensor, segments_tensors)