
    %	&h9                       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      Z7 eg d2      Z8 ed3d4g      Z9 ed5d6g      Z: ed7g      Z; e	ee      Z< e	ee      Z= e	ee      Z> e	ee      Z? e	ee      Z@ e	ee      ZA e	ee4      ZB e	ee      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e0      Z] e	ee1      Z^ e	ee2      Z_ e	ee3      Z` e	ee5      Za e	ee6      Zb e	ee7      Zc e	ee8      Zd e	ee9      Ze e	ee:      Zf e	ee;      Zg G d8 d9e      Zh G d: d;e      Zi G d< d=e      Zj G d> d?e      Zk G d@ dAe      Zl e
el      Zl G dB dCe      Zm e
emdDE      Zm G dF dGe      Zn e
endHE      Zn G dI dJe      Zo e
eodKE      Zo G dL dMe      Zp e
epdNE      Zp G dO dPe      Zq e
eqdQdRS      Zq G dT dUe      Zr e
erdVE      Zr G dW dXe      Zs e
esdYE      Zs G dZ d[e      Zt e
etd\d]S      Zt G d^ d_e      Zu e
eud`daS      Zu G db dce      Zv e
evdddeS      Zv G df dge      Zw e
ewdhE      Zw G di dje      Zx e
exdkE      Zx G dl dme      Zy e
eydnE      Zy G do dpe      Zz e
ezdqE      Zz G dr dse      Z{ e
e{dtE      Z{ G du dve      Z| e
e|dwE      Z| G dx dye      Z} e
e}dzE      Z} G d{ d|e      Z~ e
e~d}E      Z~ G d~ de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d de      Z G d de      Z G d de      Z G d de      Z e
edE      Z G d de      Z e
edE      Z G d den      Zy)zAuto Model class.    N)OrderedDict   )logging   )_BaseAutoBackboneClass_BaseAutoModelClass_LazyAutoMappingauto_class_update)CONFIG_MAPPING_NAMES(   albertAlbertModelalign
AlignModelaltclipAltCLIPModelariaAriaForConditionalGeneration)	aria_textAriaTextModel)audio-spectrogram-transformerASTModel)
autoformerAutoformerModel)bamba
BambaModelbark	BarkModel)bart	BartModelbeit	BeitModelbert	BertModel)bert-generationBertGenerationEncoderbig_birdBigBirdModel)bigbird_pegasusBigBirdPegasusModel)biogptBioGptModelbitBitModel)
blenderbotBlenderbotModel)blenderbot-smallBlenderbotSmallModelblip	BlipModel)blip-2
Blip2Model)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)ctrl	CTRLModel)cvtCvtModel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)z
donut-swinDonutSwinModel)dprDPRQuestionEncoderdptDPTModelefficientformerEfficientFormerModelefficientnetEfficientNetModelelectraElectraModel)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)gemma
GemmaModel)gemma2Gemma2Model)gemma3_textGemma3TextModel)gitGitModel)glmGlmModel)glm4	Glm4Modelglpn	GLPNModelgot_ocr2GotOcr2ForConditionalGeneration)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)granitemoesharedGraniteMoeSharedModel)
graphormerGraphormerModel)grounding-dinoGroundingDinoModel)groupvitGroupViTModel)heliumHeliumModelhiera
HieraModel)hubertHubertModelibert
IBertModel)ideficsIdeficsModel)idefics2Idefics2Model)idefics3Idefics3Model)idefics3_visionIdefics3VisionTransformerijepa
IJepaModelimagegptImageGPTModel)informerInformerModel)jamba
JambaModel)jetmoeJetMoeModel)jukeboxJukeboxModel)kosmos-2Kosmos2Model)layoutlmLayoutLMModel)
layoutlmv2LayoutLMv2Model)
layoutlmv3LayoutLMv3Model)ledLEDModellevit
LevitModel)lilt	LiltModel)llamara   llama4Llama4ForConditionalGeneration
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)mixtralMixtralModel
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)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_audio_encoderQwen2AudioEncoder)	qwen2_moeQwen2MoeModel)qwen2_vlQwen2VLModel)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_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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videomaeVideoMAEModel)vilt	ViltModel)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)P)r   AlbertForPreTrainingr#   BartForConditionalGeneration)r)   BertForPreTraining)r.   BigBirdForPreTrainingr@   BloomForCausalLMrF   CamembertForMaskedLMcolpaliColPaliForRetrievalru   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gemma3Gemma3ForConditionalGenerationr   GPT2LMHeadModelr   r  r   GPTBigCodeForCausalLMr   )r  HieraForPreTrainingr  IBertForMaskedLMr  IdeficsForVisionText2Textr   Idefics2ForConditionalGenerationr   Idefics3ForConditionalGenerationr+  LayoutLMForMaskedLMllavaLlavaForConditionalGeneration
llava_next!LlavaNextForConditionalGenerationllava_next_video&LlavaNextVideoForConditionalGenerationllava_onevision&LlavaOnevisionForConditionalGenerationr=  LongformerForMaskedLMrA  LukeForMaskedLM)rC  LxmertForPreTrainingrG  MambaForCausalLMrI  Mamba2ForCausalLMrY  MegaForMaskedLM)r[  MegatronBertForPreTrainingmistral3 Mistral3ForConditionalGenerationmllamaMllamaForConditionalGeneration)re  MobileBertForPreTrainingry  MPNetForMaskedLMr{  MptForCausalLMr}  MraForMaskedLMr  MvpForConditionalGeneration)r  NezhaForPreTrainingr  NllbMoeForConditionalGenerationr  OpenAIGPTLMHeadModel	paligemma!PaliGemmaForConditionalGenerationqwen2_audio"Qwen2AudioForConditionalGenerationr  r  RobertaForMaskedLMr  RobertaPreLayerNormForMaskedLM)r  RoCBertForPreTrainingr  RwkvForCausalLM)r  SplinterForPreTrainingr"  SqueezeBertForMaskedLMr6  *SwitchTransformersForConditionalGenerationr8  T5ForConditionalGenerationr=  TapasForMaskedLMrN  TransfoXLLMHeadModel)rP  TvltForPreTraining)rX  UniSpeechForPreTraining)rZ  UniSpeechSatForPreTrainingvideo_llava"VideoLlavaForConditionalGeneration)rb  VideoMAEForPreTrainingvipllava VipLlavaForConditionalGeneration)rg  VisualBertForPreTraining)rp  ViTMAEForPreTraining)r~  Wav2Vec2ForPreTraining)r  Wav2Vec2ConformerForPreTrainingr  XLMWithLMHeadModelr  XLMRobertaForMaskedLMr  XLMRobertaXLForMaskedLMr  XLNetLMHeadModelr  XmodForMaskedLM)Qr   AlbertForMaskedLMr  r)   BertForMaskedLMr.   BigBirdForMaskedLMr0   &BigBirdPegasusForConditionalGenerationr9   'BlenderbotSmallForConditionalGenerationr  r  rb   CodeGenForCausalLMrk   ConvBertForMaskedLMrs   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  r1  LEDForConditionalGenerationr  r?  LongT5ForConditionalGenerationr  rE  M2M100ForConditionalGenerationr  r  rK  MarianMTModelr  r[  MegatronBertForCausalLMre  MobileBertForMaskedLMru  !MoonshineForConditionalGenerationr  r  r  r  r  NezhaForMaskedLMr   r  NystromformerForMaskedLMr  r   PegasusXForConditionalGenerationr  PLBartForConditionalGeneration	pop2piano!Pop2PianoForConditionalGenerationr  QDQBertForMaskedLMr  ReformerModelWithLMHeadr  RemBertForMaskedLMr
  r  r  RoCBertForMaskedLMr  RoFormerForMaskedLMr  r  #Speech2TextForConditionalGenerationr  r  r  r  r  r~  Wav2Vec2ForMaskedLMr  WhisperForConditionalGenerationr*  r,  r.  r0  r2  r  YosoForMaskedLM)m)r   AriaTextForCausalLM)r   BambaForCausalLM)r#   BartForCausalLM)r)   BertLMHeadModel)r+   BertGenerationDecoder)r.   BigBirdForCausalLM)r0   BigBirdPegasusForCausalLM)r2   BioGptForCausalLM)r7   BlenderbotForCausalLM)r9   BlenderbotSmallForCausalLMr  )rF   CamembertForCausalLM)r`   LlamaForCausalLMr>  )rd   CohereForCausalLM)rf   Cohere2ForCausalLMrB  r  )r   Data2VecTextForCausalLM)r   DbrxForCausalLM)r   DeepseekV3ForCausalLM)r   DiffLlamaForCausalLM)r   ElectraForCausalLM)emu3Emu3ForCausalLM)r   ErnieForCausalLM)r   FalconForCausalLMr  fuyuFuyuForCausalLM)r   GemmaForCausalLM)r   Gemma2ForCausalLMr  )r   Gemma3ForCausalLMrP  )r   GlmForCausalLM)r   Glm4ForCausalLMr   r  r  r  rR  rT  rV  rX  )r   GraniteForCausalLM)r   GraniteMoeForCausalLM)r  GraniteMoeSharedForCausalLM)r	  HeliumForCausalLM)r#  JambaForCausalLM)r%  JetMoeForCausalLM)r8  r  )r:  Llama4ForCausalLM)llama4_textr  r  r  )rK  MarianForCausalLM)rU  MBartForCausalLM)rY  MegaForCausalLMrb  )r`  MistralForCausalLM)rb  MixtralForCausalLM)r  MllamaForCausalLM)rw  MoshiForCausalLMr  )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  RecurrentGemmaForCausalLMru  )r  RemBertForCausalLM)r  RobertaForCausalLM)r  RobertaPreLayerNormForCausalLM)r  RoCBertForCausalLM)r  RoFormerForCausalLMr  )speech_to_text_2Speech2Text2ForCausalLM)r$  StableLmForCausalLM)r&  Starcoder2ForCausalLMr  )trocrTrOCRForCausalLM)r  WhisperForCausalLM)r  XGLMForCausalLMr*  )r  XLMProphetNetForCausalLM)r  XLMRobertaForCausalLM)r  XLMRobertaXLForCausalLMr0  )r  XmodForCausalLM)r  ZambaForCausalLM)r  Zamba2ForCausalLM)6r%   r4   rh   rm   rp   ry   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r3  )r:  Llama4VisionModel)r  MllamaVisionModelrg  rj  rm  rp  r  r  r  r  r  r  r  r*  r-  r0  r3  r:  rC  rF  rI  r^  ra  ri  rl  ro  rr  ru  r{  r  ))r   DeiTForMaskedImageModeling)r   FocalNetForMaskedImageModeling)r.  SwinForMaskedImageModeling)r4  Swinv2ForMaskedImageModeling)rj  ViTForMaskedImageModeling)r  ImageGPTForCausalImageModeling)*)r&   BeitForImageClassification)r5   BitForImageClassification)rT   CLIPForImageClassification)rn   ConvNextForImageClassification)rq    ConvNextV2ForImageClassification)rw   CvtForImageClassification)r   $Data2VecVisionForImageClassification)r   )DeiTForImageClassification%DeiTForImageClassificationWithTeacher)r   DinatForImageClassification)r   Dinov2ForImageClassification)r   )Dinov2WithRegistersForImageClassification)r   )%EfficientFormerForImageClassification0EfficientFormerForImageClassificationWithTeacher)r   "EfficientNetForImageClassification)r   FocalNetForImageClassification)r  HieraForImageClassification)r  IJepaForImageClassification)r  ImageGPTForImageClassification)r4  )LevitForImageClassification&LevitForImageClassificationWithTeacher)rh  !MobileNetV1ForImageClassification)rk  !MobileNetV2ForImageClassification)rn  MobileViTForImageClassification)rq  !MobileViTV2ForImageClassification)r  NatForImageClassification)r  )&PerceiverForImageClassificationLearned&PerceiverForImageClassificationFourier-PerceiverForImageClassificationConvProcessing)r   PoolFormerForImageClassification)r  PvtForImageClassification)r  PvtV2ForImageClassification)r  RegNetForImageClassification)r  ResNetForImageClassification)r  SegformerForImageClassification)shieldgemma2"ShieldGemma2ForImageClassification)r  SiglipForImageClassification)r  Siglip2ForImageClassification)r+  !SwiftFormerForImageClassification)r.  SwinForImageClassification)r4  Swinv2ForImageClassification)r?  TextNetForImageClassification)rJ  !TimmWrapperForImageClassification)r_  VanForImageClassification)rj  ViTForImageClassification)rm  ViTHybridForImageClassification)rs  ViTMSNForImageClassificationr   DetrForSegmentation))r&   BeitForSemanticSegmentation)r   %Data2VecVisionForSemanticSegmentation)r   DPTForSemanticSegmentation)rk  "MobileNetV2ForSemanticSegmentation)rn   MobileViTForSemanticSegmentation)rq  "MobileViTV2ForSemanticSegmentation)r   SegformerForSemanticSegmentation)upernetUperNetForSemanticSegmentationrQ  !MaskFormerForInstanceSegmentation)r  )rO  #Mask2FormerForUniversalSegmentationr   )r  !OneFormerForUniversalSegmentation))rD  !TimesformerForVideoClassification)rb  VideoMAEForVideoClassification)r|  VivitForVideoClassification)r<   BlipForConditionalGenerationr>   Blip2ForConditionalGenerationrJ   !ChameleonForConditionalGenerationrP  r  r  instructblip$InstructBlipForConditionalGeneration)instructblipvideo)InstructBlipVideoForConditionalGenerationr)  Kosmos2ForConditionalGenerationr  r  r  r  r  r  r  
