
    UhkQ                    D   d Z ddlZddlmZ ddlmZ ddlmZmZm	Z	m
Z
 ddlmZ  ej                  e      Z eg d	      Z eg d
      Z eg d      Z eg d      Z eg d      Z eg d      Z edg      Z eg d      Z edg      Z eg d      Z edg      Z eg d      Z eg d      Z eg d      Z edg      Z eg d      Z eg d      Z  eg d      Z! eg d      Z" eg d      Z# eg d      Z$ eg d      Z% eg d      Z& eg d       Z' ed!g      Z( eg d"      Z) eg d#      Z* eg d$      Z+ eg d%      Z, eg d&      Z- eg d'      Z. eg d(      Z/ eg d)      Z0 eg d*      Z1 ed+d,g      Z2 eg d-      Z3 eg d.      Z4 eg d/      Z5 ed0g      Z6 ed1g      Z6 ed2g      Z7 eg d3      Z8 ed4d5g      Z9 ed6d7g      Z: ed8g      Z; ed9g      Z< e	ee      Z= e	ee      Z> e	ee      Z? e	ee      Z@ e	ee      ZA e	ee      ZB e	ee4      ZC e	ee      ZD e	ee      ZE e	ee      ZF e	ee      ZG e	ee      ZH e	ee      ZI e	ee      ZJ e	ee      ZK e	ee)      ZL e	ee*      ZM e	ee       ZN e	ee      ZO e	ee      ZP e	ee!      ZQ e	ee"      ZR e	ee#      ZS e	ee$      ZT e	ee&      ZU e	ee'      ZV e	ee(      ZW e	ee+      ZX e	ee,      ZY e	ee-      ZZ e	ee.      Z[ e	ee/      Z\ e	ee%      Z] e	ee0      Z^ e	ee1      Z_ e	ee2      Z` e	ee3      Za e	ee5      Zb e	ee6      Zc e	ee7      Zd e	ee8      Ze e	ee9      Zf e	ee:      Zg e	ee;      Zh e	ee<      Zi G d: d;e      Zj G d< d=e      Zk G d> d?e      Zl G d@ dAe      Zm G dB dCe      Zn e
