__init__.py 13 KB

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