From 50d3c79621fea1de2bb5c942d90333f57648c01f Mon Sep 17 00:00:00 2001
From: abiolaTresor <48957493+abiolaTresor@users.noreply.github.com>
Date: Mon, 26 Apr 2021 06:58:51 +0200
Subject: [PATCH] updating the checkpoint for GPT2ForSequence Classification to
 one with classification head (#11434)

---
 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 babe8ac1aa041c..22d009411087a2 100644
--- 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,
     )