Skip to content

GitLab

Unverified Commit 8fa0c956 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Add GPT2 to sequence classification auto model (#7630)

parent e084089e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/transformers/modeling_auto.py
View file @ 8fa0c956
...@@ -129,7 +129,7 @@ from .modeling_funnel import ( ...@@ -129,7 +129,7 @@ from .modeling_funnel import (
FunnelForTokenClassification, FunnelForTokenClassification,
FunnelModel, FunnelModel,
) )
from .modeling_gpt2 import GPT2LMHeadModel, GPT2Model from .modeling_gpt2 import GPT2ForSequenceClassification, GPT2LMHeadModel, GPT2Model
from .modeling_layoutlm import LayoutLMForMaskedLM, LayoutLMForTokenClassification, LayoutLMModel from .modeling_layoutlm import LayoutLMForMaskedLM, LayoutLMForTokenClassification, LayoutLMModel
from .modeling_longformer import ( from .modeling_longformer import (
LongformerForMaskedLM, LongformerForMaskedLM,
...@@ -377,6 +377,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict( ...@@ -377,6 +377,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(ElectraConfig, ElectraForSequenceClassification), (ElectraConfig, ElectraForSequenceClassification),
(FunnelConfig, FunnelForSequenceClassification), (FunnelConfig, FunnelForSequenceClassification),
(DebertaConfig, DebertaForSequenceClassification), (DebertaConfig, DebertaForSequenceClassification),
(GPT2Config, GPT2ForSequenceClassification),
] ]
) )
......
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