Skip to content

GitLab

"...metaportrait_pytorch.git" did not exist on "5cd956ef82be839a3161de3b722550ba8a637266"
Unverified Commit 9aedce99 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Update auto processor to check image processor created (#20021)

parent 49b77b89
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/auto/test_processor_auto.py
View file @ 9aedce99
...@@ -206,9 +206,9 @@ class AutoFeatureExtractorTest(unittest.TestCase): ...@@ -206,9 +206,9 @@ class AutoFeatureExtractorTest(unittest.TestCase):
processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert") processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert")
self.assertEqual(processor.__class__.__name__, "BertTokenizerFast") self.assertEqual(processor.__class__.__name__, "BertTokenizerFast")
def test_auto_processor_creates_feature_extractor(self): def test_auto_processor_creates_image_processor(self):
processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-convnext") processor = AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-convnext")
self.assertEqual(processor.__class__.__name__, "ConvNextFeatureExtractor") self.assertEqual(processor.__class__.__name__, "ConvNextImageProcessor")
@is_staging_test @is_staging_test
......
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