pix2struct"Pix2StructForConditionalGenerationr  "Qwen2_5_VLForConditionalGenerationr  Qwen2VLForConditionalGenerationr  r#  zvision-encoder-decoderVisionEncoderDecoderModelr  )r   )
aya_vision!AyaVisionForConditionalGenerationr'  r)  r+  )r  Emu3ForConditionalGenerationr  r  rP  r   r  r  r  r-  r2  r9  r  r  r  r  r  r  r4  )r  r  r7  r9  )r  r  )r  SmolVLMForConditionalGeneration)rT  UdopForConditionalGenerationr#  r;  ).r4  r  r6  r8  r  r@  r  r  r  r  rD  rH  rJ  r  rL  rN  r  r  r  r  rU  MBartForConditionalGenerationr  )r[  MegatronBertForMaskedLMrd  )rs  ModernBertForMaskedLMr  r  r  rh  rj  )r  PerceiverForMaskedLMrs  )r  ReformerForMaskedLMrw  r
  r  ry  r{  r  r  r  r*  r,  r.  r2  r  )	)ri   !ConditionalDetrForObjectDetection)rz   DabDetrForObjectDetection)r    DeformableDetrForObjectDetection)r   DetaForObjectDetection)r   DetrForObjectDetection)r  RTDetrForObjectDetection)r  RTDetrV2ForObjectDetection)r;  "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  r:  )r7   "BlenderbotForConditionalGenerationr<  rF  r  r   rZ  r\  r^  r`  rB  )r  MT5ForConditionalGenerationr  r   )r  PegasusForConditionalGenerationrl  rn  )r  "ProphetNetForConditionalGenerationr  )r  SeamlessM4TForTextToText)r  SeamlessM4Tv2ForTextToTextr  r  )rV  UMT5ForConditionalGeneration)r  %XLMProphetNetForConditionalGeneration)rf  rp  )r  SeamlessM4TForSpeechToText)r  SeamlessM4Tv2ForSpeechToText)zspeech-encoder-decoderSpeechEncoderDecoderModelr}  )r  SpeechT5ForSpeechToTextr  )b)r   AlbertForSequenceClassification)r#   BartForSequenceClassification)r)   BertForSequenceClassification)r.    BigBirdForSequenceClassification)r0   'BigBirdPegasusForSequenceClassification)r2   BioGptForSequenceClassification)r@   BloomForSequenceClassification)rF   "CamembertForSequenceClassification)rH   CanineForSequenceClassification)r`   LlamaForSequenceClassification)rk   !ConvBertForSequenceClassification)ru   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)r%  JetMoeForSequenceClassification)r+  !LayoutLMForSequenceClassification)r-  #LayoutLMv2ForSequenceClassification)r/  #LayoutLMv3ForSequenceClassification)r1  LEDForSequenceClassification)r6  LiltForSequenceClassification)r8  rr  )r=  #LongformerForSequenceClassification)rA  LukeForSequenceClassification)rM  !MarkupLMForSequenceClassification)rU  MBartForSequenceClassification)rY  MegaForSequenceClassification)r[  %MegatronBertForSequenceClassification)r`   MistralForSequenceClassification)rb   MixtralForSequenceClassification)re  #MobileBertForSequenceClassification)rs  #ModernBertForSequenceClassification)ry  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)r&  #Starcoder2ForSequenceClassification)r8  T5ForSequenceClassification)r=  TapasForSequenceClassification)rN  "TransfoXLForSequenceClassification)rV  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)r0   "BigBirdPegasusForQuestionAnswering)r@   BloomForQuestionAnswering)rF   CamembertForQuestionAnswering)rH   CanineForQuestionAnswering)rk   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r-  LayoutLMv2ForQuestionAnsweringr/  LayoutLMv3ForQuestionAnswering)r1  LEDForQuestionAnswering)r6  LiltForQuestionAnswering)r8  LlamaForQuestionAnswering)r=  LongformerForQuestionAnswering)rA  LukeForQuestionAnswering)rC  LxmertForQuestionAnswering)rM  MarkupLMForQuestionAnswering)rU  MBartForQuestionAnswering)rY  MegaForQuestionAnswering)r[   MegatronBertForQuestionAnswering)r`  MistralForQuestionAnswering)rb  MixtralForQuestionAnswering)re  MobileBertForQuestionAnswering)rs  ModernBertForQuestionAnswering)ry  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)r  SplinterForQuestionAnswering)r"  SqueezeBertForQuestionAnswering)r8  T5ForQuestionAnswering)rV  UMT5ForQuestionAnswering)r  XLMForQuestionAnsweringSimple)r  XLMRobertaForQuestionAnswering)r   XLMRobertaXLForQuestionAnswering)r  XLNetForQuestionAnsweringSimple)r  XmodForQuestionAnswering)r  YosoForQuestionAnswering)r=  TapasForQuestionAnswering))r<   BlipForQuestionAnsweringr)  )rd  ViltForQuestionAnswering))r+  LayoutLMForQuestionAnsweringr  r  )N)r   AlbertForTokenClassification)r)   BertForTokenClassification)r.   BigBirdForTokenClassification)r2   BioGptForTokenClassification)r@   BloomForTokenClassification)rD   BrosForTokenClassification)rF   CamembertForTokenClassification)rH   CanineForTokenClassification)rk   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   r1  )r    GPTBigCodeForTokenClassification)r   GPTNeoForTokenClassification)r   GPTNeoXForTokenClassification)r	  HeliumForTokenClassification)r  IBertForTokenClassification)r+  LayoutLMForTokenClassification)r-   LayoutLMv2ForTokenClassification)r/   LayoutLMv3ForTokenClassification)r6  LiltForTokenClassification)r8  LlamaForTokenClassification)r=   LongformerForTokenClassification)rA  LukeForTokenClassification)rM  MarkupLMForTokenClassification)rY  MegaForTokenClassification)r[  "MegatronBertForTokenClassification)r`  MistralForTokenClassification)rb  MixtralForTokenClassification)re   MobileBertForTokenClassification)rs   ModernBertForTokenClassification)ry  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)r&   Starcoder2ForTokenClassification)r8  T5ForTokenClassification)rV  UMT5ForTokenClassification)r  XLMForTokenClassification)r   XLMRobertaForTokenClassification)r  "XLMRobertaXLForTokenClassification)r  XLNetForTokenClassification)r  XmodForTokenClassification)r  YosoForTokenClassification)&)r   AlbertForMultipleChoice)r)   BertForMultipleChoice)r.   BigBirdForMultipleChoice)rF   CamembertForMultipleChoice)rH   CanineForMultipleChoice)rk   ConvBertForMultipleChoice)r   Data2VecTextForMultipleChoice)r   DebertaV2ForMultipleChoice)r   DistilBertForMultipleChoice)r   ElectraForMultipleChoice)r   ErnieForMultipleChoice)r   ErnieMForMultipleChoice)r   FlaubertForMultipleChoice)r   FNetForMultipleChoice)r   FunnelForMultipleChoice)r  IBertForMultipleChoice)r=  LongformerForMultipleChoice)rA  LukeForMultipleChoice)rY  MegaForMultipleChoice)r[  MegatronBertForMultipleChoice)re  MobileBertForMultipleChoice)ry  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)re  #MobileBertForNextSentencePrediction)r  NezhaForNextSentencePrediction)r   QDQBertForNextSentencePrediction))r   ASTForAudioClassification)r~   &Data2VecAudioForSequenceClassification)r  HubertForSequenceClassification)r
  SEWForSequenceClassification)r  SEWDForSequenceClassification)rX  "UniSpeechForSequenceClassification)rZ  %UniSpeechSatForSequenceClassification)r~  !Wav2Vec2ForSequenceClassification)r  %Wav2Vec2BertForSequenceClassification)r  *Wav2Vec2ConformerForSequenceClassification)r  WavLMForSequenceClassification)r  WhisperForAudioClassification))r~   Data2VecAudioForCTC)r  HubertForCTC)rW  MCTCTForCTC)r
  	SEWForCTC)r  
