Skip to content

GitLab

"examples/vscode:/vscode.git/clone" did not exist on "c617f988f83d57ffb3146038c193286cea892522"
Unverified Commit 58e8a761 authored by Xu Song's avatar Xu Song Committed by GitHub
Browse files

Fix gpt2 document (#9272)

parent cbe63949
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/gpt2/modeling_gpt2.py
View file @ 58e8a761
...@@ -1036,7 +1036,7 @@ class GPT2DoubleHeadsModel(GPT2PreTrainedModel): ...@@ -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
""" """
......
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