From c198ff5f1fc472fb1341f877bc1f5fb1c020c494 Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sat, 1 Jun 2019 16:28:42 -0400 Subject: [PATCH] fix typos/bugs --- hubconfs/gpt2_hubconf.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/hubconfs/gpt2_hubconf.py b/hubconfs/gpt2_hubconf.py index 45bdd4ebdc..8af60676b6 100644 --- a/hubconfs/gpt2_hubconf.py +++ b/hubconfs/gpt2_hubconf.py @@ -130,7 +130,7 @@ def gpt2LMHeadModel(*args, **kwargs): >>> predicted_token = tokenizer.decode([predicted_index]) >>> assert predicted_token == ' who' """ - model = OpenAIGPTLMHeadModel.from_pretrained(*args, **kwargs) + model = GPT2LMHeadModel.from_pretrained(*args, **kwargs) return model @@ -148,9 +148,9 @@ def gpt2DoubleHeadsModel(*args, **kwargs): # Prepare tokenized input >>> text = "Who was Jim Henson ?" - >>> indexed_tokens = tokenizer.encode(tokenized_text) + >>> indexed_tokens = tokenizer.encode(text) >>> tokens_tensor = torch.tensor([indexed_tokens]) - >>> mc_token_ids = torch.LongTensor([ [len(tokenized_text)] ]) + >>> mc_token_ids = torch.LongTensor([ [len(indexed_tokens)] ]) # Load gpt2DoubleHeadsModel >>> model = torch.hub.load('huggingface/pytorch-pretrained-BERT', 'gpt2DoubleHeadsModel', 'gpt2')