From 35c52f2f3cf85e26a85a7c52cff789983edaa62c Mon Sep 17 00:00:00 2001 From: Sukuya Date: Thu, 25 Jul 2019 16:51:11 +0800 Subject: [PATCH] Update torchscript.rst Import fixed to pytorch_transformers else torchscript flag can't be used. --- docs/source/torchscript.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/torchscript.rst b/docs/source/torchscript.rst index 1b845595672..e207068fad1 100644 --- 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) \ No newline at end of file + traced_model(tokens_tensor, segments_tensors)