From 12b612830dc76a3a91d3fe1486b1bcb77b6ac4c4 Mon Sep 17 00:00:00 2001 From: StevenBucaille Date: Thu, 24 Jul 2025 05:41:06 -0400 Subject: [PATCH] [efficientloftr] fix model_id in tests (#39621) fix: wrong EfficientLoFTR model id in tests --- tests/models/efficientloftr/test_modeling_efficientloftr.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/models/efficientloftr/test_modeling_efficientloftr.py b/tests/models/efficientloftr/test_modeling_efficientloftr.py index 50c24c41fa..1efb2e1620 100644 --- a/tests/models/efficientloftr/test_modeling_efficientloftr.py +++ b/tests/models/efficientloftr/test_modeling_efficientloftr.py @@ -273,7 +273,7 @@ class EfficientLoFTRModelTest(ModelTesterMixin, unittest.TestCase): @slow def test_model_from_pretrained(self): - from_pretrained_ids = ["stevenbucaille/efficientloftr"] + from_pretrained_ids = ["zju-community/efficientloftr"] for model_name in from_pretrained_ids: model = EfficientLoFTRForKeypointMatching.from_pretrained(model_name) self.assertIsNotNone(model) @@ -414,12 +414,12 @@ def prepare_imgs(): class EfficientLoFTRModelIntegrationTest(unittest.TestCase): @cached_property def default_image_processor(self): - return AutoImageProcessor.from_pretrained("stevenbucaille/efficientloftr") if is_vision_available() else None + return AutoImageProcessor.from_pretrained("zju-community/efficientloftr") if is_vision_available() else None @slow def test_inference(self): model = EfficientLoFTRForKeypointMatching.from_pretrained( - "stevenbucaille/efficientloftr", attn_implementation="eager" + "zju-community/efficientloftr", attn_implementation="eager" ).to(torch_device) preprocessor = self.default_image_processor images = prepare_imgs()