SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[efficientloftr] fix model_id in tests (#39621)

Browse Source 
fix: wrong EfficientLoFTR model id in tests
This commit is contained in:
StevenBucaille
2025-07-24 05:41:06 -04:00
committed by GitHub
parent 947a37e8f5
commit 12b612830d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/models/efficientloftr/test_modeling_efficientloftr.py
View File

@@ -273,7 +273,7 @@ class EfficientLoFTRModelTest(ModelTesterMixin, unittest.TestCase):
@slow @slow
def test_model_from_pretrained(self): def test_model_from_pretrained(self):
from_pretrained_ids = ["stevenbucaille/efficientloftr"] from_pretrained_ids = ["zju-community/efficientloftr"]
for model_name in from_pretrained_ids: for model_name in from_pretrained_ids:
model = EfficientLoFTRForKeypointMatching.from_pretrained(model_name) model = EfficientLoFTRForKeypointMatching.from_pretrained(model_name)
self.assertIsNotNone(model) self.assertIsNotNone(model)
@@ -414,12 +414,12 @@ def prepare_imgs():
class EfficientLoFTRModelIntegrationTest(unittest.TestCase): class EfficientLoFTRModelIntegrationTest(unittest.TestCase):
@cached_property @cached_property
def default_image_processor(self): 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 @slow
def test_inference(self): def test_inference(self):
model = EfficientLoFTRForKeypointMatching.from_pretrained( model = EfficientLoFTRForKeypointMatching.from_pretrained(
"stevenbucaille/efficientloftr", attn_implementation="eager" "zju-community/efficientloftr", attn_implementation="eager"
).to(torch_device) ).to(torch_device)
preprocessor = self.default_image_processor preprocessor = self.default_image_processor
images = prepare_imgs() images = prepare_imgs()