From cca240e86de956938bdd0ad415fd8b026fe17f09 Mon Sep 17 00:00:00 2001 From: Kahsolt Date: Sun, 4 Feb 2024 16:11:21 +0800 Subject: [PATCH] init Hubert model --- mindnlp/transformers/models/__init__.py | 4 +++- mindnlp/transformers/models/hubert/__init__.py | 17 +++++++++++++++++ 2 files changed, 20 insertions(+), 1 deletion(-) create mode 100644 mindnlp/transformers/models/hubert/__init__.py diff --git a/mindnlp/transformers/models/__init__.py b/mindnlp/transformers/models/__init__.py index dc128652..f5324de6 100644 --- a/mindnlp/transformers/models/__init__.py +++ b/mindnlp/transformers/models/__init__.py @@ -18,7 +18,7 @@ Models init from . import albert, auto, bart, bert, bloom, clip, codegen, cpm, electra, ernie, ernie_m, chatglm, chatglm2, gpt, gpt_neo, gpt_neox, gpt2, \ llama, longformer, luke, mbart, megatron_bert, mistral, mobilebert, nezha, opt, gpt_pangu, phi, roberta, rwkv, \ seamless_m4t, seamless_m4t_v2, t5, tinybert, xlm, xlm_roberta, gpt_bigcode, whisper, bark, encodec, \ - graphormer, pop2piano + graphormer, pop2piano, hubert from .albert import * from .auto import * from .bark import * @@ -61,6 +61,7 @@ from .gpt_bigcode import * from .whisper import * from .encodec import * from .pop2piano import * +from .hubert import * __all__ = [] __all__.extend(albert.__all__) @@ -105,3 +106,4 @@ __all__.extend(whisper.__all__) __all__.extend(bark.__all__) __all__.extend(encodec.__all__) __all__.extend(pop2piano.__all__) +__all__.extend(hubert.__all__) diff --git a/mindnlp/transformers/models/hubert/__init__.py b/mindnlp/transformers/models/hubert/__init__.py new file mode 100644 index 00000000..cc0ade79 --- /dev/null +++ b/mindnlp/transformers/models/hubert/__init__.py @@ -0,0 +1,17 @@ +# Copyright 2024 Huawei Technologies Co., Ltd +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================ +""" +Hubert Model. +""" -- Gitee