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