SEWDForCTC)rX  UniSpeechForCTC)rZ  UniSpeechSatForCTC)r~  Wav2Vec2ForCTC)r  Wav2Vec2BertForCTC)r  Wav2Vec2ConformerForCTC)r  WavLMForCTC))r~   (Data2VecAudioForAudioFrameClassification)rZ  'UniSpeechSatForAudioFrameClassification)r~  #Wav2Vec2ForAudioFrameClassification)r  'Wav2Vec2BertForAudioFrameClassification)r  ,Wav2Vec2ConformerForAudioFrameClassification)r   WavLMForAudioFrameClassification))r~   Data2VecAudioForXVector)rZ  UniSpeechSatForXVector)r~  Wav2Vec2ForXVector)r  Wav2Vec2BertForXVector)r  Wav2Vec2ConformerForXVector)r  WavLMForXVectorr   )r  SpeechT5ForTextToSpeech)r    )r   FastSpeech2ConformerWithHifiGan)r   MusicgenForConditionalGeneration)r  &MusicgenMelodyForConditionalGeneration)r  SeamlessM4TForTextToSpeech)r  SeamlessM4Tv2ForTextToSpeechrx  )	r   r   r;   )r>   Blip2ForImageTextRetrievalrL   rS   r[   r  r  ))r&   BeitBackbone)r5   BitBackbone)rn   ConvNextBackbone)rq   ConvNextV2Backbone)r   DinatBackbone)r   Dinov2Backbone)r   Dinov2WithRegistersBackbone)r   FocalNetBackbone)r  HieraBackbone)rS  MaskFormerSwinBackbone)r  NatBackbone)r  PvtV2Backbone)r  ResNetBackbone)rt_detr_resnetRTDetrResNetBackbone)r.  SwinBackbone)r4  Swinv2Backbone)r?  TextNetBackbonerF  )rv  VitDetBackbone)vitpose_backboneVitPoseBackboner  )
superpointSuperPointForKeypointDetection)r   r(   r-   rV   r   r   r   r   r   )r  Emu3TextModelr   r  )r:  Llama4TextModelr<  )r  MllamaTextModelrd  )r  MT5EncoderModelr  r  r  r  r  r  r  r!  )r8  T5EncoderModel)rV  UMT5EncoderModelr  r  r  )r  'PatchTSMixerForTimeSeriesClassification)r  PatchTSTForClassification)r  PatchTSMixerForRegression)r  PatchTSTForRegression)r1  Swin2SRForImageSuperResolutionc                       e Zd ZeZy)AutoModelForMaskGenerationN)__name__
__module____qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mapping     |/var/www/pru.catia.catastroantioquia-mas.com/valormas/lib/python3.12/site-packages/transformers/models/auto/modeling_auto.pyr  r  q      6Nr  r  c                       e Zd ZeZy)AutoModelForKeypointDetectionN)r  r  r  $MODEL_FOR_KEYPOINT_DETECTION_MAPPINGr  r  r  r  r  r  u      9Nr  r  c                       e Zd ZeZy)AutoModelForTextEncodingN)r  r  r  MODEL_FOR_TEXT_ENCODING_MAPPINGr  r  r  r  r  r  y      4Nr  r  c                       e Zd ZeZy)AutoModelForImageToImageN)r  r  r   MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGr  r  r  r  r  r  }  s    5Nr  r  c                       e Zd ZeZy)	AutoModelN)r  r  r  MODEL_MAPPINGr  r  r  r  r  r    s    "Nr  r  c                       e Zd ZeZy)AutoModelForPreTrainingN)r  r  r  MODEL_FOR_PRETRAINING_MAPPINGr  r  r  r  r  r    s    2Nr  r  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  r   r         0Nr  r   zcausal language modelingc                       e Zd ZeZy)AutoModelForMaskedLMN)r  r  r  MODEL_FOR_MASKED_LM_MAPPINGr  r  r  r  r  r    r  r  r  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  r  r        ;Nr  r  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  r  r    s    >Nr  r  zsequence classificationc                       e Zd ZeZy)AutoModelForQuestionAnsweringN)r  r  r  $MODEL_FOR_QUESTION_ANSWERING_MAPPINGr  r  r  r  r  r    r  r  r  zquestion answeringc                       e Zd ZeZy)"AutoModelForTableQuestionAnsweringN)r  r  r  *MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGr  r  r  r  r  r        ?Nr  r  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  r  r    s    @Nr  r  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  r  r    s    BNr  r  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    r	  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  r"  r"    r  r  r"  znext sentence predictionc                       e Zd ZeZy)AutoModelForImageClassificationN)r  r  r  &MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGr  r  r  r  r%  r%    r	  r  r%  zimage classificationc                       e Zd ZeZy)'AutoModelForZeroShotImageClassificationN)r  r  r  0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGr  r  r  r  r(  r(    s    ENr  r(  zzero-shot image classificationc                       e Zd ZeZy)AutoModelForImageSegmentationN)r  r  r  $MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGr  r  r  r  r+  r+    r  r  r+  zimage segmentationc                       e Zd ZeZy) AutoModelForSemanticSegmentationN)r  r  r  'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGr  r  r  r  r.  r.        <Nr  r.  zsemantic segmentationc                       e Zd ZeZy)!AutoModelForUniversalSegmentationN)r  r  r  (MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGr  r  r  r  r2  r2    s    =Nr  r2  zuniversal image segmentationc                       e Zd ZeZy) AutoModelForInstanceSegmentationN)r  r  r  'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGr  r  r  r  r5  r5  !  r0  r  r5  zinstance segmentationc                       e Zd ZeZy)AutoModelForObjectDetectionN)r  r  r  "MODEL_FOR_OBJECT_DETECTION_MAPPINGr  r  r  r  r8  r8  *      7Nr  r8  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<  1      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  rC  rC  A  r	  r  rC  zvideo classificationc                       e Zd ZeZy)AutoModelForVision2SeqN)r  r  r  MODEL_FOR_VISION_2_SEQ_MAPPINGr  r  r  r  rF  rF  H  s    3Nr  rF  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  rI  rI  O  r  r  rI  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  rL  rL  V  r	  r  rL  zaudio classificationc                       e Zd ZeZy)AutoModelForCTCN)r  r  r  MODEL_FOR_CTC_MAPPINGr  r  r  r  rO  rO  ]  s    *Nr  rO  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  rR  rR  d  r:  r  rR  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  rU  rU  m  r>  r  rU  z"audio frame (token) classificationc                       e Zd ZeZy)AutoModelForAudioXVectorN)r  r  r  MODEL_FOR_AUDIO_XVECTOR_MAPPINGr  r  r  r  rX  rX  v  r  r  rX  c                       e Zd ZeZy)AutoModelForTextToSpectrogramN)r  r  r  %MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGr  r  r  r  r[  r[  z  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  ra  ra    r  r  ra  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  rd  rd    r0  r  rd  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  rn  zAutoModelWithLMHead.from_config  s,    B 		
 w"6**r  c                 ^    t        j                  dt               t        |   |g|i |S ri  )rj  rk  rl  rm  from_pretrained)ro  pretrained_model_name_or_path
