fix typo

cc @thomwolf
2019-06-01 17:29:07 -04:00
parent 466a96543a
commit 8f97f6c57f
2 changed files with 2 additions and 2 deletions
--- a/pytorch_pretrained_bert/modeling_openai.py
+++ b/pytorch_pretrained_bert/modeling_openai.py
@@ -789,7 +789,7 @@ class OpenAIGPTDoubleHeadsModel(OpenAIGPTPreTrainedModel):

    config = modeling_openai.OpenAIGPTConfig()

-    model = modeling_openai.OpenAIGPTLMHeadModel(config)
+    model = modeling_openai.OpenAIGPTDoubleHeadsModel(config)
    lm_logits, multiple_choice_logits = model(input_ids, mc_token_ids)
    ```
    """