Add SegformerFeatureExtractor to Auto API (#15410)

ee5de663 · NielsRogge · GitHub · 0f69b924 · ee5de663
Unverified Commit ee5de663 authored Jan 31, 2022 by NielsRogge Committed by GitHub Jan 31, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/transformers/models/auto/feature_extraction_auto.py src/transformers/models/auto/feature_extraction_auto.py +1 -0

No files found.
--- a/src/transformers/models/auto/feature_extraction_auto.py
+++ b/src/transformers/models/auto/feature_extraction_auto.py
@@ -46,6 +46,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
        ("perceiver", "PerceiverFeatureExtractor"),
        ("swin", "ViTFeatureExtractor"),
        ("vit_mae", "ViTFeatureExtractor"),
+        ("segformer", "SegformerFeatureExtractor"),
    ]
 )