d6217fb30c
* add raw scaffold * implement feat extract layers * make style * remove + * correctly convert weights * make feat extractor work * make feature extraction proj work * run forward pass * finish forward pass * Succesful decoding example * remove unused files * more changes * add wav2vec tokenizer * add new structure * fix run forward * add other layer norm architecture * finish 2nd structure * add model tests * finish tests for tok and model * clean-up * make style * finish docstring for model and config * make style * correct docstring * correct tests * change checkpoints to fairseq * fix examples * finish wav2vec2 * make style * apply sylvains suggestions * apply lysandres suggestions * change print to log.info * re-add assert statement * add input_values as required input name * finish wav2vec2 tokenizer * Update tests/test_tokenization_wav2vec2.py Co-authored-by: Lysandre Debut <lysandre@huggingface.co> * apply sylvains suggestions Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
71 lines
1.4 KiB
Python
71 lines
1.4 KiB
Python
# flake8: noqa
|
|
# There's no way to ignore "F401 '...' imported but unused" warnings in this
|
|
# module, but to preserve other warnings. So, don't check this module at all.
|
|
|
|
# Copyright 2020 The HuggingFace Team. All rights reserved.
|
|
#
|
|
# 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.
|
|
|
|
from . import (
|
|
albert,
|
|
auto,
|
|
bart,
|
|
barthez,
|
|
bert,
|
|
bert_generation,
|
|
bert_japanese,
|
|
bertweet,
|
|
blenderbot,
|
|
blenderbot_small,
|
|
camembert,
|
|
convbert,
|
|
ctrl,
|
|
deberta,
|
|
dialogpt,
|
|
distilbert,
|
|
dpr,
|
|
electra,
|
|
encoder_decoder,
|
|
flaubert,
|
|
fsmt,
|
|
funnel,
|
|
gpt2,
|
|
herbert,
|
|
layoutlm,
|
|
led,
|
|
longformer,
|
|
lxmert,
|
|
marian,
|
|
mbart,
|
|
mmbt,
|
|
mobilebert,
|
|
mpnet,
|
|
mt5,
|
|
openai,
|
|
pegasus,
|
|
phobert,
|
|
prophetnet,
|
|
rag,
|
|
reformer,
|
|
retribert,
|
|
roberta,
|
|
squeezebert,
|
|
t5,
|
|
tapas,
|
|
transfo_xl,
|
|
wav2vec2,
|
|
xlm,
|
|
xlm_roberta,
|
|
xlnet,
|
|
)
|