diff --git a/src/transformers/models/auto/modeling_auto.py b/src/transformers/models/auto/modeling_auto.py index 3d6d4eaea787..5bc4ef935969 100644 --- a/src/transformers/models/auto/modeling_auto.py +++ b/src/transformers/models/auto/modeling_auto.py @@ -130,7 +130,7 @@ class _BaseModelWithGenerate(PreTrainedModel, GenerationMixin): ("dpr", "DPRQuestionEncoder"), ("dpt", "DPTModel"), ("efficientformer", "EfficientFormerModel"), - ("efficientloftr", "EfficientLoFTRModel"), + ("efficientloftr", ("EfficientLoFTRModel", "EfficientLoFTRForKeypointMatching")), ("efficientnet", "EfficientNetModel"), ("electra", "ElectraModel"), ("emu3", "Emu3Model"),