From 0891f0ec6e3d6a9dd3a4d2b1dbd2e1759b6bd1b0 Mon Sep 17 00:00:00 2001 From: Alex M Date: Mon, 18 Aug 2025 11:58:47 +0100 Subject: [PATCH] add-loftr-keypoints-to-map --- src/transformers/models/auto/modeling_auto.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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"),