From 77b76672e2b6072df80c36abfe880043f7efc814 Mon Sep 17 00:00:00 2001 From: Mishig Davaadorj Date: Wed, 29 Jun 2022 16:19:37 +0200 Subject: [PATCH] Fix img seg tests (load checkpoints from `hf-internal-testing`) (#17939) * Revert "Skip failing test until they are fixed." This reverts commit 8f400775fc5bc1011a2674dcfd5408d30d69f678. * Use `tiny-detr` checkpts from `hf-internal-testing` --- tests/pipelines/test_pipelines_image_segmentation.py | 3 +-- tests/pipelines/test_pipelines_object_detection.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/pipelines/test_pipelines_image_segmentation.py b/tests/pipelines/test_pipelines_image_segmentation.py index b3ec6cd9b2..405ad4e8a9 100644 --- a/tests/pipelines/test_pipelines_image_segmentation.py +++ b/tests/pipelines/test_pipelines_image_segmentation.py @@ -146,10 +146,9 @@ class ImageSegmentationPipelineTests(unittest.TestCase, metaclass=PipelineTestCa def test_small_model_tf(self): pass - @unittest.skip("Model has moved, skip until it's fixed.") @require_torch def test_small_model_pt(self): - model_id = "mishig/tiny-detr-mobilenetsv3-panoptic" + model_id = "hf-internal-testing/tiny-detr-mobilenetsv3-panoptic" model = AutoModelForImageSegmentation.from_pretrained(model_id) feature_extractor = AutoFeatureExtractor.from_pretrained(model_id) diff --git a/tests/pipelines/test_pipelines_object_detection.py b/tests/pipelines/test_pipelines_object_detection.py index 3346ca0df9..85463ac815 100644 --- a/tests/pipelines/test_pipelines_object_detection.py +++ b/tests/pipelines/test_pipelines_object_detection.py @@ -104,10 +104,9 @@ class ObjectDetectionPipelineTests(unittest.TestCase, metaclass=PipelineTestCase def test_small_model_tf(self): pass - @unittest.skip("Model has moved, skip until it's fixed.") @require_torch def test_small_model_pt(self): - model_id = "mishig/tiny-detr-mobilenetsv3" + model_id = "hf-internal-testing/tiny-detr-mobilenetsv3" model = AutoModelForObjectDetection.from_pretrained(model_id) feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)