__init__.py 1.64 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
# 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,
NielsRogge's avatar
NielsRogge committed
24
    beit,
25
26
27
28
    bert,
    bert_generation,
    bert_japanese,
    bertweet,
Vasudev Gupta's avatar
Vasudev Gupta committed
29
    big_bird,
Vasudev Gupta's avatar
Vasudev Gupta committed
30
    bigbird_pegasus,
31
32
33
    blenderbot,
    blenderbot_small,
    camembert,
NielsRogge's avatar
NielsRogge committed
34
    canine,
Suraj Patil's avatar
Suraj Patil committed
35
    clip,
abhishek thakur's avatar
abhishek thakur committed
36
    convbert,
37
    cpm,
38
39
    ctrl,
    deberta,
NielsRogge's avatar
NielsRogge committed
40
    deit,
NielsRogge's avatar
NielsRogge committed
41
    detr,
42
43
44
45
46
47
48
49
50
    dialogpt,
    distilbert,
    dpr,
    electra,
    encoder_decoder,
    flaubert,
    fsmt,
    funnel,
    gpt2,
Suraj Patil's avatar
Suraj Patil committed
51
    gpt_neo,
52
53
54
55
    herbert,
    layoutlm,
    led,
    longformer,
NielsRogge's avatar
NielsRogge committed
56
    luke,
57
    lxmert,
Suraj Patil's avatar
Suraj Patil committed
58
    m2m_100,
59
60
    marian,
    mbart,
61
    megatron_bert,
62
63
64
65
66
67
68
69
70
71
    mmbt,
    mobilebert,
    mpnet,
    mt5,
    openai,
    pegasus,
    phobert,
    prophetnet,
    rag,
    reformer,
72
    rembert,
73
74
    retribert,
    roberta,
75
    roformer,
Suraj Patil's avatar
Suraj Patil committed
76
    speech_to_text,
77
78
79
80
    squeezebert,
    t5,
    tapas,
    transfo_xl,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
81
    visual_bert,
82
    vit,
Patrick von Platen's avatar
Patrick von Platen committed
83
    wav2vec2,
84
85
86
87
    xlm,
    xlm_roberta,
    xlnet,
)