Add GPT2 to sequence classification auto model (#7630)

8fa0c956 · Lysandre Debut · GitHub · e084089e · 8fa0c956
Unverified Commit 8fa0c956 authored Oct 07, 2020 by Lysandre Debut Committed by GitHub Oct 07, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/modeling_auto.py src/transformers/modeling_auto.py +2 -1

No files found.
--- a/src/transformers/modeling_auto.py
+++ b/src/transformers/modeling_auto.py
@@ -129,7 +129,7 @@ from .modeling_funnel import (
    FunnelForTokenClassification,
    FunnelModel,
 )
-from .modeling_gpt2 import GPT2LMHeadModel, GPT2Model
+from .modeling_gpt2 import GPT2ForSequenceClassification, GPT2LMHeadModel, GPT2Model
 from .modeling_layoutlm import LayoutLMForMaskedLM, LayoutLMForTokenClassification, LayoutLMModel
 from .modeling_longformer import (
    LongformerForMaskedLM,
@@ -377,6 +377,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
        (ElectraConfig, ElectraForSequenceClassification),
        (FunnelConfig, FunnelForSequenceClassification),
        (DebertaConfig, DebertaForSequenceClassification),
+        (GPT2Config, GPT2ForSequenceClassification),
    ]
 )