"docs/source/vscode:/vscode.git/clone" did not exist on "6dc4c36acbaddcfd7493f71d459ccf123c70e523"
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
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)
......
......@@ -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)
......
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