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

[Tests] Update model_id in AIMv2 Tests (#39281)

Browse Source 
* Update model_id in tests

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yaswanth Gali
2025-07-09 01:16:32 +05:30
committed by GitHub
parent 7ef592c96c
commit 0e1c281745
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/models/aimv2/test_modeling_aimv2.py
View File

@@ -598,7 +598,7 @@ class Aimv2ModelTest(Aimv2ModelTesterMixin, PipelineTesterMixin, unittest.TestCa
class Aimv2ModelIntegrationTest(unittest.TestCase):
@slow
def test_inference(self):
model_name = "yaswanthgali/aimv2-large-patch14-224-lit-HF"
model_name = "apple/aimv2-large-patch14-224-lit"
model = Aimv2Model.from_pretrained(model_name, device_map=torch_device)
processor = AutoProcessor.from_pretrained(model_name)
@@ -622,8 +622,8 @@ class Aimv2ModelIntegrationTest(unittest.TestCase):
)
# handle device
expected_logits = torch.tensor([[34.2415, 24.6724]]).to(model.device)
self.assertTrue(torch.allclose(outputs.logits_per_image, expected_logits, atol=1e-3))
expected_logits = torch.tensor([[33.3550, 26.4255]]).to(model.device)
torch.testing.assert_close(outputs.logits_per_image, expected_logits, atol=1e-3, rtol=1e-3)
@require_vision
@@ -631,7 +631,7 @@ class Aimv2ModelIntegrationTest(unittest.TestCase):
class Aimv2VisionModelIntegrationTests(unittest.TestCase):
@slow
def test_inference(self):
model_name = "yaswanthgali/aimv2-large-patch14-224-HF"
model_name = "apple/aimv2-large-patch14-224"
model = Aimv2VisionModel.from_pretrained(model_name, device_map=torch_device)
processor = AutoImageProcessor.from_pretrained(model_name)
@@ -662,7 +662,7 @@ class Aimv2VisionModelIntegrationTests(unittest.TestCase):
@slow
def test_inference_for_native_resolution(self):
model_name = "yaswanthgali/aimv2-large-patch14-native-HF"
model_name = "apple/aimv2-large-patch14-native"
model = Aimv2VisionModel.from_pretrained(model_name, device_map="auto")
processor = AutoImageProcessor.from_pretrained(model_name)