updating the checkpoint for GPT2ForSequence Classification to one wit…

…h classification head (#11434)
huggingface · Apr 26, 2021 · 30f0658 · 30f0658
1 parent 35cd8ee
commit 30f0658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/gpt2/modeling_gpt2.py b/src/transformers/models/gpt2/modeling_gpt2.py
@@ -1236,7 +1236,7 @@ def __init__(self, config):
     @add_start_docstrings_to_model_forward(GPT2_INPUTS_DOCSTRING)
     @add_code_sample_docstrings(
         tokenizer_class=_TOKENIZER_FOR_DOC,
-        checkpoint="microsoft/dialogrpt",
+        checkpoint="microsoft/DialogRPT-updown",
         output_type=SequenceClassifierOutputWithPast,
         config_class=_CONFIG_FOR_DOC,
     )