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
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()