"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "abca1741cf065749c44ef5d77f6f632c50beb070"
Unverified Commit 77b76672 authored by Mishig Davaadorj's avatar Mishig Davaadorj Committed by GitHub
Browse files

Fix img seg tests (load checkpoints from `hf-internal-testing`) (#17939)

* Revert "Skip failing test until they are fixed."

This reverts commit 8f400775.

* Use `tiny-detr` checkpts from `hf-internal-testing`
parent 3cff4cc5
...@@ -146,10 +146,9 @@ class ImageSegmentationPipelineTests(unittest.TestCase, metaclass=PipelineTestCa ...@@ -146,10 +146,9 @@ class ImageSegmentationPipelineTests(unittest.TestCase, metaclass=PipelineTestCa
def test_small_model_tf(self): def test_small_model_tf(self):
pass pass
@unittest.skip("Model has moved, skip until it's fixed.")
@require_torch @require_torch
def test_small_model_pt(self): 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) model = AutoModelForImageSegmentation.from_pretrained(model_id)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_id) feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)
......
...@@ -104,10 +104,9 @@ class ObjectDetectionPipelineTests(unittest.TestCase, metaclass=PipelineTestCase ...@@ -104,10 +104,9 @@ class ObjectDetectionPipelineTests(unittest.TestCase, metaclass=PipelineTestCase
def test_small_model_tf(self): def test_small_model_tf(self):
pass pass
@unittest.skip("Model has moved, skip until it's fixed.")
@require_torch @require_torch
def test_small_model_pt(self): 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) model = AutoModelForObjectDetection.from_pretrained(model_id)
feature_extractor = AutoFeatureExtractor.from_pretrained(model_id) feature_extractor = AutoFeatureExtractor.from_pretrained(model_id)
......
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