Add GPT2 to sequence classification auto model (#7630)

2020-10-07 05:20:05 -04:00
parent e084089eb9
commit 8fa0c956b3
1 changed files with 2 additions and 1 deletions
@@ -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),
    ]
 )