en      Zn G dD dEe      Zo e
eodFG      Zo G dH dIe      Zp e
epdJG      Zp G dK dLe      Zq e
eqdMG      Zq G dN dOe      Zr e
erdPG      Zr G dQ dRe      Zs e
esdSdTU      Zs G dV dWe      Zt e
etdXG      Zt G dY dZe      Zu e
eud[G      Zu G d\ d]e      Zv e
evd^d_U      Zv G d` dae      Zw e
ewdbdcU      Zw G dd dee      Zx e
exdfdgU      Zx G dh die      Zy e
eydjG      Zy G dk dle      Zz e
ezdmG      Zz G dn doe      Z{ e
e{dpG      Z{ G dq dre      Z| e
e|dsG      Z| G dt due      Z} e
e}dvG      Z} G dw dxe      Z~ e
e~dyG      Z~ G dz d{e      Z e
ed|G      Z G d} d~e      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d de      Z G d de      Z G d de      Z G d de      Z e
edG      Z G d de      Z e
edG      Z G d dep      Zg dZy)zAuto Model class.    N)OrderedDict   )logging   )_BaseAutoBackboneClass_BaseAutoModelClass_LazyAutoMappingauto_class_update)CONFIG_MAPPING_NAMES(?  albertAlbertModelalign
AlignModelaltclipAltCLIPModel)aria	AriaModel)	aria_textAriaTextModel)audio-spectrogram-transformerASTModel)
autoformerAutoformerModel)
aya_visionAyaVisionModel)bamba
BambaModelbark	BarkModel)bart	BartModelbeit	BeitModelbert	BertModel)bert-generationBertGenerationEncoderbig_birdBigBirdModel)bigbird_pegasusBigBirdPegasusModel)biogptBioGptModelbitBitModel)bitnetBitNetModel)
blenderbotBlenderbotModel)blenderbot-smallBlenderbotSmallModelblip	BlipModel)blip-2
Blip2Model)blip_2_qformerBlip2QFormerModel)bloom
BloomModel)bridgetowerBridgeTowerModel)bros	BrosModel)	camembertCamembertModel)canineCanineModel)	chameleonChameleonModelchinese_clipChineseCLIPModel)chinese_clip_vision_modelChineseCLIPVisionModel)clap	ClapModelclip	CLIPModelclip_text_modelCLIPTextModel)clip_vision_modelCLIPVisionModelclipsegCLIPSegModel)clvp!ClvpModelForConditionalGeneration)
code_llama
LlamaModel)codegenCodeGenModel)cohereCohereModel)cohere2Cohere2Modelconditional_detrConditionalDetrModel)convbertConvBertModelconvnextConvNextModel
convnextv2ConvNextV2Model)cpmantCpmAntModelcsmCsmForConditionalGeneration)ctrl	CTRLModel)cvtCvtModel)d_fine
DFineModeldab-detrDabDetrModel)dacDacModel)data2vec-audioData2VecAudioModeldata2vec-textData2VecTextModeldata2vec-visionData2VecVisionModel)dbrx	DbrxModeldebertaDebertaModel
deberta-v2DebertaV2Model)decision_transformerDecisionTransformerModel)deepseek_v3DeepseekV3Modeldeformable_detrDeformableDetrModeldeit	DeiTModel	depth_proDepthProModeldeta	DetaModeldetr	DetrModel)	diffllamaDiffLlamaModeldinat
DinatModeldinov2Dinov2Modeldinov2_with_registersDinov2WithRegistersModel
distilbertDistilBertModel)
donut-swinDonutSwinModel)dprDPRQuestionEncoderdptDPTModelefficientformerEfficientFormerModelefficientnetEfficientNetModelelectraElectraModel)emu3	Emu3Model)encodecEncodecModel)ernie
ErnieModel)ernie_mErnieMModel)esmEsmModel)falconFalconModel)falcon_mambaFalconMambaModelfastspeech2_conformerFastSpeech2ConformerModelflaubertFlaubertModel)flava
FlavaModel)fnet	FNetModelfocalnetFocalNetModel)fsmt	FSMTModel)funnel)FunnelModelFunnelBaseModel)fuyu	FuyuModel)gemma
GemmaModel)gemma2Gemma2Model)gemma3Gemma3Model)gemma3_textGemma3TextModel)gitGitModel)glmGlmModel)glm4	Glm4Modelglpn	GLPNModel)got_ocr2GotOcr2Model)gpt-sw3	GPT2Model)gpt2r   )gpt_bigcodeGPTBigCodeModel)gpt_neoGPTNeoModel)gpt_neoxGPTNeoXModel)gpt_neox_japaneseGPTNeoXJapaneseModel)gptj	GPTJModelzgptsan-japanese&GPTSanJapaneseForConditionalGeneration)graniteGraniteModel)
granitemoeGraniteMoeModel)granitemoehybridGraniteMoeHybridModel)granitemoesharedGraniteMoeSharedModel)
graphormerGraphormerModel)grounding-dinoGroundingDinoModel)groupvitGroupViTModel)heliumHeliumModelhgnet_v2HGNetV2Backbonehiera
HieraModel)hubertHubertModelibert
IBertModel)ideficsIdeficsModel)idefics2Idefics2Model)idefics3Idefics3Model)idefics3_visionIdefics3VisionTransformerijepa
IJepaModelimagegptImageGPTModel)informerInformerModel)instructblipInstructBlipModel)instructblipvideoInstructBlipVideoModel)internvlInternVLModel)internvl_visionInternVLVisionModel)jamba
JambaModel)janus
JanusModel)jetmoeJetMoeModel)jukeboxJukeboxModel)kosmos-2Kosmos2Model)layoutlmLayoutLMModel)
layoutlmv2LayoutLMv2Model)
layoutlmv3LayoutLMv3Model)ledLEDModellevit
LevitModel)lilt	LiltModel)llamarf   llama4Llama4ForConditionalGeneration)llama4_textLlama4TextModel)llava
LlavaModel)
llava_nextLlavaNextModel)llava_next_videoLlavaNextVideoModel)llava_onevisionLlavaOnevisionModel
longformerLongformerModel)longt5LongT5Model)luke	LukeModel)lxmertLxmertModel)m2m_100M2M100Model)mamba
MambaModel)mamba2Mamba2Model)marianMarianModel)markuplmMarkupLMModel)mask2formerMask2FormerModel)
maskformerMaskFormerModel)maskformer-swinMaskFormerSwinModel)mbart
MBartModel)mctct
MCTCTModel)mega	MegaModel)megatron-bertMegatronBertModel)zmgp-strMgpstrForSceneTextRecognition)mimi	MimiModel)mistralMistralModel)mistral3Mistral3Model)mixtralMixtralModelmlcdMLCDVisionModel)mllamaMllamaModel
mobilebertMobileBertModelmobilenet_v1MobileNetV1Modelmobilenet_v2MobileNetV2Model	mobilevitMobileViTModelmobilevitv2MobileViTV2Model)
modernbertModernBertModel)	moonshineMoonshineModel)moshi
MoshiModel)mpnet
MPNetModel)mptMptModel)mraMraModel)mt5MT5Model)musicgenMusicgenModel)musicgen_melodyMusicgenMelodyModel)mvpMvpModelnatNatModel)nemotronNemotronModel)nezha
NezhaModel)nllb-moeNllbMoeModelnystromformerNystromformerModel)olmo	OlmoModel)olmo2
Olmo2Model)olmoe
OlmoeModelzomdet-turboOmDetTurboForObjectDetection)	oneformerOneFormerModel)
open-llamaOpenLlamaModel)
openai-gptOpenAIGPTModel)optOPTModel)owlv2
Owlv2Model)owlvitOwlViTModel)	paligemmaPaliGemmaModel)patchtsmixerPatchTSMixerModel)patchtstPatchTSTModel)pegasusPegasusModel)	pegasus_xPegasusXModel)	perceiverPerceiverModel)	persimmonPersimmonModel)phiPhiModel)phi3	Phi3Model)phi4_multimodalPhi4MultimodalModel)phimoePhimoeModel)pixtralPixtralVisionModel)plbartPLBartModel
poolformerPoolFormerModel)
prophetnetProphetNetModelpvtPvtModel)pvt_v2
PvtV2Model)qdqbertQDQBertModel)qwen2
Qwen2Model)
qwen2_5_vlQwen2_5_VLModel)qwen2_5_vl_textQwen2_5_VLTextModel)qwen2_audio_encoderQwen2AudioEncoder)	qwen2_moeQwen2MoeModel)qwen2_vlQwen2VLModel)qwen2_vl_textQwen2VLTextModel)qwen3
Qwen3Model)	qwen3_moeQwen3MoeModel)recurrent_gemmaRecurrentGemmaModelreformerReformerModelregnetRegNetModelrembertRemBertModelresnetResNetModel	retribertRetriBertModelrobertaRobertaModelroberta-prelayernormRobertaPreLayerNormModelroc_bertRoCBertModelroformerRoFormerModel)rt_detrRTDetrModel)
rt_detr_v2RTDetrV2Model)rwkv	RwkvModelsamSamModelsam_hq
SamHQModel)sam_hq_vision_modelSamHQVisionModel)sam_vision_modelSamVisionModel)seamless_m4tSeamlessM4TModel)seamless_m4t_v2SeamlessM4Tv2Model	segformerSegformerModel)seggptSegGptModel)sewSEWModel)sew-d	SEWDModelsiglipSiglipModelsiglip2Siglip2Modelsiglip_vision_modelSiglipVisionModel)smolvlmSmolVLMModel)smolvlm_visionSmolVLMVisionTransformer)speech_to_textSpeech2TextModel)speecht5SpeechT5Model)splinterSplinterModelsqueezebertSqueezeBertModel)stablelmStableLmModel)
starcoder2Starcoder2Model)	superglueSuperGlueForKeypointMatchingswiftformerSwiftFormerModelswin	SwinModelswin2srSwin2SRModelswinv2Swinv2Model)switch_transformersSwitchTransformersModel)t5T5Modeltable-transformerTableTransformerModel)tapas
TapasModel)textnetTextNetModel)time_series_transformerTimeSeriesTransformerModel)timesfmTimesFmModeltimesformerTimesformerModeltimm_backboneTimmBackbonetimm_wrapperTimmWrapperModel)trajectory_transformerTrajectoryTransformerModel)
transfo-xlTransfoXLModel)tvlt	TvltModel)tvpTvpModel)udop	UdopModel)umt5	UMT5Model)	unispeechUniSpeechModel)unispeech-satUniSpeechSatModel)univnetUnivNetModelvanVanModel)video_llavaVideoLlavaModelvideomaeVideoMAEModel)vilt	ViltModel)vipllavaVipLlavaModel)zvision-text-dual-encoderVisionTextDualEncoderModel)visual_bertVisualBertModelvitViTModel
vit_hybridViTHybridModelvit_maeViTMAEModelvit_msnViTMSNModelvitdetVitDetModelvits	VitsModelvivit
VivitModel)wav2vec2Wav2Vec2Model)wav2vec2-bertWav2Vec2BertModel)wav2vec2-conformerWav2Vec2ConformerModel)wavlm
WavLMModel)whisperWhisperModel)xclip
XCLIPModel)xglm	XGLMModelxlmXLMModel)xlm-prophetnetXLMProphetNetModelxlm-robertaXLMRobertaModelxlm-roberta-xlXLMRobertaXLModel)xlnet
XLNetModel)xmod	XmodModelyolos
YolosModel)yoso	YosoModel)zamba
ZambaModel)zamba2Zamba2Model)Q)r   AlbertForPreTrainingr$   BartForConditionalGeneration)r*   BertForPreTraining)r/   BigBirdForPreTrainingrE   BloomForCausalLMrK   CamembertForMaskedLMcolpaliColPaliForRetrievalr}   CTRLLMHeadModelr   Data2VecTextForMaskedLMr   DebertaForMaskedLMr   DebertaV2ForMaskedLMr   DistilBertForMaskedLM)r   ElectraForPreTraining)r   ErnieForPreTrainingr   FalconMambaForCausalLMr   FlaubertWithLMHeadModel)r   FlavaForPreTraining)r   FNetForPreTrainingr   FSMTForConditionalGeneration)r   FunnelForPreTrainingr   Gemma3ForConditionalGenerationr   GPT2LMHeadModelr   r	  r  GPTBigCodeForCausalLMr  )r!  HieraForPreTrainingr&  IBertForMaskedLMr(  IdeficsForVisionText2Textr*   Idefics2ForConditionalGenerationr,   Idefics3ForConditionalGenerationrB  JanusForConditionalGenerationrJ  LayoutLMForMaskedLMr]  LlavaForConditionalGenerationr_  !LlavaNextForConditionalGenerationra  &LlavaNextVideoForConditionalGenerationrc  &LlavaOnevisionForConditionalGenerationrf  LongformerForMaskedLMrj  LukeForMaskedLM)rl  LxmertForPreTrainingrp  MambaForCausalLMrr  Mamba2ForCausalLMr  MegaForMaskedLM)r  MegatronBertForPreTrainingr   Mistral3ForConditionalGenerationr  MllamaForConditionalGeneration)r  MobileBertForPreTrainingr  MPNetForMaskedLMr  MptForCausalLMr  MraForMaskedLMr  MvpForConditionalGeneration)r  NezhaForPreTrainingr  NllbMoeForConditionalGenerationr  OpenAIGPTLMHeadModelr  !PaliGemmaForConditionalGenerationqwen2_audio"Qwen2AudioForConditionalGenerationr  r!  RobertaForMaskedLMr$  RobertaPreLayerNormForMaskedLM)r'  RoCBertForPreTrainingr0  RwkvForCausalLM)rZ  SplinterForPreTrainingr]  SqueezeBertForMaskedLMrq  *SwitchTransformersForConditionalGenerationrs  T5ForConditionalGenerationrx  TapasForMaskedLMr  TransfoXLLMHeadModel)r  TvltForPreTraining)r  UniSpeechForPreTraining)r  UniSpeechSatForPreTrainingr  "VideoLlavaForConditionalGeneration)r  VideoMAEForPreTrainingr   VipLlavaForConditionalGeneration)r  VisualBertForPreTraining)r  ViTMAEForPreTraining)r  Wav2Vec2ForPreTraining)r  Wav2Vec2ConformerForPreTrainingr  XLMWithLMHeadModelr  XLMRobertaForMaskedLMr  XLMRobertaXLForMaskedLMr  XLNetLMHeadModelr  XmodForMaskedLM)Qr   AlbertForMaskedLMr  r*   BertForMaskedLMr/   BigBirdForMaskedLMr1   &BigBirdPegasusForConditionalGenerationr<   'BlenderbotSmallForConditionalGenerationr  r  rg   CodeGenForCausalLMrp   ConvBertForMaskedLMrx   CpmAntForCausalLMr  r  r  r  r  r   ElectraForMaskedLMzencoder-decoderEncoderDecoderModelr   ErnieForMaskedLMr   EsmForMaskedLMr  r  r   FNetForMaskedLMr  r   FunnelForMaskedLMr   GitForCausalLMr  r
  r  r  GPTNeoForCausalLMr  GPTNeoXForCausalLMr  GPTNeoXJapaneseForCausalLMr	  GPTJForCausalLMr  r  r  rP  LEDForConditionalGenerationr"  rh  LongT5ForConditionalGenerationr$  rn  M2M100ForConditionalGenerationr'  r)  rt  MarianMTModelr+  r  MegatronBertForCausalLMr  MobileBertForMaskedLMr  !MoonshineForConditionalGenerationr3  r5  r7  r9  r  NezhaForMaskedLMr<  r  NystromformerForMaskedLMr>  r   PegasusXForConditionalGenerationr  PLBartForConditionalGeneration	pop2piano!Pop2PianoForConditionalGenerationr  QDQBertForMaskedLMr  ReformerModelWithLMHeadr  RemBertForMaskedLMrE  rG  r'  RoCBertForMaskedLMr*  RoFormerForMaskedLMrJ  rV  #Speech2TextForConditionalGenerationrM  rO  rQ  rS  rU  r  Wav2Vec2ForMaskedLMr  WhisperForConditionalGenerationrc  re  rg  ri  rk  r  YosoForMaskedLM)o)r   AriaTextForCausalLM)r   BambaForCausalLM)r$   BartForCausalLM)r*   BertLMHeadModel)r,   BertGenerationDecoder)r/   BigBirdForCausalLM)r1   BigBirdPegasusForCausalLM)r3   BioGptForCausalLM)r8   BitNetForCausalLM)r:   BlenderbotForCausalLM)r<   BlenderbotSmallForCausalLMr  )rK   CamembertForCausalLM)re   LlamaForCausalLMrw  )ri   CohereForCausalLM)rk   Cohere2ForCausalLMr{  r  )r   Data2VecTextForCausalLM)r   DbrxForCausalLM)r   DeepseekV3ForCausalLM)r   DiffLlamaForCausalLM)r   ElectraForCausalLM)r   Emu3ForCausalLM)r   ErnieForCausalLM)r   FalconForCausalLMr  r   FuyuForCausalLM)r   GemmaForCausalLM)r   Gemma2ForCausalLMr  )r   Gemma3ForCausalLMr  )r   GlmForCausalLM)r   Glm4ForCausalLMr   GotOcr2ForConditionalGenerationr  r
  r  r  r  r  r  )r  GraniteForCausalLM)r  GraniteMoeForCausalLM)r  GraniteMoeHybridForCausalLM)r  GraniteMoeSharedForCausalLM)r  HeliumForCausalLM)r@  JambaForCausalLM)rD  JetMoeForCausalLM)rW  r  )rY  Llama4ForCausalLM)r[  r  r'  r)  )rt  MarianForCausalLM)r~  MBartForCausalLM)r  MegaForCausalLMr  )r  MistralForCausalLM)r  MixtralForCausalLM)r  MllamaForCausalLM)r  MoshiForCausalLMr5  )r  MusicgenForCausalLM)r  MusicgenMelodyForCausalLM)r  MvpForCausalLM)r  NemotronForCausalLM)r  OlmoForCausalLM)r  Olmo2ForCausalLM)r  OlmoeForCausalLM)r  OpenLlamaForCausalLMr>  )r  OPTForCausalLM)r  PegasusForCausalLM)r  PersimmonForCausalLM)r  PhiForCausalLM)r  Phi3ForCausalLM)r  Phi4MultimodalForCausalLM)r  PhimoeForCausalLM)r  PLBartForCausalLM)r  ProphetNetForCausalLM)r  QDQBertLMHeadModel)r  Qwen2ForCausalLM)r  Qwen2MoeForCausalLM)r  Qwen3ForCausalLM)r  Qwen3MoeForCausalLM)r  RecurrentGemmaForCausalLMr  )r  RemBertForCausalLM)r!  RobertaForCausalLM)r$  RobertaPreLayerNormForCausalLM)r'  RoCBertForCausalLM)r*  RoFormerForCausalLMrJ  )speech_to_text_2Speech2Text2ForCausalLM)r_  StableLmForCausalLM)ra  Starcoder2ForCausalLMrU  )trocrTrOCRForCausalLM)r  WhisperForCausalLM)r  XGLMForCausalLMrc  )r  XLMProphetNetForCausalLM)r  XLMRobertaForCausalLM)r  XLMRobertaXLForCausalLMri  )r  XmodForCausalLM)r  ZambaForCausalLM)r  Zamba2ForCausalLM)7r&   r5   rm   rr   ru   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r0  r3  rR  )rY  Llama4VisionModelr  )r  MllamaVisionModelr  r  r  r  r  r  r  r  r  r@  rO  re  rh  rk  rn  ru  r  r  r  r  r  r  r  r  r  r  r  r  ))r   DeiTForMaskedImageModeling)r   FocalNetForMaskedImageModeling)ri  SwinForMaskedImageModeling)ro  Swinv2ForMaskedImageModeling)r  ViTForMaskedImageModeling)r4  ImageGPTForCausalImageModeling),)r'   BeitForImageClassification)r6   BitForImageClassification)rY   CLIPForImageClassification)rs   ConvNextForImageClassification)rv    ConvNextV2ForImageClassification)r   CvtForImageClassification)r   $Data2VecVisionForImageClassification)r   )DeiTForImageClassification%DeiTForImageClassificationWithTeacher)r   DinatForImageClassification)r   Dinov2ForImageClassification)r   )Dinov2WithRegistersForImageClassification)r   DonutSwinForImageClassification)r   )%EfficientFormerForImageClassification0EfficientFormerForImageClassificationWithTeacher)r   "EfficientNetForImageClassification)r   FocalNetForImageClassification)r  HGNetV2ForImageClassification)r!  HieraForImageClassification)r1  IJepaForImageClassification)r4  ImageGPTForImageClassification)rS  )LevitForImageClassification&LevitForImageClassificationWithTeacher)r  !MobileNetV1ForImageClassification)r  !MobileNetV2ForImageClassification)r  MobileViTForImageClassification)r  !MobileViTV2ForImageClassification)r  NatForImageClassification)r  )&PerceiverForImageClassificationLearned&PerceiverForImageClassificationFourier-PerceiverForImageClassificationConvProcessing)r   PoolFormerForImageClassification)r  PvtForImageClassification)r  PvtV2ForImageClassification)r  RegNetForImageClassification)r  ResNetForImageClassification)rA  SegformerForImageClassification)shieldgemma2"ShieldGemma2ForImageClassification)rJ  SiglipForImageClassification)rM  Siglip2ForImageClassification)rf  !SwiftFormerForImageClassification)ri  SwinForImageClassification)ro  Swinv2ForImageClassification)rz  TextNetForImageClassification)r  !TimmWrapperForImageClassification)r  VanForImageClassification)r  ViTForImageClassification)r  ViTHybridForImageClassification)r  ViTMSNForImageClassificationr   DetrForSegmentation))r'   BeitForSemanticSegmentation)r   %Data2VecVisionForSemanticSegmentation)r   DPTForSemanticSegmentation)r  "MobileNetV2ForSemanticSegmentation)r   MobileViTForSemanticSegmentation)r  "MobileViTV2ForSemanticSegmentation)rA   SegformerForSemanticSegmentation)upernetUperNetForSemanticSegmentationrz  !MaskFormerForInstanceSegmentation)rQ  )rx  #Mask2FormerForUniversalSegmentationr\  )r  !OneFormerForUniversalSegmentation))r  !TimesformerForVideoClassification)r  VideoMAEForVideoClassification)r  VivitForVideoClassification)r?   BlipForConditionalGenerationrA   Blip2ForConditionalGenerationrO   !ChameleonForConditionalGenerationr  r  r  r8  $InstructBlipForConditionalGeneration)r:  )InstructBlipVideoForConditionalGenerationrH  Kosmos2ForConditionalGenerationr  r  r  r   r.  r0  r@  
pix2struct"Pix2StructForConditionalGenerationr  "Qwen2_5_VLForConditionalGenerationr  Qwen2VLForConditionalGenerationrZ  r]  zvision-encoder-decoderVisionEncoderDecoderModelr  )")r   AriaForConditionalGeneration)r   !AyaVisionForConditionalGenerationrc  re  rg  )r   Emu3ForConditionalGenerationr  r  r  r  r  r  r  ri  )r<   InternVLForConditionalGenerationr  rl  rX  r  r  r  r   r.  r0  r@  rn  )r  r  rq  rs  )rD  r  )rR  SmolVLMForConditionalGeneration)r  UdopForConditionalGenerationr]  ru  ).rm  r  ro  rq  r  ry  r  r  r  r  r}  r  r  r  r  r  r  r  r"  r$  r~  MBartForConditionalGenerationr+  )r  MegatronBertForMaskedLMr  )r  ModernBertForMaskedLMr3  r7  r9  r  r  )r  PerceiverForMaskedLMr  )r  ReformerForMaskedLMr  rE  rG  r  r  rM  rS  r  rc  re  rg  rk  r  )
)rn   !ConditionalDetrForObjectDetection)r   DFineForObjectDetection)r   DabDetrForObjectDetection)r    DeformableDetrForObjectDetection)r   DetaForObjectDetection)r   DetrForObjectDetection)r,  RTDetrForObjectDetection)r.  RTDetrV2ForObjectDetection)rv  "TableTransformerForObjectDetection)r  YolosForObjectDetection))r  GroundingDinoForObjectDetectionr  )r  Owlv2ForObjectDetection)r  OwlViTForObjectDetection))depth_anythingDepthAnythingForDepthEstimation)r   DepthProForDepthEstimation)r   DPTForDepthEstimation)r   GLPNForDepthEstimation)prompt_depth_anything%PromptDepthAnythingForDepthEstimation)zoedepthZoeDepthForDepthEstimation)r  rs  )r:   "BlenderbotForConditionalGenerationru  r  r  r  granite_speech%GraniteSpeechForConditionalGenerationr  r  r  r  r}  )r  MT5ForConditionalGenerationr9  r<  )r  PegasusForConditionalGenerationr  r  )r  "ProphetNetForConditionalGenerationrB  )r<  SeamlessM4TForTextToText)r>  SeamlessM4Tv2ForTextToTextrO  rQ  )r  UMT5ForConditionalGeneration)r  %XLMProphetNetForConditionalGeneration)	r  r  r  )r<  SeamlessM4TForSpeechToText)r>  SeamlessM4Tv2ForSpeechToText)zspeech-encoder-decoderSpeechEncoderDecoderModelr  )rX  SpeechT5ForSpeechToTextr  )b)r   AlbertForSequenceClassification)r$   BartForSequenceClassification)r*   BertForSequenceClassification)r/    BigBirdForSequenceClassification)r1   'BigBirdPegasusForSequenceClassification)r3   BioGptForSequenceClassification)rE   BloomForSequenceClassification)rK   "CamembertForSequenceClassification)rM   CanineForSequenceClassification)re   LlamaForSequenceClassification)rp   !ConvBertForSequenceClassification)r}   CTRLForSequenceClassification)r   %Data2VecTextForSequenceClassification)r    DebertaForSequenceClassification)r   "DebertaV2ForSequenceClassification)r   "DiffLlamaForSequenceClassification)r   #DistilBertForSequenceClassification)r    ElectraForSequenceClassification)r   ErnieForSequenceClassification)r   ErnieMForSequenceClassification)r   EsmForSequenceClassification)r   FalconForSequenceClassification)r   !FlaubertForSequenceClassification)r   FNetForSequenceClassification)r   FunnelForSequenceClassification)r   GemmaForSequenceClassification)r   Gemma2ForSequenceClassification)r   GlmForSequenceClassification)r   Glm4ForSequenceClassification)r   GPT2ForSequenceClassification)r   r  )r  #GPTBigCodeForSequenceClassification)r  GPTNeoForSequenceClassification)r   GPTNeoXForSequenceClassification)r	  GPTJForSequenceClassification)r  HeliumForSequenceClassification)r&  IBertForSequenceClassification)r@  JambaForSequenceClassification)rD  JetMoeForSequenceClassification)rJ  !LayoutLMForSequenceClassification)rL  #LayoutLMv2ForSequenceClassification)rN  #LayoutLMv3ForSequenceClassification)rP  LEDForSequenceClassification)rU  LiltForSequenceClassification)rW  r  )rf  #LongformerForSequenceClassification)rj  LukeForSequenceClassification)rv  !MarkupLMForSequenceClassification)r~  MBartForSequenceClassification)r  MegaForSequenceClassification)r  %MegatronBertForSequenceClassification)r   MistralForSequenceClassification)r   MixtralForSequenceClassification)r  #MobileBertForSequenceClassification)r  #ModernBertForSequenceClassification)r  MPNetForSequenceClassification)r  MptForSequenceClassification)r  MraForSequenceClassification)r  MT5ForSequenceClassification)r  MvpForSequenceClassification)r  !NemotronForSequenceClassification)r  NezhaForSequenceClassification)r  &NystromformerForSequenceClassification)r  "OpenLlamaForSequenceClassification)r  "OpenAIGPTForSequenceClassification)r  OPTForSequenceClassification)r  "PerceiverForSequenceClassification)r  "PersimmonForSequenceClassification)r  PhiForSequenceClassification)r  Phi3ForSequenceClassification)r  PhimoeForSequenceClassification)r  PLBartForSequenceClassification)r   QDQBertForSequenceClassification)r  Qwen2ForSequenceClassification)r  !Qwen2MoeForSequenceClassification)r  Qwen3ForSequenceClassification)r  !Qwen3MoeForSequenceClassification)r  !ReformerForSequenceClassification)r   RemBertForSequenceClassification)r!   RobertaForSequenceClassification)r$  ,RobertaPreLayerNormForSequenceClassification)r'   RoCBertForSequenceClassification)r*  !RoFormerForSequenceClassification)r]  $SqueezeBertForSequenceClassification)r_  !StableLmForSequenceClassification)ra  #Starcoder2ForSequenceClassification)rs  T5ForSequenceClassification)rx  TapasForSequenceClassification)r  "TransfoXLForSequenceClassification)r  UMT5ForSequenceClassification)r  XLMForSequenceClassification)r  #XLMRobertaForSequenceClassification)r  %XLMRobertaXLForSequenceClassification)r  XLNetForSequenceClassification)r  XmodForSequenceClassification)r  YosoForSequenceClassification)r  ZambaForSequenceClassification)r  Zamba2ForSequenceClassification)H)r   AlbertForQuestionAnswering)r$   BartForQuestionAnswering)r*   BertForQuestionAnswering)r/   BigBirdForQuestionAnswering)r1   "BigBirdPegasusForQuestionAnswering)rE   BloomForQuestionAnswering)rK   CamembertForQuestionAnswering)rM   CanineForQuestionAnswering)rp   ConvBertForQuestionAnswering)r    Data2VecTextForQuestionAnswering)r   DebertaForQuestionAnswering)r   DebertaV2ForQuestionAnswering)r   DiffLlamaForQuestionAnswering)r   DistilBertForQuestionAnswering)r   ElectraForQuestionAnswering)r   ErnieForQuestionAnswering)r   ErnieMForQuestionAnswering)r   FalconForQuestionAnswering)r   "FlaubertForQuestionAnsweringSimple)r   FNetForQuestionAnswering)r   FunnelForQuestionAnswering)r   GPT2ForQuestionAnswering)r  GPTNeoForQuestionAnswering)r  GPTNeoXForQuestionAnswering)r	  GPTJForQuestionAnswering)r&  IBertForQuestionAnsweringrL  LayoutLMv2ForQuestionAnsweringrN  LayoutLMv3ForQuestionAnswering)rP  LEDForQuestionAnswering)rU  LiltForQuestionAnswering)rW  LlamaForQuestionAnswering)rf  LongformerForQuestionAnswering)rj  LukeForQuestionAnswering)rl  LxmertForQuestionAnswering)rv  MarkupLMForQuestionAnswering)r~  MBartForQuestionAnswering)r  MegaForQuestionAnswering)r   MegatronBertForQuestionAnswering)r  MistralForQuestionAnswering)r  MixtralForQuestionAnswering)r  MobileBertForQuestionAnswering)r  ModernBertForQuestionAnswering)r  MPNetForQuestionAnswering)r  MptForQuestionAnswering)r  MraForQuestionAnswering)r  MT5ForQuestionAnswering)r  MvpForQuestionAnswering)r  NemotronForQuestionAnswering)r  NezhaForQuestionAnswering)r  !NystromformerForQuestionAnswering)r  OPTForQuestionAnswering)r  QDQBertForQuestionAnswering)r  Qwen2ForQuestionAnswering)r  Qwen2MoeForQuestionAnswering)r  Qwen3ForQuestionAnswering)r  Qwen3MoeForQuestionAnswering)r  ReformerForQuestionAnswering)r  RemBertForQuestionAnswering)r!  RobertaForQuestionAnswering)r$  'RobertaPreLayerNormForQuestionAnswering)r'  RoCBertForQuestionAnswering)r*  RoFormerForQuestionAnswering)rZ  SplinterForQuestionAnswering)r]  SqueezeBertForQuestionAnswering)rs  T5ForQuestionAnswering)r  UMT5ForQuestionAnswering)r  XLMForQuestionAnsweringSimple)r  XLMRobertaForQuestionAnswering)r   XLMRobertaXLForQuestionAnswering)r  XLNetForQuestionAnsweringSimple)r  XmodForQuestionAnswering)r  YosoForQuestionAnswering)rx  TapasForQuestionAnswering))r?   BlipForQuestionAnsweringre  )r  ViltForQuestionAnswering))rJ  LayoutLMForQuestionAnsweringr"  r$  )N)r   AlbertForTokenClassification)r*   BertForTokenClassification)r/   BigBirdForTokenClassification)r3   BioGptForTokenClassification)rE   BloomForTokenClassification)rI   BrosForTokenClassification)rK   CamembertForTokenClassification)rM   CanineForTokenClassification)rp   ConvBertForTokenClassification)r   "Data2VecTextForTokenClassification)r   DebertaForTokenClassification)r   DebertaV2ForTokenClassification)r   DiffLlamaForTokenClassification)r    DistilBertForTokenClassification)r   ElectraForTokenClassification)r   ErnieForTokenClassification)r   ErnieMForTokenClassification)r   EsmForTokenClassification)r   FalconForTokenClassification)r   FlaubertForTokenClassification)r   FNetForTokenClassification)r   FunnelForTokenClassification)r   GemmaForTokenClassification)r   Gemma2ForTokenClassification)r   GlmForTokenClassification)r   Glm4ForTokenClassification)r   GPT2ForTokenClassification)r   rp  )r   GPTBigCodeForTokenClassification)r  GPTNeoForTokenClassification)r  GPTNeoXForTokenClassification)r  HeliumForTokenClassification)r&  IBertForTokenClassification)rJ  LayoutLMForTokenClassification)rL   LayoutLMv2ForTokenClassification)rN   LayoutLMv3ForTokenClassification)rU  LiltForTokenClassification)rW  LlamaForTokenClassification)rf   LongformerForTokenClassification)rj  LukeForTokenClassification)rv  MarkupLMForTokenClassification)r  MegaForTokenClassification)r  "MegatronBertForTokenClassification)r  MistralForTokenClassification)r  MixtralForTokenClassification)r   MobileBertForTokenClassification)r   ModernBertForTokenClassification)r  MPNetForTokenClassification)r  MptForTokenClassification)r  MraForTokenClassification)r  MT5ForTokenClassification)r  NemotronForTokenClassification)r  NezhaForTokenClassification)r  #NystromformerForTokenClassification)r  PersimmonForTokenClassification)r  PhiForTokenClassification)r  Phi3ForTokenClassification)r  QDQBertForTokenClassification)r  Qwen2ForTokenClassification)r  Qwen2MoeForTokenClassification)r  Qwen3ForTokenClassification)r  Qwen3MoeForTokenClassification)r  RemBertForTokenClassification)r!  RobertaForTokenClassification)r$  )RobertaPreLayerNormForTokenClassification)r'  RoCBertForTokenClassification)r*  RoFormerForTokenClassification)r]  !SqueezeBertForTokenClassification)r_  StableLmForTokenClassification)ra   Starcoder2ForTokenClassification)rs  T5ForTokenClassification)r  UMT5ForTokenClassification)r  XLMForTokenClassification)r   XLMRobertaForTokenClassification)r  "XLMRobertaXLForTokenClassification)r  XLNetForTokenClassification)r  XmodForTokenClassification)r  YosoForTokenClassification)&)r   AlbertForMultipleChoice)r*   BertForMultipleChoice)r/   BigBirdForMultipleChoice)rK   CamembertForMultipleChoice)rM   CanineForMultipleChoice)rp   ConvBertForMultipleChoice)r   Data2VecTextForMultipleChoice)r   DebertaV2ForMultipleChoice)r   DistilBertForMultipleChoice)r   ElectraForMultipleChoice)r   ErnieForMultipleChoice)r   ErnieMForMultipleChoice)r   FlaubertForMultipleChoice)r   FNetForMultipleChoice)r   FunnelForMultipleChoice)r&  IBertForMultipleChoice)rf  LongformerForMultipleChoice)rj  LukeForMultipleChoice)r  MegaForMultipleChoice)r  MegatronBertForMultipleChoice)r  MobileBertForMultipleChoice)r  MPNetForMultipleChoice)r  MraForMultipleChoice)r  NezhaForMultipleChoice)r  NystromformerForMultipleChoice)r  QDQBertForMultipleChoice)r  RemBertForMultipleChoice)r!  RobertaForMultipleChoice)r$  $RobertaPreLayerNormForMultipleChoice)r'  RoCBertForMultipleChoice)r*  RoFormerForMultipleChoice)r]  SqueezeBertForMultipleChoice)r  XLMForMultipleChoice)r  XLMRobertaForMultipleChoice)r  XLMRobertaXLForMultipleChoice)r  XLNetForMultipleChoice)r  XmodForMultipleChoice)r  YosoForMultipleChoice))r*   BertForNextSentencePrediction)r   ErnieForNextSentencePrediction)r   FNetForNextSentencePrediction)r  %MegatronBertForNextSentencePrediction)r  #MobileBertForNextSentencePrediction)r  NezhaForNextSentencePrediction)r   QDQBertForNextSentencePrediction))r   ASTForAudioClassification)r   &Data2VecAudioForSequenceClassification)r#  HubertForSequenceClassification)rE  SEWForSequenceClassification)rG  SEWDForSequenceClassification)r  "UniSpeechForSequenceClassification)r  %UniSpeechSatForSequenceClassification)r  !Wav2Vec2ForSequenceClassification)r  %Wav2Vec2BertForSequenceClassification)r  *Wav2Vec2ConformerForSequenceClassification)r  WavLMForSequenceClassification)r  WhisperForAudioClassification))r   Data2VecAudioForCTC)r#  HubertForCTC)r  MCTCTForCTC)rE  	SEWForCTC)rG  
SEWDForCTC)r  UniSpeechForCTC)r  UniSpeechSatForCTC)r  Wav2Vec2ForCTC)r  Wav2Vec2BertForCTC)r  Wav2Vec2ConformerForCTC)r  WavLMForCTC))r   (Data2VecAudioForAudioFrameClassification)r  'UniSpeechSatForAudioFrameClassification)r  #Wav2Vec2ForAudioFrameClassification)r  'Wav2Vec2BertForAudioFrameClassification)r  ,Wav2Vec2ConformerForAudioFrameClassification)r   WavLMForAudioFrameClassification))r   Data2VecAudioForXVector)r  UniSpeechSatForXVector)r  Wav2Vec2ForXVector)r  Wav2Vec2BertForXVector)r  Wav2Vec2ConformerForXVector)r  WavLMForXVectorr   )rX  SpeechT5ForTextToSpeech)	r!   rz   )r   FastSpeech2ConformerWithHifiGan)r   MusicgenForConditionalGeneration)r  &MusicgenMelodyForConditionalGeneration)qwen2_5_omni#Qwen2_5OmniForConditionalGeneration)r<  SeamlessM4TForTextToSpeech)r>  SeamlessM4Tv2ForTextToSpeechr  )	r   r   r>   )rA   Blip2ForImageTextRetrievalrQ   rX   r`   rI  rL  ))r'   BeitBackbone)r6   BitBackbone)rs   ConvNextBackbone)rv   ConvNextV2Backbone)r   DinatBackbone)r   Dinov2Backbone)r   Dinov2WithRegistersBackbone)r   FocalNetBackboner  )r!  HieraBackbone)r|  MaskFormerSwinBackbone)r  NatBackbone)r  PvtV2Backbone)r  ResNetBackbone)rt_detr_resnetRTDetrResNetBackbone)ri  SwinBackbone)ro  Swinv2Backbone)rz  TextNetBackboner  )r  VitDetBackbone)vitpose_backboneVitPoseBackboner2  r5  )
superpointSuperPointForKeypointDetection)r   r)   r.   r[   r   r   r   r   r   )r   Emu3TextModelr   r%  )rY  r\  re  )r  MllamaTextModelr  )r  MT5EncoderModelr  r  r  r   r#  r&  r)  r\  )rs  T5EncoderModel)r  UMT5EncoderModelr  r  r  )r  'PatchTSMixerForTimeSeriesClassification)r  PatchTSTForClassification)r  PatchTSMixerForRegression)r  PatchTSTForRegression)r~  TimesFmModelForPrediction)rl  Swin2SRForImageSuperResolutionc                       e Zd ZeZy)AutoModelForMaskGenerationN)__name__
__module____qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mapping     x/var/www/catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/transformers/models/auto/modeling_auto.pyr  r        6Nr&  r  c                       e Zd ZeZy)AutoModelForKeypointDetectionN)r   r!  r"  $MODEL_FOR_KEYPOINT_DETECTION_MAPPINGr$  r%  r&  r'  r*  r*        9Nr&  r*  c                       e Zd ZeZy)AutoModelForTextEncodingN)r   r!  r"  MODEL_FOR_TEXT_ENCODING_MAPPINGr$  r%  r&  r'  r.  r.        4Nr&  r.  c                       e Zd ZeZy)AutoModelForImageToImageN)r   r!  r"   MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGr$  r%  r&  r'  r2  r2    s    5Nr&  r2  c                       e Zd ZeZy)	AutoModelN)r   r!  r"  MODEL_MAPPINGr$  r%  r&  r'  r5  r5    s    "Nr&  r5  c                       e Zd ZeZy)AutoModelForPreTrainingN)r   r!  r"  MODEL_FOR_PRETRAINING_MAPPINGr$  r%  r&  r'  r8  r8    s    2Nr&  r8  pretraining)head_docc                       e Zd ZeZy)_AutoModelWithLMHeadN)r   r!  r"  MODEL_WITH_LM_HEAD_MAPPINGr$  r%  r&  r'  r=  r=        /Nr&  r=  zlanguage modelingc                       e Zd ZeZy)AutoModelForCausalLMN)r   r!  r"  MODEL_FOR_CAUSAL_LM_MAPPINGr$  r%  r&  r'  rA  rA        0Nr&  rA  zcausal language modelingc                       e Zd ZeZy)AutoModelForMaskedLMN)r   r!  r"  MODEL_FOR_MASKED_LM_MAPPINGr$  r%  r&  r'  rE  rE    rC  r&  rE  zmasked language modelingc                       e Zd ZeZy)AutoModelForSeq2SeqLMN)r   r!  r"  &MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGr$  r%  r&  r'  rH  rH        ;Nr&  rH  z&sequence-to-sequence language modelingzgoogle-t5/t5-base)r;  checkpoint_for_examplec                       e Zd ZeZy)"AutoModelForSequenceClassificationN)r   r!  r"  )MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGr$  r%  r&  r'  rM  rM    s    >Nr&  rM  zsequence classificationc                       e Zd ZeZy)AutoModelForQuestionAnsweringN)r   r!  r"  $MODEL_FOR_QUESTION_ANSWERING_MAPPINGr$  r%  r&  r'  rP  rP    r,  r&  rP  zquestion answeringc                       e Zd ZeZy)"AutoModelForTableQuestionAnsweringN)r   r!  r"  *MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGr$  r%  r&  r'  rS  rS        ?Nr&  rS  ztable question answeringzgoogle/tapas-base-finetuned-wtqc                       e Zd ZeZy)#AutoModelForVisualQuestionAnsweringN)r   r!  r"  +MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGr$  r%  r&  r'  rW  rW  	  s    @Nr&  rW  zvisual question answeringzdandelin/vilt-b32-finetuned-vqac                       e Zd ZeZy)%AutoModelForDocumentQuestionAnsweringN)r   r!  r"  -MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGr$  r%  r&  r'  rZ  rZ    s    BNr&  rZ  zdocument question answeringz/impira/layoutlm-document-qa", revision="52e01b3c                       e Zd ZeZy)AutoModelForTokenClassificationN)r   r!  r"  &MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGr$  r%  r&  r'  r]  r]    rJ  r&  r]  ztoken classificationc                       e Zd ZeZy)AutoModelForMultipleChoiceN)r   r!  r"  !MODEL_FOR_MULTIPLE_CHOICE_MAPPINGr$  r%  r&  r'  r`  r`  &  r(  r&  r`  zmultiple choicec                       e Zd ZeZy)"AutoModelForNextSentencePredictionN)r   r!  r"  *MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGr$  r%  r&  r'  rc  rc  -  rU  r&  rc  znext sentence predictionc                       e Zd ZeZy)AutoModelForImageClassificationN)r   r!  r"  &MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGr$  r%  r&  r'  rf  rf  6  rJ  r&  rf  zimage classificationc                       e Zd ZeZy)'AutoModelForZeroShotImageClassificationN)r   r!  r"  0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGr$  r%  r&  r'  ri  ri  =  s    ENr&  ri  zzero-shot image classificationc                       e Zd ZeZy)AutoModelForImageSegmentationN)r   r!  r"  $MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGr$  r%  r&  r'  rl  rl  F  r,  r&  rl  zimage segmentationc                       e Zd ZeZy) AutoModelForSemanticSegmentationN)r   r!  r"  'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGr$  r%  r&  r'  ro  ro  M      <Nr&  ro  zsemantic segmentationc                       e Zd ZeZy) AutoModelForTimeSeriesPredictionN)r   r!  r"  (MODEL_FOR_TIME_SERIES_PREDICTION_MAPPINGr$  r%  r&  r'  rs  rs  V      =Nr&  rs  ztime-series predictionc                       e Zd ZeZy)!AutoModelForUniversalSegmentationN)r   r!  r"  (MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGr$  r%  r&  r'  rw  rw  _  ru  r&  rw  zuniversal image segmentationc                       e Zd ZeZy) AutoModelForInstanceSegmentationN)r   r!  r"  'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGr$  r%  r&  r'  rz  rz  h  rq  r&  rz  zinstance segmentationc                       e Zd ZeZy)AutoModelForObjectDetectionN)r   r!  r"  "MODEL_FOR_OBJECT_DETECTION_MAPPINGr$  r%  r&  r'  r}  r}  q      7Nr&  r}  zobject detectionc                       e Zd ZeZy)#AutoModelForZeroShotObjectDetectionN)r   r!  r"  ,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGr$  r%  r&  r'  r  r  x      ANr&  r  zzero-shot object detectionc                       e Zd ZeZy)AutoModelForDepthEstimationN)r   r!  r"  "MODEL_FOR_DEPTH_ESTIMATION_MAPPINGr$  r%  r&  r'  r  r    r  r&  r  zdepth estimationc                       e Zd ZeZy)AutoModelForVideoClassificationN)r   r!  r"  &MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGr$  r%  r&  r'  r  r    rJ  r&  r  zvideo classificationc                       e Zd ZeZy)AutoModelForVision2SeqN)r   r!  r"  MODEL_FOR_VISION_2_SEQ_MAPPINGr$  r%  r&  r'  r  r    s    3Nr&  r  zvision-to-text modelingc                       e Zd ZeZy)AutoModelForImageTextToTextN)r   r!  r"  $MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPINGr$  r%  r&  r'  r  r    r,  r&  r  zimage-text-to-text modelingc                       e Zd ZeZy)AutoModelForAudioClassificationN)r   r!  r"  &MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGr$  r%  r&  r'  r  r    rJ  r&  r  zaudio classificationc                       e Zd ZeZy)AutoModelForCTCN)r   r!  r"  MODEL_FOR_CTC_MAPPINGr$  r%  r&  r'  r  r    s    *Nr&  r  z%connectionist temporal classificationc                       e Zd ZeZy)AutoModelForSpeechSeq2SeqN)r   r!  r"  "MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGr$  r%  r&  r'  r  r    r  r&  r  z,sequence-to-sequence speech-to-text modelingc                       e Zd ZeZy)$AutoModelForAudioFrameClassificationN)r   r!  r"  ,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGr$  r%  r&  r'  r  r    r  r&  r  z"audio frame (token) classificationc                       e Zd ZeZy)AutoModelForAudioXVectorN)r   r!  r"  MODEL_FOR_AUDIO_XVECTOR_MAPPINGr$  r%  r&  r'  r  r    r0  r&  r  c                       e Zd ZeZy)AutoModelForTextToSpectrogramN)r   r!  r"  %MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGr$  r%  r&  r'  r  r    s    :Nr&  r  c                       e Zd ZeZy)AutoModelForTextToWaveformN)r   r!  r"  "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPINGr$  r%  r&  r'  r  r    r  r&  r  c                       e Zd ZeZy)AutoBackboneN)r   r!  r"  MODEL_FOR_BACKBONE_MAPPINGr$  r%  r&  r'  r  r    r?  r&  r  zaudio retrieval via x-vectorc                       e Zd ZeZy)AutoModelForMaskedImageModelingN)r   r!  r"  'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGr$  r%  r&  r'  r  r    rq  r&  r  zmasked image modelingc                   <     e Zd Ze fd       Ze fd       Z xZS )AutoModelWithLMHeadc                 V    t        j                  dt               t        |   |      S NzThe class `AutoModelWithLMHead` is deprecated and will be removed in a future version. Please use `AutoModelForCausalLM` for causal language models, `AutoModelForMaskedLM` for masked language models and `AutoModelForSeq2SeqLM` for encoder-decoder models.)warningswarnFutureWarningsuperfrom_config)clsconfig	__class__s     r'  r  zAutoModelWithLMHead.from_config  s,    B 		
 w"6**r&  c                 ^    t        j                  dt               t        |   |g|i |S r  )r  r  r  r  from_pretrained)r  pretrained_model_name_or_path
