riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-14 18:18:24 +06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Merge pull request #899 from sukuya/master

Browse Source 
Fixed import to use torchscript flag.
This commit is contained in:
Thomas Wolf 2019-07-25 15:03:21 +02:00 committed by GitHub
commit 46cc9dd2b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/torchscript.rst
View File

@ -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")