diff --git a/src/transformers/models/auto/feature_extraction_auto.py b/src/transformers/models/auto/feature_extraction_auto.py index 54c03a3dcb58..0096ef157a9d 100644 --- a/src/transformers/models/auto/feature_extraction_auto.py +++ b/src/transformers/models/auto/feature_extraction_auto.py @@ -34,6 +34,7 @@ [ ("beit", "BeitFeatureExtractor"), ("deit", "DeiTFeatureExtractor"), + ("hubert", "Wav2Vec2FeatureExtractor"), ("speech_to_text", "Speech2TextFeatureExtractor"), ("vit", "ViTFeatureExtractor"), ("wav2vec2", "Wav2Vec2FeatureExtractor"), diff --git a/src/transformers/models/hubert/__init__.py b/src/transformers/models/hubert/__init__.py index f62cc14bd76d..1d88b206a47a 100644 --- a/src/transformers/models/hubert/__init__.py +++ b/src/transformers/models/hubert/__init__.py @@ -21,6 +21,7 @@ _import_structure = { + ".wav2vec2.feature_extraction_wav2vec2": ["Wav2Vec2FeatureExtractor"], "configuration_hubert": ["HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "HubertConfig"], } @@ -43,6 +44,7 @@ ] if TYPE_CHECKING: + from ..wav2vec2.feature_extraction_wav2vec2 import Wav2Vec2FeatureExtractor from .configuration_hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig if is_torch_available():