diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py index 370e314710aae3..38793792fd073e 100644 --- a/src/transformers/models/auto/modeling_auto.py +++ b/src/transformers/models/auto/modeling_auto.py @@ -28,6 +28,10 @@ MODEL_MAPPING_NAMES = OrderedDict( [ # Base model mapping + ("albert", "AlbertModel"), + ("bart", "BartModel"), + ("beit", "BeitModel"), + ("bert", "BertModel"), ("bert-generation", "BertGenerationEncoder"), ("big_bird", "BigBirdModel"), ("bigbird_pegasus", "BigBirdPegasusModel"), @@ -56,6 +60,7 @@ ("flaubert", "FlaubertModel"), ("flava", "FlavaModel"), ("fnet", "FNetModel"), + ("fsmt", "FSMTModel"), ("funnel", ("FunnelModel", "FunnelBaseModel")), ("glpn", "GLPNModel"), ("gpt2", "GPT2Model"),