Commit c198ff5f authored by VictorSanh's avatar VictorSanh
Browse files

fix typos/bugs

parent 592d1e3a
......@@ -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')
......
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