Skip to content

GitLab

Commit 592d1e3a authored by VictorSanh's avatar VictorSanh
Browse files

fix typos

parent f836130b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
hubconfs/gpt2_hubconf.py
View file @ 592d1e3a
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()
......
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