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