Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ee8e80a060d65ab349743ffcb5842365eb0e5606"
Unverified Commit 303c1d69 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Update processor mapping for hub snippets (#27477)

parent 067c4a31
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
utils/update_metadata.py
View file @ 303c1d69
...@@ -197,9 +197,9 @@ def get_frameworks_table() -> pd.DataFrame: ...@@ -197,9 +197,9 @@ def get_frameworks_table() -> pd.DataFrame:
processors[t] = "AutoProcessor" processors[t] = "AutoProcessor"
elif t in transformers_module.models.auto.tokenization_auto.TOKENIZER_MAPPING_NAMES: elif t in transformers_module.models.auto.tokenization_auto.TOKENIZER_MAPPING_NAMES:
processors[t] = "AutoTokenizer" 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: 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" processors[t] = "AutoFeatureExtractor"
else: else:
# Default to AutoTokenizer if a model has nothing, for backward compatibility. # Default to AutoTokenizer if a model has nothing, for backward compatibility.
......
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