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

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

This reverts commit 8f400775fc.

* Use `tiny-detr` checkpts from `hf-internal-testing`
This commit is contained in:
Mishig Davaadorj
2022-06-29 16:19:37 +02:00
committed by GitHub
parent 3cff4cc587
commit 77b76672e2
2 changed files with 2 additions and 4 deletions

View File

@@ -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)