From 303c1d69f3efa44b565bc5618ff819daf4a0e80f Mon Sep 17 00:00:00 2001 From: amyeroberts <22614925+amyeroberts@users.noreply.github.com> Date: Tue, 14 Nov 2023 20:05:54 +0000 Subject: [PATCH] Update processor mapping for hub snippets (#27477) --- utils/update_metadata.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/update_metadata.py b/utils/update_metadata.py index 9a233a082..2104d53b6 100644 --- a/utils/update_metadata.py +++ b/utils/update_metadata.py @@ -197,9 +197,9 @@ def get_frameworks_table() -> pd.DataFrame: processors[t] = "AutoProcessor" elif t in transformers_module.models.auto.tokenization_auto.TOKENIZER_MAPPING_NAMES: processors[t] = "AutoTokenizer" - elif t in transformers_module.models.auto.feature_extraction_auto.FEATURE_EXTRACTOR_MAPPING_NAMES: - processors[t] = "AutoFeatureExtractor" elif t in transformers_module.models.auto.image_processing_auto.IMAGE_PROCESSOR_MAPPING_NAMES: + processors[t] = "AutoImageProcessor" + elif t in transformers_module.models.auto.feature_extraction_auto.FEATURE_EXTRACTOR_MAPPING_NAMES: processors[t] = "AutoFeatureExtractor" else: # Default to AutoTokenizer if a model has nothing, for backward compatibility.