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

LayoutLM Config (#9539)

parent e45eba3b
...@@ -460,6 +460,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict( ...@@ -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( ...@@ -475,7 +476,6 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(TransfoXLConfig, TransfoXLForSequenceClassification), (TransfoXLConfig, TransfoXLForSequenceClassification),
(MPNetConfig, MPNetForSequenceClassification), (MPNetConfig, MPNetForSequenceClassification),
(TapasConfig, TapasForSequenceClassification), (TapasConfig, TapasForSequenceClassification),
(LayoutLMConfig, LayoutLMForSequenceClassification),
] ]
) )
......
...@@ -241,7 +241,7 @@ class AutoModelTest(unittest.TestCase): ...@@ -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__}"
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