SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix gpt2 document (#9272)

Browse Source
This commit is contained in:
Xu Song
2020-12-23 18:34:15 +08:00
committed by GitHub
parent cbe63949d7
commit 58e8a7611f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/gpt2/modeling_gpt2.py
View File

@@ -1036,7 +1036,7 @@ class GPT2DoubleHeadsModel(GPT2PreTrainedModel):
>>> mc_token_ids = torch.tensor([cls_token_location]) # Batch size: 1 >>> mc_token_ids = torch.tensor([cls_token_location]) # Batch size: 1
>>> outputs = model(input_ids, mc_token_ids=mc_token_ids) >>> outputs = model(input_ids, mc_token_ids=mc_token_ids)
>>> lm_logits = outputs.lm_logits >>> lm_logits = outputs.logits
>>> mc_logits = outputs.mc_logits >>> mc_logits = outputs.mc_logits
""" """