From 592d1e3aae81e6a085a6bf16581fefde9053eaed Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sat, 1 Jun 2019 16:19:32 -0400 Subject: [PATCH] fix typos --- hubconfs/gpt2_hubconf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hubconfs/gpt2_hubconf.py b/hubconfs/gpt2_hubconf.py index 0e802407f2..45bdd4ebdc 100644 --- a/hubconfs/gpt2_hubconf.py +++ b/hubconfs/gpt2_hubconf.py @@ -1,5 +1,5 @@ from pytorch_pretrained_bert.tokenization_gpt2 import GPT2Tokenizer -from pytorch_pretrained_bert.modeling_openai import ( +from pytorch_pretrained_bert.modeling_gpt2 import ( GPT2Model, GPT2LMHeadModel, GPT2DoubleHeadsModel @@ -152,7 +152,7 @@ def gpt2DoubleHeadsModel(*args, **kwargs): >>> tokens_tensor = torch.tensor([indexed_tokens]) >>> mc_token_ids = torch.LongTensor([ [len(tokenized_text)] ]) - # Load openAIGPTDoubleHeadsModel + # Load gpt2DoubleHeadsModel >>> model = torch.hub.load('huggingface/pytorch-pretrained-BERT', 'gpt2DoubleHeadsModel', 'gpt2') >>> model.eval()