Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "3ae2e86baffc1fea8b8b93695fb5a10941fd63dc"
Unverified Commit ee5de663 authored by NielsRogge's avatar NielsRogge Committed by GitHub
Browse files

Add SegformerFeatureExtractor to Auto API (#15410)

parent 0f69b924
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/models/auto/feature_extraction_auto.py
View file @ ee5de663
...@@ -46,6 +46,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict( ...@@ -46,6 +46,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
("perceiver", "PerceiverFeatureExtractor"), ("perceiver", "PerceiverFeatureExtractor"),
("swin", "ViTFeatureExtractor"), ("swin", "ViTFeatureExtractor"),
("vit_mae", "ViTFeatureExtractor"), ("vit_mae", "ViTFeatureExtractor"),
("segformer", "SegformerFeatureExtractor"),
] ]
) )
......
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