From 58e8a7611f21ee6f6b41601d84a2387589be61ab Mon Sep 17 00:00:00 2001 From: Xu Song Date: Wed, 23 Dec 2020 18:34:15 +0800 Subject: [PATCH] Fix gpt2 document (#9272) --- src/transformers/models/gpt2/modeling_gpt2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/gpt2/modeling_gpt2.py b/src/transformers/models/gpt2/modeling_gpt2.py index cc3f288903..d34bcd7165 100644 --- 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 """