model_argskwargsrq  s       r  rs  z#AutoModelWithLMHead.from_pretrained  s8    B 		
 w&'D\z\U[\\r  )r  r  r  classmethodrn  rs  __classcell__)rq  s   @r  rg  rg    s+    + + ] ]r  rg  )__doc__rj  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_IMAGE_TO_IMAGE_MAPPING_NAMESr  r  r  r  'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGr&  r)  r,  r/  r6  r3  rD  rG  rJ  MODEL_FOR_RETRIEVAL_MAPPINGr  r  r  MODEL_FOR_IMAGE_MAPPINGre  r9  r=  rA  r  r  r  r  r  r   r#  rM  rP  rS  rV  rY  r\  r_  rb  r  r  r  ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGr  r  r  r  r  r  r  r  r   r  r  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r2  r5  r8  r<  r@  rC  rF  rI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  r  r  r  <module>r     s
     #   5 
		H	%!bd L	 '2RT' #l $/SU$  n %0oq% !f !,8:! x 1<1 - 1< 	71 - 0;?A0 ,F .9 	(. * 1<
1 - 1< 	<1 - 2=2 . 0;0 , (3( $6 %0*% ! .9 ". *H %002% !h ,7, ( 6A6 2 ,7
, ( 0;0 ,@ ,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 $/$  4 +6+ ' .98. * )4!) %F 6AC16 2 2=5-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, ( $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$  
>(; > %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 ]. ]r  