__init__.py 3.79 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
    auto,
21
    autoformer,
Yoach Lacombe's avatar
Yoach Lacombe committed
22
    bark,
23
24
    bart,
    barthez,
25
    bartpho,
NielsRogge's avatar
NielsRogge committed
26
    beit,
27
28
29
30
    bert,
    bert_generation,
    bert_japanese,
    bertweet,
Vasudev Gupta's avatar
Vasudev Gupta committed
31
    big_bird,
Vasudev Gupta's avatar
Vasudev Gupta committed
32
    bigbird_pegasus,
Kamal Raj Kanakarajan's avatar
Kamal Raj Kanakarajan committed
33
    biogpt,
NielsRogge's avatar
NielsRogge committed
34
    bit,
35
36
    blenderbot,
    blenderbot_small,
Younes Belkada's avatar
Younes Belkada committed
37
    blip,
NielsRogge's avatar
NielsRogge committed
38
    blip_2,
Younes Belkada's avatar
Younes Belkada committed
39
    bloom,
40
    bridgetower,
Jinho Park's avatar
Jinho Park committed
41
    bros,
42
    byt5,
43
    camembert,
NielsRogge's avatar
NielsRogge committed
44
    canine,
45
    chinese_clip,
46
    clap,
Suraj Patil's avatar
Suraj Patil committed
47
    clip,
NielsRogge's avatar
NielsRogge committed
48
    clipseg,
Susnato Dhar's avatar
Susnato Dhar committed
49
    clvp,
50
    code_llama,
rooa's avatar
rooa committed
51
    codegen,
52
    conditional_detr,
abhishek thakur's avatar
abhishek thakur committed
53
    convbert,
NielsRogge's avatar
NielsRogge committed
54
    convnext,
Alara Dirik's avatar
Alara Dirik committed
55
    convnextv2,
56
    cpm,
57
    cpmant,
58
    ctrl,
NielsRogge's avatar
NielsRogge committed
59
    cvt,
60
    data2vec,
61
    deberta,
62
    deberta_v2,
63
    decision_transformer,
NielsRogge's avatar
NielsRogge committed
64
    deformable_detr,
NielsRogge's avatar
NielsRogge committed
65
    deit,
Sylvain Gugger's avatar
Sylvain Gugger committed
66
    deprecated,
NielsRogge's avatar
NielsRogge committed
67
    deta,
NielsRogge's avatar
NielsRogge committed
68
    detr,
69
    dialogpt,
70
    dinat,
NielsRogge's avatar
NielsRogge committed
71
    dinov2,
72
    distilbert,
73
    dit,
NielsRogge's avatar
NielsRogge committed
74
    donut,
75
    dpr,
NielsRogge's avatar
NielsRogge committed
76
    dpt,
77
    efficientformer,
Alara Dirik's avatar
Alara Dirik committed
78
    efficientnet,
79
    electra,
80
    encodec,
81
    encoder_decoder,
82
    ernie,
83
    ernie_m,
84
    esm,
Matt's avatar
Matt committed
85
    falcon,
86
    flaubert,
87
    flava,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
88
    fnet,
NielsRogge's avatar
NielsRogge committed
89
    focalnet,
90
91
    fsmt,
    funnel,
Pablo Montalvo's avatar
Pablo Montalvo committed
92
    fuyu,
93
    git,
NielsRogge's avatar
NielsRogge committed
94
    glpn,
95
    gpt2,
96
    gpt_bigcode,
Suraj Patil's avatar
Suraj Patil committed
97
    gpt_neo,
98
    gpt_neox,
99
    gpt_neox_japanese,
100
    gpt_sw3,
Stella Biderman's avatar
Stella Biderman committed
101
    gptj,
102
    gptsan_japanese,
103
    graphormer,
104
    groupvit,
105
    herbert,
106
107
    hubert,
    ibert,
108
    idefics,
NielsRogge's avatar
NielsRogge committed
109
    imagegpt,
110
    informer,
NielsRogge's avatar
NielsRogge committed
111
    instructblip,
112
    jukebox,
Yih-Dar's avatar
Yih-Dar committed
113
    kosmos2,
114
    layoutlm,
115
    layoutlmv2,
NielsRogge's avatar
NielsRogge committed
116
    layoutlmv3,
117
    layoutxlm,
118
    led,
119
    levit,
NielsRogge's avatar
NielsRogge committed
120
    lilt,
Jason Phang's avatar
Jason Phang committed
121
    llama,
122
    llava,
123
    longformer,
Daniel Stancl's avatar
Daniel Stancl committed
124
    longt5,
NielsRogge's avatar
NielsRogge committed
125
    luke,
126
    lxmert,
Suraj Patil's avatar
Suraj Patil committed
127
    m2m_100,
128
    marian,
NielsRogge's avatar
NielsRogge committed
129
    markuplm,
Alara Dirik's avatar
Alara Dirik committed
130
    mask2former,
131
    maskformer,
132
    mbart,
133
    mbart50,
134
    mega,
135
    megatron_bert,
136
    megatron_gpt2,
wangpeng's avatar
wangpeng committed
137
    mgp_str,
138
    mistral,
139
    mixtral,
Ryokan RI's avatar
Ryokan RI committed
140
    mluke,
141
    mobilebert,
142
    mobilenet_v1,
