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

LayoutLM Config (#9539)

Browse Source
This commit is contained in:
Lysandre Debut
2021-01-12 10:03:50 -05:00
committed by GitHub
parent e45eba3b1c
commit a1100fac67
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/auto/modeling_auto.py
View File

@@ -460,6 +460,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(LongformerConfig, LongformerForSequenceClassification), (LongformerConfig, LongformerForSequenceClassification),
(RobertaConfig, RobertaForSequenceClassification), (RobertaConfig, RobertaForSequenceClassification),
(SqueezeBertConfig, SqueezeBertForSequenceClassification), (SqueezeBertConfig, SqueezeBertForSequenceClassification),
(LayoutLMConfig, LayoutLMForSequenceClassification),
(BertConfig, BertForSequenceClassification), (BertConfig, BertForSequenceClassification),
(XLNetConfig, XLNetForSequenceClassification), (XLNetConfig, XLNetForSequenceClassification),
(MobileBertConfig, MobileBertForSequenceClassification), (MobileBertConfig, MobileBertForSequenceClassification),
@@ -475,7 +476,6 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(TransfoXLConfig, TransfoXLForSequenceClassification), (TransfoXLConfig, TransfoXLForSequenceClassification),
(MPNetConfig, MPNetForSequenceClassification), (MPNetConfig, MPNetForSequenceClassification),
(TapasConfig, TapasForSequenceClassification), (TapasConfig, TapasForSequenceClassification),
(LayoutLMConfig, LayoutLMForSequenceClassification),
] ]
) )

4
tests/test_modeling_auto.py
View File

@@ -241,7 +241,7 @@ class AutoModelTest(unittest.TestCase):
for parent_config, parent_model in mapping[: index + 1]: for parent_config, parent_model in mapping[: index + 1]:
assert not issubclass( assert not issubclass(
child_config, parent_config child_config, parent_config
), "{child_config.__name__} is child of {parent_config.__name__}" ), f"{child_config.__name__} is child of {parent_config.__name__}"
assert not issubclass( assert not issubclass(
child_model, parent_model child_model, parent_model
), "{child_config.__name__} is child of {parent_config.__name__}" ), f"{child_config.__name__} is child of {parent_config.__name__}"