Skip to content

GitLab

Unverified Commit e1782653 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Add entries to `FEATURE_EXTRACTOR_MAPPING_NAMES` (#20551) 


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 8639cfb4
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
src/transformers/models/auto/feature_extraction_auto.py
View file @ e1782653
...@@ -73,11 +73,15 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict( ...@@ -73,11 +73,15 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
("regnet", "ConvNextFeatureExtractor"), ("regnet", "ConvNextFeatureExtractor"),
("resnet", "ConvNextFeatureExtractor"), ("resnet", "ConvNextFeatureExtractor"),
("segformer", "SegformerFeatureExtractor"), ("segformer", "SegformerFeatureExtractor"),
("sew", "Wav2Vec2FeatureExtractor"),
("sew-d", "Wav2Vec2FeatureExtractor"),
("speech_to_text", "Speech2TextFeatureExtractor"), ("speech_to_text", "Speech2TextFeatureExtractor"),
("swin", "ViTFeatureExtractor"), ("swin", "ViTFeatureExtractor"),
("swinv2", "ViTFeatureExtractor"), ("swinv2", "ViTFeatureExtractor"),
("table-transformer", "DetrFeatureExtractor"), ("table-transformer", "DetrFeatureExtractor"),
("timesformer", "VideoMAEFeatureExtractor"), ("timesformer", "VideoMAEFeatureExtractor"),
("unispeech", "Wav2Vec2FeatureExtractor"),
("unispeech-sat", "Wav2Vec2FeatureExtractor"),
("van", "ConvNextFeatureExtractor"), ("van", "ConvNextFeatureExtractor"),
("videomae", "VideoMAEFeatureExtractor"), ("videomae", "VideoMAEFeatureExtractor"),
("vilt", "ViltFeatureExtractor"), ("vilt", "ViltFeatureExtractor"),
...@@ -86,6 +90,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict( ...@@ -86,6 +90,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
("vit_msn", "ViTFeatureExtractor"), ("vit_msn", "ViTFeatureExtractor"),
("wav2vec2", "Wav2Vec2FeatureExtractor"), ("wav2vec2", "Wav2Vec2FeatureExtractor"),
("wav2vec2-conformer", "Wav2Vec2FeatureExtractor"), ("wav2vec2-conformer", "Wav2Vec2FeatureExtractor"),
("wavlm", "Wav2Vec2FeatureExtractor"),
("whisper", "WhisperFeatureExtractor"), ("whisper", "WhisperFeatureExtractor"),
("xclip", "CLIPFeatureExtractor"), ("xclip", "CLIPFeatureExtractor"),
("yolos", "YolosFeatureExtractor"), ("yolos", "YolosFeatureExtractor"),
......
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