From a4c0a09ab4fe58cd49c68b4762b1412a59252ee6 Mon Sep 17 00:00:00 2001 From: anton-l Date: Wed, 1 Sep 2021 13:09:22 +0300 Subject: [PATCH 1/2] Add Hubert to the auto feature extractor --- src/transformers/models/auto/feature_extraction_auto.py | 1 + 1 file changed, 1 insertion(+) 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"), From b61ee662e9d2fd0fe44e0360fb88791ee848c511 Mon Sep 17 00:00:00 2001 From: anton-l Date: Wed, 1 Sep 2021 15:42:17 +0300 Subject: [PATCH 2/2] Fix import structure --- src/transformers/models/hubert/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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():