model_argskwargsr  s       r'  r  z#AutoModelWithLMHead.from_pretrained  s8    B 		
 w&'D\z\U[\\r&  )r   r!  r"  classmethodr  r  __classcell__)r  s   @r'  r  r    s+    + + ] ]r&  r  )Tr  r  r  r  'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGrB  r  r[  r  rg  MODEL_FOR_IMAGE_MAPPINGrm  r3  r+  r{  r  rF  r#  ra  rd  r~  r9  rQ  rp  rI  rN  r  rT  r/  r  r  rt  r^  rx  r  r  MODEL_FOR_RETRIEVAL_MAPPINGr  rX  r6  r>  rj  r  ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGr5  r  r  r  r  rA  r  r  rf  rl  r2  rz  r*  r  r.  r  rE  r`  rc  r}  r8  rP  ro  rH  rM  r  rS  r  r  r]  rw  r  r  rW  rZ  r  ri  r  r  )__doc__r  collectionsr   utilsr   auto_factoryr   r   r	   r
   configuration_autor   
get_loggerr   loggerMODEL_MAPPING_NAMES#MODEL_FOR_PRETRAINING_MAPPING_NAMES MODEL_WITH_LM_HEAD_MAPPING_NAMES!MODEL_FOR_CAUSAL_LM_MAPPING_NAMESMODEL_FOR_IMAGE_MAPPING_NAMES-MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMES-MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING_NAMES,MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMES*MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMES-MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING_NAMES-MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING_NAMES.MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPING_NAMES,MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING_NAMES$MODEL_FOR_VISION_2_SEQ_MAPPING_NAMES!MODEL_FOR_RETRIEVAL_MAPPING_NAMES*MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING_NAMES!MODEL_FOR_MASKED_LM_MAPPING_NAMES(MODEL_FOR_OBJECT_DETECTION_MAPPING_NAMES2MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING_NAMES(MODEL_FOR_DEPTH_ESTIMATION_MAPPING_NAMES,MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMES(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMES/MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMES*MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMES0MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMES1MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING_NAMES3MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING_NAMES,MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMES'MODEL_FOR_MULTIPLE_CHOICE_MAPPING_NAMES0MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING_NAMES,MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMESMODEL_FOR_CTC_MAPPING_NAMES2MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPING_NAMES%MODEL_FOR_AUDIO_XVECTOR_MAPPING_NAMES+MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING_NAMES(MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING_NAMES6MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING_NAMES MODEL_FOR_BACKBONE_MAPPING_NAMES'MODEL_FOR_MASK_GENERATION_MAPPING_NAMES*MODEL_FOR_KEYPOINT_DETECTION_MAPPING_NAMES%MODEL_FOR_TEXT_ENCODING_MAPPING_NAMES2MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING_NAMES.MODEL_FOR_TIME_SERIES_REGRESSION_MAPPING_NAMES.MODEL_FOR_TIME_SERIES_PREDICTION_MAPPING_NAMES&MODEL_FOR_IMAGE_TO_IMAGE_MAPPING_NAMESr6  r9  r>  rB  r  rg  rj  rm  rp  r{  rx  r  r  r  r  rX  r[  rF  r  r  r~  r  r  rI  rN  rQ  rT  r^  ra  rd  r  r  r  r  r  r  r  r  r#  r+  r/  r  r  rt  r3  r  r*  r.  r2  r5  r8  r=  rA  rE  rH  rM  rP  rS  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rs  rw  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  __all__r%  r&  r'  <module>r     sZ     #   5 
		H	%!AC J
 '2SU' #n $/SU$  n %0qs% !j !,9;! z 1<1 - 1< 	71 - 0;AC0 ,J .9 	(. * 1<
