__init__.py 3.31 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 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,
17
    align,
Jongjyh's avatar
Jongjyh committed
18
    altclip,
19
    audio_spectrogram_transformer,
20
21
22
    auto,
    bart,
    barthez,
23
    bartpho,
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,
Kamal Raj Kanakarajan's avatar
Kamal Raj Kanakarajan committed
31
    biogpt,
NielsRogge's avatar
NielsRogge committed
32
    bit,
33
34
    blenderbot,
    blenderbot_small,
Younes Belkada's avatar
Younes Belkada committed
35
    blip,
NielsRogge's avatar
NielsRogge committed
36
    blip_2,
Younes Belkada's avatar
Younes Belkada committed
37
    bloom,
Lysandre Debut's avatar
Lysandre Debut committed
38
    bort,
39
    bridgetower,
40
    byt5,
41
    camembert,
NielsRogge's avatar
NielsRogge committed
42
    canine,
43
    chinese_clip,
44
    clap,
Suraj Patil's avatar
Suraj Patil committed
45
    clip,
NielsRogge's avatar
NielsRogge committed
46
    clipseg,
rooa's avatar
rooa committed
47
    codegen,
48
    conditional_detr,
abhishek thakur's avatar
abhishek thakur committed
49
    convbert,
NielsRogge's avatar
NielsRogge committed
50
    convnext,
Alara Dirik's avatar
Alara Dirik committed
51
    convnextv2,
52
    cpm,
53
    ctrl,
NielsRogge's avatar
NielsRogge committed
54
    cvt,
55
    data2vec,
56
    deberta,
57
    deberta_v2,
58
    decision_transformer,
NielsRogge's avatar
NielsRogge committed
59
    deformable_detr,
NielsRogge's avatar
NielsRogge committed
60
    deit,
NielsRogge's avatar
NielsRogge committed
61
    deta,
NielsRogge's avatar
NielsRogge committed
62
    detr,
63
    dialogpt,
64
    dinat,
65
    distilbert,
66
    dit,
NielsRogge's avatar
NielsRogge committed
67
    donut,
68
    dpr,
NielsRogge's avatar
NielsRogge committed
69
    dpt,
70
    efficientformer,
Alara Dirik's avatar
Alara Dirik committed
71
    efficientnet,
72
73
    electra,
    encoder_decoder,
74
    ernie,
75
    ernie_m,
76
    esm,
77
    flaubert,
78
    flava,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
79
    fnet,
80
81
    fsmt,
    funnel,
82
    git,
NielsRogge's avatar
NielsRogge committed
83
    glpn,
84
    gpt2,
Suraj Patil's avatar
Suraj Patil committed
85
    gpt_neo,
86
    gpt_neox,
87
    gpt_neox_japanese,
88
    gpt_sw3,
Stella Biderman's avatar
Stella Biderman committed
89
    gptj,
90
    gptsan_japanese,
91
    graphormer,
92
    groupvit,
93
    herbert,
94
95
    hubert,
    ibert,
NielsRogge's avatar
NielsRogge committed
96
    imagegpt,
97
    informer,
98
    jukebox,
99
    layoutlm,
100
    layoutlmv2,
NielsRogge's avatar
NielsRogge committed
101
    layoutlmv3,
102
    layoutxlm,
103
    led,
104
    levit,
NielsRogge's avatar
NielsRogge committed
105
    lilt,
Jason Phang's avatar
Jason Phang committed
106
    llama,
107
    longformer,
Daniel Stancl's avatar
Daniel Stancl committed
108
    longt5,
NielsRogge's avatar
NielsRogge committed
109
    luke,
110
    lxmert,
Suraj Patil's avatar
Suraj Patil committed
111
    m2m_100,
112
    marian,
NielsRogge's avatar
NielsRogge committed
113
    markuplm,
Alara Dirik's avatar
Alara Dirik committed
114
    mask2former,
115
    maskformer,
116
    mbart,
117
    mbart50,
Chan Woo Kim's avatar
Chan Woo Kim committed
118
    mctct,
119
    mega,
120
    megatron_bert,
121
    megatron_gpt2,
