diff --git a/src/transformers/__init__.py b/src/transformers/__init__.py index bfc3e682f..b98dfa0e6 100755 --- a/src/transformers/__init__.py +++ b/src/transformers/__init__.py @@ -1087,10 +1087,10 @@ if is_tf_available(): _import_structure["models.layoutlm"].extend( [ "TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LIST", - "TFLayoutLMMainLayer", "TFLayoutLMForMaskedLM", "TFLayoutLMForSequenceClassification", "TFLayoutLMForTokenClassification", + "TFLayoutLMMainLayer", "TFLayoutLMModel", "TFLayoutLMPreTrainedModel", ]