Add missing feature extractors (#13374)

2021-09-01 17:10:49 +02:00 · 2021-09-01 17:10:49 +02:00 · 33b7c9a8aa
parent 2406892a2e
commit 33b7c9a8aa
1 changed files with 3 additions and 0 deletions
--- a/src/transformers/models/auto/feature_extraction_auto.py
+++ b/src/transformers/models/auto/feature_extraction_auto.py
@ -38,6 +38,9 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
        ("speech_to_text", "Speech2TextFeatureExtractor"),
        ("vit", "ViTFeatureExtractor"),
        ("wav2vec2", "Wav2Vec2FeatureExtractor"),
+        ("detr", "DetrFeatureExtractor"),
+        ("layoutlmv2", "LayoutLMv2FeatureExtractor"),
+        ("clip", "ClipFeatureExtractor"),
    ]
 )