__init__.py 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396
  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 typing import TYPE_CHECKING
  15. from ..utils import _LazyModule
  16. from ..utils.import_utils import define_import_structure
  17. if TYPE_CHECKING:
  18. from .aimv2 import *
  19. from .albert import *
  20. from .align import *
  21. from .altclip import *
  22. from .arcee import *
  23. from .aria import *
  24. from .audio_spectrogram_transformer import *
  25. from .auto import *
  26. from .autoformer import *
  27. from .aya_vision import *
  28. from .bamba import *
  29. from .bark import *
  30. from .bart import *
  31. from .barthez import *
  32. from .bartpho import *
  33. from .beit import *
  34. from .bert import *
  35. from .bert_generation import *
  36. from .bert_japanese import *
  37. from .bertweet import *
  38. from .big_bird import *
  39. from .bigbird_pegasus import *
  40. from .biogpt import *
  41. from .bit import *
  42. from .bitnet import *
  43. from .blenderbot import *
  44. from .blenderbot_small import *
  45. from .blip import *
  46. from .blip_2 import *
  47. from .bloom import *
  48. from .blt import *
  49. from .bridgetower import *
  50. from .bros import *
  51. from .byt5 import *
  52. from .camembert import *
  53. from .canine import *
  54. from .chameleon import *
  55. from .chinese_clip import *
  56. from .clap import *
  57. from .clip import *
  58. from .clipseg import *
  59. from .clvp import *
  60. from .code_llama import *
  61. from .codegen import *
  62. from .cohere import *
  63. from .cohere2 import *
  64. from .cohere2_vision import *
  65. from .colpali import *
  66. from .colqwen2 import *
  67. from .conditional_detr import *
  68. from .convbert import *
  69. from .convnext import *
  70. from .convnextv2 import *
  71. from .cpm import *
  72. from .cpmant import *
  73. from .csm import *
  74. from .ctrl import *
  75. from .cvt import *
  76. from .d_fine import *
  77. from .dab_detr import *
  78. from .dac import *
  79. from .data2vec import *
  80. from .dbrx import *
  81. from .deberta import *
  82. from .deberta_v2 import *
  83. from .decision_transformer import *
  84. from .deepseek_v2 import *
  85. from .deepseek_v3 import *
  86. from .deepseek_vl import *
  87. from .deepseek_vl_hybrid import *
  88. from .deformable_detr import *
  89. from .deit import *
  90. from .deprecated import *
  91. from .depth_anything import *
  92. from .depth_pro import *
  93. from .detr import *
  94. from .dia import *
  95. from .dialogpt import *
  96. from .diffllama import *
  97. from .dinat import *
  98. from .dinov2 import *
  99. from .dinov2_with_registers import *
  100. from .dinov3_convnext import *
  101. from .dinov3_vit import *
  102. from .distilbert import *
  103. from .dit import *
  104. from .donut import *
  105. from .dots1 import *
  106. from .dpr import *
  107. from .dpt import *
  108. from .edgetam import *
  109. from .edgetam_video import *
  110. from .efficientloftr import *
  111. from .efficientnet import *
  112. from .electra import *
  113. from .emu3 import *
  114. from .encodec import *
  115. from .encoder_decoder import *
  116. from .ernie import *
  117. from .esm import *
  118. from .evolla import *
  119. from .exaone4 import *
  120. from .falcon import *
  121. from .falcon_h1 import *
  122. from .falcon_mamba import *
  123. from .fastspeech2_conformer import *
  124. from .flaubert import *
  125. from .flava import *
  126. from .flex_olmo import *
  127. from .florence2 import *
  128. from .fnet import *
  129. from .focalnet import *
  130. from .fsmt import *
  131. from .funnel import *
  132. from .fuyu import *
  133. from .gemma import *
  134. from .gemma2 import *
  135. from .gemma3 import *
  136. from .gemma3n import *
  137. from .git import *
  138. from .glm import *
  139. from .glm4 import *
  140. from .glpn import *
  141. from .got_ocr2 import *
  142. from .gpt2 import *
  143. from .gpt_bigcode import *
  144. from .gpt_neo import *
  145. from .gpt_neox import *
  146. from .gpt_neox_japanese import *
  147. from .gpt_oss import *
  148. from .gpt_sw3 import *
  149. from .gptj import *
  150. from .granite import *
  151. from .granite_speech import *
  152. from .granitemoe import *
  153. from .granitemoehybrid import *
  154. from .granitemoeshared import *
  155. from .grounding_dino import *
  156. from .groupvit import *
  157. from .helium import *
  158. from .herbert import *
  159. from .hgnet_v2 import *
  160. from .hiera import *
  161. from .hubert import *
  162. from .hunyuan_v1_dense import *
  163. from .hunyuan_v1_moe import *
  164. from .ibert import *
  165. from .idefics import *
  166. from .idefics2 import *
  167. from .idefics3 import *
  168. from .ijepa import *
  169. from .imagegpt import *
  170. from .informer import *
  171. from .instructblip import *
  172. from .instructblipvideo import *
  173. from .internvl import *
  174. from .jamba import *
  175. from .janus import *
  176. from .jetmoe import *
  177. from .kosmos2 import *
  178. from .kyutai_speech_to_text import *
  179. from .layoutlm import *
  180. from .layoutlmv2 import *
  181. from .layoutlmv3 import *
  182. from .layoutxlm import *
  183. from .led import *
  184. from .levit import *
  185. from .lfm2 import *
  186. from .lfm2_vl import *
  187. from .lightglue import *
  188. from .lilt import *
  189. from .llama import *
  190. from .llama4 import *
  191. from .llava import *
  192. from .llava_next import *
  193. from .llava_next_video import *
  194. from .llava_onevision import *
  195. from .longcat_flash import *
  196. from .longformer import *
  197. from .longt5 import *
  198. from .luke import *
  199. from .lxmert import *
  200. from .m2m_100 import *
  201. from .mamba import *
  202. from .mamba2 import *
  203. from .marian import *
  204. from .markuplm import *
  205. from .mask2former import *
  206. from .maskformer import *
  207. from .mbart import *
  208. from .mbart50 import *
  209. from .megatron_bert import *
  210. from .megatron_gpt2 import *
  211. from .mgp_str import *
  212. from .mimi import *
  213. from .minimax import *
  214. from .ministral import *
  215. from .mistral import *
  216. from .mistral3 import *
  217. from .mixtral import *
  218. from .mlcd import *
  219. from .mllama import *
  220. from .mluke import *
  221. from .mobilebert import *
  222. from .mobilenet_v1 import *
  223. from .mobilenet_v2 import *
  224. from .mobilevit import *
  225. from .mobilevitv2 import *
  226. from .modernbert import *
  227. from .modernbert_decoder import *
  228. from .moonshine import *
  229. from .moshi import *
  230. from .mpnet import *
  231. from .mpt import *
  232. from .mra import *
  233. from .mt5 import *
  234. from .musicgen import *
  235. from .musicgen_melody import *
  236. from .mvp import *
  237. from .myt5 import *
  238. from .nemotron import *
  239. from .nllb import *
  240. from .nllb_moe import *
  241. from .nougat import *
  242. from .nystromformer import *
  243. from .olmo import *
  244. from .olmo2 import *
  245. from .olmo3 import *
  246. from .olmoe import *
  247. from .omdet_turbo import *
  248. from .oneformer import *
  249. from .openai import *
  250. from .opt import *
  251. from .ovis2 import *
  252. from .owlv2 import *
  253. from .owlvit import *
  254. from .paligemma import *
  255. from .parakeet import *
  256. from .patchtsmixer import *
  257. from .patchtst import *
  258. from .pegasus import *
  259. from .pegasus_x import *
  260. from .perceiver import *
  261. from .perception_lm import *
  262. from .persimmon import *
  263. from .phi import *
  264. from .phi3 import *
  265. from .phi4_multimodal import *
  266. from .phimoe import *
  267. from .phobert import *
  268. from .pix2struct import *
  269. from .pixtral import *
  270. from .plbart import *
  271. from .poolformer import *
  272. from .pop2piano import *
  273. from .prompt_depth_anything import *
  274. from .prophetnet import *
  275. from .pvt import *
  276. from .pvt_v2 import *
  277. from .qwen2 import *
  278. from .qwen2_5_omni import *
  279. from .qwen2_5_vl import *
  280. from .qwen2_audio import *
  281. from .qwen2_moe import *
  282. from .qwen2_vl import *
  283. from .qwen3 import *
  284. from .qwen3_moe import *
  285. from .qwen3_next import *
  286. from .qwen3_omni_moe import *
  287. from .qwen3_vl import *
  288. from .qwen3_vl_moe import *
  289. from .rag import *
  290. from .recurrent_gemma import *
  291. from .reformer import *
  292. from .regnet import *
  293. from .rembert import *
  294. from .resnet import *
  295. from .roberta import *
  296. from .roberta_prelayernorm import *
  297. from .roc_bert import *
  298. from .roformer import *
  299. from .rt_detr import *
  300. from .rt_detr_v2 import *
  301. from .rwkv import *
  302. from .sam import *
  303. from .sam2 import *
  304. from .sam2_video import *
  305. from .sam_hq import *
  306. from .seamless_m4t import *
  307. from .seamless_m4t_v2 import *
  308. from .seed_oss import *
  309. from .segformer import *
  310. from .seggpt import *
  311. from .sew import *
  312. from .sew_d import *
  313. from .shieldgemma2 import *
  314. from .siglip import *
  315. from .siglip2 import *
  316. from .smolvlm import *
  317. from .speech_encoder_decoder import *
  318. from .speech_to_text import *
  319. from .speecht5 import *
  320. from .splinter import *
  321. from .squeezebert import *
  322. from .stablelm import *
  323. from .starcoder2 import *
  324. from .superglue import *
  325. from .superpoint import *
  326. from .swiftformer import *
  327. from .swin import *
  328. from .swin2sr import *
  329. from .swinv2 import *
  330. from .switch_transformers import *
  331. from .t5 import *
  332. from .t5gemma import *
  333. from .table_transformer import *
  334. from .tapas import *
  335. from .textnet import *
  336. from .time_series_transformer import *
  337. from .timesfm import *
  338. from .timesformer import *
  339. from .timm_backbone import *
  340. from .timm_wrapper import *
  341. from .trocr import *
  342. from .tvp import *
  343. from .udop import *
  344. from .umt5 import *
  345. from .unispeech import *
  346. from .unispeech_sat import *
  347. from .univnet import *
  348. from .upernet import *
  349. from .vaultgemma import *
  350. from .video_llava import *
  351. from .videomae import *
  352. from .vilt import *
  353. from .vipllava import *
  354. from .vision_encoder_decoder import *
  355. from .vision_text_dual_encoder import *
  356. from .visual_bert import *
  357. from .vit import *
  358. from .vit_mae import *
  359. from .vit_msn import *
  360. from .vitdet import *
  361. from .vitmatte import *
  362. from .vitpose import *
  363. from .vitpose_backbone import *
  364. from .vits import *
  365. from .vivit import *
  366. from .vjepa2 import *
  367. from .voxtral import *
  368. from .wav2vec2 import *
  369. from .wav2vec2_bert import *
  370. from .wav2vec2_conformer import *
  371. from .wav2vec2_phoneme import *
  372. from .wav2vec2_with_lm import *
  373. from .wavlm import *
  374. from .whisper import *
  375. from .x_clip import *
  376. from .xcodec import *
  377. from .xglm import *
  378. from .xlm import *
  379. from .xlm_roberta import *
  380. from .xlm_roberta_xl import *
  381. from .xlnet import *
  382. from .xlstm import *
  383. from .xmod import *
  384. from .yolos import *
  385. from .yoso import *
  386. from .zamba import *
  387. from .zamba2 import *
  388. from .zoedepth import *
  389. else:
  390. import sys
  391. _file = globals()["__file__"]
  392. sys.modules[__name__] = _LazyModule(__name__, _file, define_import_structure(_file), module_spec=__spec__)