wangpeng's avatar
wangpeng committed
122
    mgp_str,
Ryokan RI's avatar
Ryokan RI committed
123
    mluke,
124
125
    mmbt,
    mobilebert,
126
    mobilenet_v1,
127
    mobilenet_v2,
128
    mobilevit,
129
130
    mpnet,
    mt5,
StevenTang1998's avatar
StevenTang1998 committed
131
    mvp,
132
    nat,
133
    nezha,
Lysandre Debut's avatar
Lysandre Debut committed
134
    nllb,
novice's avatar
novice committed
135
    nystromformer,
Jitesh Jain's avatar
Jitesh Jain committed
136
    oneformer,
137
    openai,
Younes Belkada's avatar
Younes Belkada committed
138
    opt,
139
    owlvit,
140
    pegasus,
Jason Phang's avatar
Jason Phang committed
141
    pegasus_x,
NielsRogge's avatar
NielsRogge committed
142
    perceiver,
143
    phobert,
Younes Belkada's avatar
Younes Belkada committed
144
    pix2struct,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
145
    plbart,
Tanay Mehta's avatar
Tanay Mehta committed
146
    poolformer,
147
    prophetnet,
148
    qdqbert,
149
    rag,
Li-Huai (Allan) Lin's avatar
Li-Huai (Allan) Lin committed
150
    realm,
151
    reformer,
Francesco Saverio Zuppichini's avatar
Francesco Saverio Zuppichini committed
152
    regnet,
153
    rembert,
154
    resnet,
155
156
    retribert,
    roberta,
157
    roberta_prelayernorm,
Weiwe Shi's avatar
Weiwe Shi committed
158
    roc_bert,
159
    roformer,
NielsRogge's avatar
NielsRogge committed
160
    segformer,
161
162
    sew,
    sew_d,
163
    speech_encoder_decoder,
Suraj Patil's avatar
Suraj Patil committed
164
    speech_to_text,
165
    speech_to_text_2,
166
    speecht5,
Ori Ram's avatar
Ori Ram committed
167
    splinter,
168
    squeezebert,
novice's avatar
novice committed
169
    swin,
NielsRogge's avatar
NielsRogge committed
170
    swin2sr,
171
    swinv2,
172
    switch_transformers,
173
    t5,
174
    table_transformer,
175
    tapas,
NielsRogge's avatar
NielsRogge committed
176
    tapex,
177
    time_series_transformer,
178
    timesformer,
Carl's avatar
Carl committed
179
    trajectory_transformer,
180
    transfo_xl,
181
    trocr,
Zineng Tang's avatar
Zineng Tang committed
182
    tvlt,
183
184
    unispeech,
    unispeech_sat,
NielsRogge's avatar
NielsRogge committed
185
    upernet,
186
    van,
NielsRogge's avatar
NielsRogge committed
187
    videomae,
NielsRogge's avatar
NielsRogge committed
188
    vilt,
189
    vision_encoder_decoder,
Suraj Patil's avatar
Suraj Patil committed
190
    vision_text_dual_encoder,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
191
    visual_bert,
192
    vit,
NielsRogge's avatar
NielsRogge committed
193
    vit_hybrid,
NielsRogge's avatar
NielsRogge committed
194
    vit_mae,
195
    vit_msn,
Patrick von Platen's avatar
Patrick von Platen committed
196
    wav2vec2,
197
    wav2vec2_conformer,
198
    wav2vec2_phoneme,
199
    wav2vec2_with_lm,
Patrick von Platen's avatar
Patrick von Platen committed
200
    wavlm,
201
    whisper,
NielsRogge's avatar
NielsRogge committed
202
    x_clip,
Suraj Patil's avatar
Suraj Patil committed
203
    xglm,
204
    xlm,
205
    xlm_prophetnet,
206
    xlm_roberta,
207
    xlm_roberta_xl,
208
    xlnet,
Jannis Vamvas's avatar
Jannis Vamvas committed
209
    xmod,
NielsRogge's avatar
NielsRogge committed
210
    yolos,
novice's avatar
novice committed
211
    yoso,
212
)