"src/nni_manager/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "6568eaee0f6ce42a9d19215dd6a5ecca1d96652e"
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(
[
("beit", "BeitFeatureExtractor"),
("deit", "DeiTFeatureExtractor"),
("hubert", "Wav2Vec2FeatureExtractor"),
("speech_to_text", "Speech2TextFeatureExtractor"),
("vit", "ViTFeatureExtractor"),
("wav2vec2", "Wav2Vec2FeatureExtractor"),
......
......@@ -21,6 +21,7 @@ from ...file_utils import _LazyModule, is_tf_available, is_torch_available
_import_structure = {
".wav2vec2.feature_extraction_wav2vec2": ["Wav2Vec2FeatureExtractor"],
"configuration_hubert": ["HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "HubertConfig"],
}
......@@ -43,6 +44,7 @@ if is_tf_available():
]
if TYPE_CHECKING:
from ..wav2vec2.feature_extraction_wav2vec2 import Wav2Vec2FeatureExtractor
from .configuration_hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig
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