Skip to content

GitLab

Unverified Commit 33b7c9a8 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Add missing feature extractors (#13374)

parent 2406892a
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/transformers/models/auto/feature_extraction_auto.py
View file @ 33b7c9a8
...@@ -38,6 +38,9 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict( ...@@ -38,6 +38,9 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
("speech_to_text", "Speech2TextFeatureExtractor"), ("speech_to_text", "Speech2TextFeatureExtractor"),
("vit", "ViTFeatureExtractor"), ("vit", "ViTFeatureExtractor"),
("wav2vec2", "Wav2Vec2FeatureExtractor"), ("wav2vec2", "Wav2Vec2FeatureExtractor"),
("detr", "DetrFeatureExtractor"),
("layoutlmv2", "LayoutLMv2FeatureExtractor"),
("clip", "ClipFeatureExtractor"),
] ]
) )
......
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