Skip to content

GitLab

Unverified Commit 35c52f2f authored by Sukuya's avatar Sukuya Committed by GitHub
Browse files

Update torchscript.rst 

Import fixed to pytorch_transformers else torchscript flag can't be used.
parent 067923d3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docs/source/torchscript.rst
View file @ 35c52f2f
...@@ -74,7 +74,7 @@ according to a ``BertConfig`` class and then saved to disk under the filename `` ...@@ -74,7 +74,7 @@ according to a ``BertConfig`` class and then saved to disk under the filename ``
.. code-block:: python .. code-block:: python
from pytorch_pretrained_bert import BertModel, BertTokenizer, BertConfig from pytorch_transformers import BertModel, BertTokenizer, BertConfig
import torch import torch
enc = BertTokenizer.from_pretrained("bert-base-uncased") 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 ...@@ -129,4 +129,4 @@ Using the traced model for inference is as simple as using its ``__call__`` dund
.. code-block:: python .. code-block:: python
traced_model(tokens_tensor, segments_tensors) traced_model(tokens_tensor, segments_tensors)
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment