SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix for XLMRobertaConfig inherits from RobertaConfig

Browse Source 
hat/tip @stefan-it
This commit is contained in:
Julien Chaumond
2020-01-22 17:50:24 -05:00
parent 1a8e87be4e
commit 34a3c25a30
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/modeling_auto.py
View File

@@ -130,8 +130,8 @@ MODEL_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertModel), (DistilBertConfig, DistilBertModel),
(AlbertConfig, AlbertModel), (AlbertConfig, AlbertModel),
(CamembertConfig, CamembertModel), (CamembertConfig, CamembertModel),
(RobertaConfig, RobertaModel),
(XLMRobertaConfig, XLMRobertaModel), (XLMRobertaConfig, XLMRobertaModel),
(RobertaConfig, RobertaModel),
(BertConfig, BertModel), (BertConfig, BertModel),
(OpenAIGPTConfig, OpenAIGPTModel), (OpenAIGPTConfig, OpenAIGPTModel),
(GPT2Config, GPT2Model), (GPT2Config, GPT2Model),
@@ -148,8 +148,8 @@ MODEL_WITH_LM_HEAD_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertForMaskedLM), (DistilBertConfig, DistilBertForMaskedLM),
(AlbertConfig, AlbertForMaskedLM), (AlbertConfig, AlbertForMaskedLM),
(CamembertConfig, CamembertForMaskedLM), (CamembertConfig, CamembertForMaskedLM),
(RobertaConfig, RobertaForMaskedLM),
(XLMRobertaConfig, XLMRobertaForMaskedLM), (XLMRobertaConfig, XLMRobertaForMaskedLM),
(RobertaConfig, RobertaForMaskedLM),
(BertConfig, BertForMaskedLM), (BertConfig, BertForMaskedLM),
(OpenAIGPTConfig, OpenAIGPTLMHeadModel), (OpenAIGPTConfig, OpenAIGPTLMHeadModel),
(GPT2Config, GPT2LMHeadModel), (GPT2Config, GPT2LMHeadModel),
@@ -165,8 +165,8 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertForSequenceClassification), (DistilBertConfig, DistilBertForSequenceClassification),
(AlbertConfig, AlbertForSequenceClassification), (AlbertConfig, AlbertForSequenceClassification),
(CamembertConfig, CamembertForSequenceClassification), (CamembertConfig, CamembertForSequenceClassification),
(RobertaConfig, RobertaForSequenceClassification),
(XLMRobertaConfig, XLMRobertaForSequenceClassification), (XLMRobertaConfig, XLMRobertaForSequenceClassification),
(RobertaConfig, RobertaForSequenceClassification),
(BertConfig, BertForSequenceClassification), (BertConfig, BertForSequenceClassification),
(XLNetConfig, XLNetForSequenceClassification), (XLNetConfig, XLNetForSequenceClassification),
(XLMConfig, XLMForSequenceClassification), (XLMConfig, XLMForSequenceClassification),
@@ -187,8 +187,8 @@ MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING = OrderedDict(
[ [
(DistilBertConfig, DistilBertForTokenClassification), (DistilBertConfig, DistilBertForTokenClassification),
(CamembertConfig, CamembertForTokenClassification), (CamembertConfig, CamembertForTokenClassification),
(RobertaConfig, RobertaForTokenClassification),
(XLMRobertaConfig, XLMRobertaForTokenClassification), (XLMRobertaConfig, XLMRobertaForTokenClassification),
(RobertaConfig, RobertaForTokenClassification),
(BertConfig, BertForTokenClassification), (BertConfig, BertForTokenClassification),
(XLNetConfig, XLNetForTokenClassification), (XLNetConfig, XLNetForTokenClassification),
] ]

2
src/transformers/tokenization_auto.py
View File

@@ -60,8 +60,8 @@ TOKENIZER_MAPPING = OrderedDict(
(DistilBertConfig, DistilBertTokenizer), (DistilBertConfig, DistilBertTokenizer),
(AlbertConfig, AlbertTokenizer), (AlbertConfig, AlbertTokenizer),
(CamembertConfig, CamembertTokenizer), (CamembertConfig, CamembertTokenizer),
(RobertaConfig, RobertaTokenizer),
(XLMRobertaConfig, XLMRobertaTokenizer), (XLMRobertaConfig, XLMRobertaTokenizer),
(RobertaConfig, RobertaTokenizer),
(BertConfig, BertTokenizer), (BertConfig, BertTokenizer),
(OpenAIGPTConfig, OpenAIGPTTokenizer), (OpenAIGPTConfig, OpenAIGPTTokenizer),
(GPT2Config, GPT2Tokenizer), (GPT2Config, GPT2Tokenizer),