Fix gpt2 document (#9272)

2020-12-23 18:34:15 +08:00 · 2020-12-23 18:34:15 +08:00 · 58e8a7611f
parent cbe63949d7
commit 58e8a7611f
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/models/gpt2/modeling_gpt2.py
+++ b/src/transformers/models/gpt2/modeling_gpt2.py
@ -1036,7 +1036,7 @@ class GPT2DoubleHeadsModel(GPT2PreTrainedModel):
            >>> mc_token_ids = torch.tensor([cls_token_location])  # Batch size: 1

            >>> outputs = model(input_ids, mc_token_ids=mc_token_ids)
-            >>> lm_logits = outputs.lm_logits
+            >>> lm_logits = outputs.logits
            >>> mc_logits = outputs.mc_logits

        """