1 - 1< 	<1 - 2=2 . 0;0 , (3( $6 %0*% ! .9#%. *N %002% !h ,7, (  6A6 2 ,7
, ( 0;0 ,B ,7
, ( 3>df3 /P .9JL. *\ 4? 	/4 0 5@5 1 7B7 3 0;PR0 ,h +6(*+ 'X 4?
4 0 0;0 ,$ * " 6A
6 2 )4
) % /: 	?// + ,7, ( :E: 6 $/$  6 +6+ ' +6 + ' .98. * )4!) %F 6AC16 2 2=5-2 . 2=02 . *55* & !!57JK 01EGj k -.BDde ./CEfg *:G+ ' *:F* & 4DP4 0 (8D( $ +;G+ ' +;G+ ' ,<H, ( *:F* & "22FHl!m '7D( $ //CEfg .>K/ + 1AM1 - //CEfg *+?A^_ *:G+ ' &66JLt%u "/?L0 , &66JLt%u ")9F* & -=I- ) (8D( $ .>J. * *:F* & %55IKr$s !-=J. * *:F* & ))=?Z[ %56JLt%u "/?L0 , #33GIn"o (8E) % &66JLt%u "-.BDde $45IKr$s !'7D( $ #33GIn"o /?L0 , ,<H, ( ,<H, ( $44HJp#q  7!4 7:$7 :52 562 6## # i(	31 3 ,,Cm\ 0. 0 ))=H[\ 1. 1 ))=Hbc 1. 1 ))=Hbc </ < *5. ?)< ? &7&1J& "
:$7 : !22OZn o @)< @ &7&'<& "A*= A '8'(<' #C,? C ):)*L) %<&9 < #44S^t"u 7!4 7 //ITef @)< @ &7&1K& "
<&9 < #44S^t"u F.A F +<+6V+ '
:$7 : !22OZn o =': = $5$/F$  
>': > $5$/G$  
>(; > %6%0N% !
=': = $5$/F$  
8"5 8 00KVhi B*= B '8'2N' #
8"5 8 00KVhi <&9 < #44S^t"u 40 4 ++ALef :"5 : 00KVst <&9 < #44S^t"u +) + $O>ef8 3 8 .(V 
B+> B (9(3W( $
52 5;$7 ;8!4 80) 0 --EPno =&9 = #44S^u"v ]. ],Ur&  