Unverified Commit 2406892a authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Add `Hubert` to the `AutoFeatureExtractor` (#13366)

* Add Hubert to the auto feature extractor

* Fix import structure
parent 6b353264
...@@ -34,6 +34,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict( ...@@ -34,6 +34,7 @@ FEATURE_EXTRACTOR_MAPPING_NAMES = OrderedDict(
[ [
("beit", "BeitFeatureExtractor"), ("beit", "BeitFeatureExtractor"),
("deit", "DeiTFeatureExtractor"), ("deit", "DeiTFeatureExtractor"),
("hubert", "Wav2Vec2FeatureExtractor"),
("speech_to_text", "Speech2TextFeatureExtractor"), ("speech_to_text", "Speech2TextFeatureExtractor"),
("vit", "ViTFeatureExtractor"), ("vit", "ViTFeatureExtractor"),
("wav2vec2", "Wav2Vec2FeatureExtractor"), ("wav2vec2", "Wav2Vec2FeatureExtractor"),
......
...@@ -21,6 +21,7 @@ from ...file_utils import _LazyModule, is_tf_available, is_torch_available ...@@ -21,6 +21,7 @@ from ...file_utils import _LazyModule, is_tf_available, is_torch_available
_import_structure = { _import_structure = {
".wav2vec2.feature_extraction_wav2vec2": ["Wav2Vec2FeatureExtractor"],
"configuration_hubert": ["HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "HubertConfig"], "configuration_hubert": ["HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "HubertConfig"],
} }
...@@ -43,6 +44,7 @@ if is_tf_available(): ...@@ -43,6 +44,7 @@ if is_tf_available():
] ]
if TYPE_CHECKING: if TYPE_CHECKING:
from ..wav2vec2.feature_extraction_wav2vec2 import Wav2Vec2FeatureExtractor
from .configuration_hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig from .configuration_hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig
if is_torch_available(): if is_torch_available():
......
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