__init__.py 1.52 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# 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,
Vasudev Gupta's avatar
Vasudev Gupta committed
28
    big_bird,
29
30
31
    blenderbot,
    blenderbot_small,
    camembert,
abhishek thakur's avatar
abhishek thakur committed
32
    convbert,
33
    cpm,
34
35
36
37
38
39
40
41
42
43
44
    ctrl,
    deberta,
    dialogpt,
    distilbert,
    dpr,
    electra,
    encoder_decoder,
    flaubert,
    fsmt,
    funnel,
    gpt2,
Suraj Patil's avatar
Suraj Patil committed
45
    gpt_neo,
46
47
48
49
50
    herbert,
    layoutlm,
    led,
    longformer,
    lxmert,
Suraj Patil's avatar
Suraj Patil committed
51
    m2m_100,
52
53
    marian,
    mbart,
54
    megatron_bert,
55
56
57
58
59
60
61
62
63
64
65
66
    mmbt,
    mobilebert,
    mpnet,
    mt5,
    openai,
    pegasus,
    phobert,
    prophetnet,
    rag,
    reformer,
    retribert,
    roberta,
Suraj Patil's avatar
Suraj Patil committed
67
    speech_to_text,
68
69
70
71
    squeezebert,
    t5,
    tapas,
    transfo_xl,
72
    vit,
Patrick von Platen's avatar
Patrick von Platen committed
73
    wav2vec2,
74
75
76
77
    xlm,
    xlm_roberta,
    xlnet,
)