__init__.py 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289
  1. # Copyright 2020 The HuggingFace Team. All rights reserved.
  2. #
  3. # Licensed under the Apache License, Version 2.0 (the "License");
  4. # you may not use this file except in compliance with the License.
  5. # You may obtain a copy of the License at
  6. #
  7. # http://www.apache.org/licenses/LICENSE-2.0
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. from . import (
  15. albert,
  16. align,
  17. altclip,
  18. audio_spectrogram_transformer,
  19. auto,
  20. autoformer,
  21. bark,
  22. bart,
  23. barthez,
  24. bartpho,
  25. beit,
  26. bert,
  27. bert_generation,
  28. bert_japanese,
  29. bertweet,
  30. big_bird,
  31. bigbird_pegasus,
  32. biogpt,
  33. bit,
  34. blenderbot,
  35. blenderbot_small,
  36. blip,
  37. blip_2,
  38. bloom,
  39. bridgetower,
  40. bros,
  41. byt5,
  42. camembert,
  43. canine,
  44. chameleon,
  45. chinese_clip,
  46. clap,
  47. clip,
  48. clipseg,
  49. clvp,
  50. code_llama,
  51. codegen,
  52. cohere,
  53. conditional_detr,
  54. convbert,
  55. convnext,
  56. convnextv2,
  57. cpm,
  58. cpmant,
  59. ctrl,
  60. cvt,
  61. dac,
  62. data2vec,
  63. dbrx,
  64. deberta,
  65. deberta_v2,
  66. decision_transformer,
  67. deformable_detr,
  68. deit,
  69. deprecated,
  70. depth_anything,
  71. detr,
  72. dialogpt,
  73. dinat,
  74. dinov2,
  75. distilbert,
  76. dit,
  77. donut,
  78. dpr,
  79. dpt,
  80. efficientnet,
  81. electra,
  82. encodec,
  83. encoder_decoder,
  84. ernie,
  85. esm,
  86. falcon,
  87. falcon_mamba,
  88. fastspeech2_conformer,
  89. flaubert,
  90. flava,
  91. fnet,
  92. focalnet,
  93. fsmt,
  94. funnel,
  95. fuyu,
  96. gemma,
  97. gemma2,
  98. git,
  99. glm,
  100. glpn,
  101. gpt2,
  102. gpt_bigcode,
  103. gpt_neo,
  104. gpt_neox,
  105. gpt_neox_japanese,
  106. gpt_sw3,
  107. gptj,
  108. granite,
  109. granitemoe,
  110. grounding_dino,
  111. groupvit,
  112. herbert,
  113. hiera,
  114. hubert,
  115. ibert,
  116. idefics,
  117. idefics2,
  118. idefics3,
  119. imagegpt,
  120. informer,
  121. instructblip,
  122. instructblipvideo,
  123. jamba,
  124. jetmoe,
  125. kosmos2,
  126. layoutlm,
  127. layoutlmv2,
  128. layoutlmv3,
  129. layoutxlm,
  130. led,
  131. levit,
  132. lilt,
  133. llama,
  134. llava,
  135. llava_next,
  136. llava_next_video,
  137. llava_onevision,
  138. longformer,
  139. longt5,
  140. luke,
  141. lxmert,
  142. m2m_100,
  143. mamba,
  144. mamba2,
  145. marian,
  146. markuplm,
  147. mask2former,
  148. maskformer,
  149. mbart,
  150. mbart50,
  151. megatron_bert,
  152. megatron_gpt2,
  153. mgp_str,
  154. mimi,
  155. mistral,
  156. mixtral,
  157. mllama,
  158. mluke,
  159. mobilebert,
  160. mobilenet_v1,
  161. mobilenet_v2,
  162. mobilevit,
  163. mobilevitv2,
  164. moshi,
  165. mpnet,
  166. mpt,
  167. mra,
  168. mt5,
  169. musicgen,
  170. musicgen_melody,
  171. mvp,
  172. myt5,
  173. nemotron,
  174. nllb,
  175. nllb_moe,
  176. nougat,
  177. nystromformer,
  178. olmo,
  179. olmoe,
  180. omdet_turbo,
  181. oneformer,
  182. openai,
  183. opt,
  184. owlv2,
  185. owlvit,
  186. paligemma,
  187. patchtsmixer,
  188. patchtst,
  189. pegasus,
  190. pegasus_x,
  191. perceiver,
  192. persimmon,
  193. phi,
  194. phi3,
  195. phimoe,
  196. phobert,
  197. pix2struct,
  198. pixtral,
  199. plbart,
  200. poolformer,
  201. pop2piano,
  202. prophetnet,
  203. pvt,
  204. pvt_v2,
  205. qwen2,
  206. qwen2_audio,
  207. qwen2_moe,
  208. qwen2_vl,
  209. rag,
  210. recurrent_gemma,
  211. reformer,
  212. regnet,
  213. rembert,
  214. resnet,
  215. roberta,
  216. roberta_prelayernorm,
  217. roc_bert,
  218. roformer,
  219. rt_detr,
  220. rwkv,
  221. sam,
  222. seamless_m4t,
  223. seamless_m4t_v2,
  224. segformer,
  225. seggpt,
  226. sew,
  227. sew_d,
  228. siglip,
  229. speech_encoder_decoder,
  230. speech_to_text,
  231. speecht5,
  232. splinter,
  233. squeezebert,
  234. stablelm,
  235. starcoder2,
  236. superpoint,
  237. swiftformer,
  238. swin,
  239. swin2sr,
  240. swinv2,
  241. switch_transformers,
  242. t5,
  243. table_transformer,
  244. tapas,
  245. time_series_transformer,
  246. timesformer,
  247. timm_backbone,
  248. trocr,
  249. tvp,
  250. udop,
  251. umt5,
  252. unispeech,
  253. unispeech_sat,
  254. univnet,
  255. upernet,
  256. video_llava,
  257. videomae,
  258. vilt,
  259. vipllava,
  260. vision_encoder_decoder,
  261. vision_text_dual_encoder,
  262. visual_bert,
  263. vit,
  264. vit_mae,
  265. vit_msn,
  266. vitdet,
  267. vitmatte,
  268. vits,
  269. vivit,
  270. wav2vec2,
  271. wav2vec2_bert,
  272. wav2vec2_conformer,
  273. wav2vec2_phoneme,
  274. wav2vec2_with_lm,
  275. wavlm,
  276. whisper,
  277. x_clip,
  278. xglm,
  279. xlm,
  280. xlm_roberta,
  281. xlm_roberta_xl,
  282. xlnet,
  283. xmod,
  284. yolos,
  285. yoso,
  286. zamba,
  287. zoedepth,
  288. )