"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a2c90a7f7b1f8a2a8217c962a04a1a65638121d5"
Unverified Commit 17846646 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix `MobileViTV2` checkpoint name (#24018)



* fix

* fix

* Apply suggestions from code review
Co-authored-by: default avataramyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: default avataramyeroberts <22614925+amyeroberts@users.noreply.github.com>
parent 649ffbf5
...@@ -306,14 +306,14 @@ class MobileViTV2ModelIntegrationTest(unittest.TestCase): ...@@ -306,14 +306,14 @@ class MobileViTV2ModelIntegrationTest(unittest.TestCase):
@cached_property @cached_property
def default_image_processor(self): def default_image_processor(self):
return ( return (
MobileViTImageProcessor.from_pretrained("shehan97/mobilevitv2-1.0-imagenet1k-256") MobileViTImageProcessor.from_pretrained("apple/mobilevitv2-1.0-imagenet1k-256")
if is_vision_available() if is_vision_available()
else None else None
) )
@slow @slow
def test_inference_image_classification_head(self): def test_inference_image_classification_head(self):
model = MobileViTV2ForImageClassification.from_pretrained("shehan97/mobilevitv2-1.0-imagenet1k-256").to( model = MobileViTV2ForImageClassification.from_pretrained("apple/mobilevitv2-1.0-imagenet1k-256").to(
torch_device torch_device
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment