Skip to content

GitLab

Unverified Commit 30f06589 authored by abiolaTresor's avatar abiolaTresor Committed by GitHub
Browse files

updating the checkpoint for GPT2ForSequence Classification to one with classification head (#11434)

parent 35cd8eed
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/gpt2/modeling_gpt2.py
View file @ 30f06589
...@@ -1236,7 +1236,7 @@ class GPT2ForSequenceClassification(GPT2PreTrainedModel): ...@@ -1236,7 +1236,7 @@ class GPT2ForSequenceClassification(GPT2PreTrainedModel):
@add_start_docstrings_to_model_forward(GPT2_INPUTS_DOCSTRING) @add_start_docstrings_to_model_forward(GPT2_INPUTS_DOCSTRING)
@add_code_sample_docstrings( @add_code_sample_docstrings(
tokenizer_class=_TOKENIZER_FOR_DOC, tokenizer_class=_TOKENIZER_FOR_DOC,
checkpoint="microsoft/dialogrpt", checkpoint="microsoft/DialogRPT-updown",
output_type=SequenceClassifierOutputWithPast, output_type=SequenceClassifierOutputWithPast,
config_class=_CONFIG_FOR_DOC, config_class=_CONFIG_FOR_DOC,
) )
......
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