From 210d407245f42e50eff723297f2b1cbb9793ab2e Mon Sep 17 00:00:00 2001 From: thomwolf Date: Mon, 18 Feb 2019 00:55:39 +0100 Subject: [PATCH] updating init --- pytorch_pretrained_bert/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_pretrained_bert/__init__.py b/pytorch_pretrained_bert/__init__.py index eeb8392728..7cb28e2d2b 100644 --- a/pytorch_pretrained_bert/__init__.py +++ b/pytorch_pretrained_bert/__init__.py @@ -2,6 +2,7 @@ __version__ = "0.6.0" from .tokenization import BertTokenizer, BasicTokenizer, WordpieceTokenizer from .tokenization_openai import OpenAIGPTTokenizer from .tokenization_transfo_xl import (TransfoXLTokenizer, TransfoXLCorpus) +from .tokenization_gpt2 import GPT2Tokenizer from .modeling import (BertConfig, BertModel, BertForPreTraining, BertForMaskedLM, BertForNextSentencePrediction,