143
    mobilenet_v2,
144
    mobilevit,
Shehan Munasinghe's avatar
Shehan Munasinghe committed
145
    mobilevitv2,
146
    mpnet,
147
    mpt,
148
    mra,
149
    mt5,
Sanchit Gandhi's avatar
Sanchit Gandhi committed
150
    musicgen,
StevenTang1998's avatar
StevenTang1998 committed
151
    mvp,
152
    nat,
153
    nezha,
Lysandre Debut's avatar
Lysandre Debut committed
154
    nllb,
155
    nllb_moe,
NielsRogge's avatar
NielsRogge committed
156
    nougat,
novice's avatar
novice committed
157
    nystromformer,
Jitesh Jain's avatar
Jitesh Jain committed
158
    oneformer,
159
    openai,
Younes Belkada's avatar
Younes Belkada committed
160
    opt,
NielsRogge's avatar
NielsRogge committed
161
    owlv2,
162
    owlvit,
163
    patchtsmixer,
164
    patchtst,
165
    pegasus,
Jason Phang's avatar
Jason Phang committed
166
    pegasus_x,
NielsRogge's avatar
NielsRogge committed
167
    perceiver,
168
    persimmon,
Susnato Dhar's avatar
Susnato Dhar committed
169
    phi,
170
    phobert,
Younes Belkada's avatar
Younes Belkada committed
171
    pix2struct,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
172
    plbart,
Tanay Mehta's avatar
Tanay Mehta committed
173
    poolformer,
Susnato Dhar's avatar
Susnato Dhar committed
174
    pop2piano,
175
    prophetnet,
Rinat's avatar
Rinat committed
176
    pvt,
177
    qdqbert,
178
    rag,
Li-Huai (Allan) Lin's avatar
Li-Huai (Allan) Lin committed
179
    realm,
180
    reformer,
Francesco Saverio Zuppichini's avatar
Francesco Saverio Zuppichini committed
181
    regnet,
182
    rembert,
183
    resnet,
184
    roberta,
185
    roberta_prelayernorm,
Weiwe Shi's avatar
Weiwe Shi committed
186
    roc_bert,
187
    roformer,
Sylvain Gugger's avatar
Sylvain Gugger committed
188
    rwkv,
189
    sam,
190
    seamless_m4t,
Yoach Lacombe's avatar
Yoach Lacombe committed
191
    seamless_m4t_v2,
NielsRogge's avatar
NielsRogge committed
192
    segformer,
193
194
    sew,
    sew_d,
195
    speech_encoder_decoder,
Suraj Patil's avatar
Suraj Patil committed
196
    speech_to_text,
197
    speech_to_text_2,
198
    speecht5,
Ori Ram's avatar
Ori Ram committed
199
    splinter,
200
    squeezebert,
Shehan Munasinghe's avatar
Shehan Munasinghe committed
201
    swiftformer,
novice's avatar
novice committed
202
    swin,
NielsRogge's avatar
NielsRogge committed
203
    swin2sr,
204
    swinv2,
205
    switch_transformers,
206
    t5,
207
    table_transformer,
208
    tapas,
209
    time_series_transformer,
210
    timesformer,
amyeroberts's avatar
amyeroberts committed
211
    timm_backbone,
212
    trocr,
Zineng Tang's avatar
Zineng Tang committed
213
    tvlt,
jiqing-feng's avatar
jiqing-feng committed
214
    tvp,
215
    umt5,
216
217
    unispeech,
    unispeech_sat,
218
    univnet,
NielsRogge's avatar
NielsRogge committed
219
    upernet,
NielsRogge's avatar
NielsRogge committed
220
    videomae,
NielsRogge's avatar
NielsRogge committed
221
    vilt,
222
    vipllava,
223
    vision_encoder_decoder,
Suraj Patil's avatar
Suraj Patil committed
224
    vision_text_dual_encoder,
Gunjan Chhablani's avatar
Gunjan Chhablani committed
225
    visual_bert,
226
    vit,
NielsRogge's avatar
NielsRogge committed
227
    vit_hybrid,
NielsRogge's avatar
NielsRogge committed
228
    vit_mae,
229
    vit_msn,
NielsRogge's avatar
NielsRogge committed
230
    vitdet,
NielsRogge's avatar
NielsRogge committed
231
    vitmatte,
Matthijs Hollemans's avatar
Matthijs Hollemans committed
232
    vits,
Jegor Kit拧kerkin's avatar
Jegor Kit拧kerkin committed
233
    vivit,
Patrick von Platen's avatar
Patrick von Platen committed
234
    wav2vec2,
235
    wav2vec2_conformer,
236
    wav2vec2_phoneme,
237
    wav2vec2_with_lm,
Patrick von Platen's avatar
Patrick von Platen committed
238
    wavlm,
239
    whisper,
NielsRogge's avatar
NielsRogge committed
240
    x_clip,
Suraj Patil's avatar
Suraj Patil committed
241
    xglm,
242
    xlm,
243
    xlm_prophetnet,
244
    xlm_roberta,
245
    xlm_roberta_xl,
246
    xlnet,
Jannis Vamvas's avatar
Jannis Vamvas committed
247
    xmod,
NielsRogge's avatar
NielsRogge committed
248
    yolos,
novice's avatar
novice committed
249
    yoso,
250
)