Skip to content

GitLab

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

LayoutLM Config (#9539)

parent e45eba3b
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
src/transformers/models/auto/modeling_auto.py
View file @ a1100fac
......@@ -460,6 +460,7 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(LongformerConfig, LongformerForSequenceClassification),
(RobertaConfig, RobertaForSequenceClassification),
(SqueezeBertConfig, SqueezeBertForSequenceClassification),
(LayoutLMConfig, LayoutLMForSequenceClassification),
(BertConfig, BertForSequenceClassification),
(XLNetConfig, XLNetForSequenceClassification),
(MobileBertConfig, MobileBertForSequenceClassification),
......@@ -475,7 +476,6 @@ MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING = OrderedDict(
(TransfoXLConfig, TransfoXLForSequenceClassification),
(MPNetConfig, MPNetForSequenceClassification),
(TapasConfig, TapasForSequenceClassification),
(LayoutLMConfig, LayoutLMForSequenceClassification),
]
)
......
tests/test_modeling_auto.py
View file @ a1100fac
......@@ -241,7 +241,7 @@ class AutoModelTest(unittest.TestCase):
for parent_config, parent_model in mapping[: index + 1]:
assert not issubclass(
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(
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