
    fThkQ                       S r SSKrSSKJr  SSKJr  SSKJrJrJ	r	J
r
  SSKJr  \R                  " \5      r\" / S	Q5      r\" / S
Q5      r\" / SQ5      r\" / SQ5      r\" / SQ5      r\" / SQ5      r\" S/5      r\" / SQ5      r\" S/5      r\" / SQ5      r\" S/5      r\" / SQ5      r\" / SQ5      r\" / SQ5      r\" S/5      r\" / SQ5      r\" / SQ5      r \" / SQ5      r!\" / SQ5      r"\" / SQ5      r#\" / SQ5      r$\" / SQ5      r%\" / SQ5      r&\" / S Q5      r'\" S!/5      r(\" / S"Q5      r)\" / S#Q5      r*\" / S$Q5      r+\" / S%Q5      r,\" / S&Q5      r-\" / S'Q5      r.\" / S(Q5      r/\" / S)Q5      r0\" / S*Q5      r1\" S+S,/5      r2\" / S-Q5      r3\" / S.Q5      r4\" / S/Q5      r5\" S0/5      r6\" S1/5      r6\" S2/5      r7\" / S3Q5      r8\" S4S5/5      r9\" S6S7/5      r:\" S8/5      r;\" S9/5      r<\	" \\5      r=\	" \\5      r>\	" \\5      r?\	" \\5      r@\	" \\5      rA\	" \\5      rB\	" \\45      rC\	" \\5      rD\	" \\5      rE\	" \\5      rF\	" \\5      rG\	" \\5      rH\	" \\5      rI\	" \\5      rJ\	" \\5      rK\	" \\)5      rL\	" \\*5      rM\	" \\ 5      rN\	" \\5      rO\	" \\5      rP\	" \\!5      rQ\	" \\"5      rR\	" \\#5      rS\	" \\$5      rT\	" \\&5      rU\	" \\'5      rV\	" \\(5      rW\	" \\+5      rX\	" \\,5      rY\	" \\-5      rZ\	" \\.5      r[\	" \\/5      r\\	" \\%5      r]\	" \\05      r^\	" \\15      r_\	" \\25      r`\	" \\35      ra\	" \\55      rb\	" \\65      rc\	" \\75      rd\	" \\85      re\	" \\95      rf\	" \\:5      rg\	" \\;5      rh\	" \\<5      ri " S: S;\5      rj " S< S=\5      rk " S> S?\5      rl " S@ SA\5      rm " SB SC\5      rn\
" \n5      rn " SD SE\5      ro\
" \oSFSG9ro " SH SI\5      rp\
" \pSJSG9rp " SK SL\5      rq\
" \qSMSG9rq " SN SO\5      rr\
" \rSPSG9rr " SQ SR\5      rs\
" \sSSSTSU9rs " SV SW\5      rt\
" \tSXSG9rt " SY SZ\5      ru\
" \uS[SG9ru " S\ S]\5      rv\
" \vS^S_SU9rv " S` Sa\5      rw\
" \wSbScSU9rw " Sd Se\5      rx\
" \xSfSgSU9rx " Sh Si\5      ry\
" \ySjSG9ry " Sk Sl\5      rz\
" \zSmSG9rz " Sn So\5      r{\
" \{SpSG9r{ " Sq Sr\5      r|\
" \|SsSG9r| " St Su\5      r}\
" \}SvSG9r} " Sw Sx\5      r~\
" \~SySG9r~ " Sz S{\5      r\
" \S|SG9r " S} S~\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\5      r " S S\5      r " S S\5      r " S S\5      r\
" \SSG9r " S S\5      r\
" \SSG9r " S S\p5      r/ SQrg)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                       \ rS rSr\rSrg)AutoModelForMaskGenerationi   N)__name__
__module____qualname____firstlineno__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mapping__static_attributes__r       ^/var/www/auris/envauris/lib/python3.13/site-packages/transformers/models/auto/modeling_auto.pyr  r        6Nr(  r  c                       \ rS rSr\rSrg)AutoModelForKeypointDetectioni  r   N)r!  r"  r#  r$  $MODEL_FOR_KEYPOINT_DETECTION_MAPPINGr&  r'  r   r(  r)  r,  r,        9Nr(  r,  c                       \ rS rSr\rSrg)AutoModelForTextEncodingi  r   N)r!  r"  r#  r$  MODEL_FOR_TEXT_ENCODING_MAPPINGr&  r'  r   r(  r)  r0  r0        4Nr(  r0  c                       \ rS rSr\rSrg)AutoModelForImageToImagei  r   N)r!  r"  r#  r$   MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGr&  r'  r   r(  r)  r4  r4    s    5Nr(  r4  c                       \ rS rSr\rSrg)	AutoModeli  r   N)r!  r"  r#  r$  MODEL_MAPPINGr&  r'  r   r(  r)  r7  r7    s    "Nr(  r7  c                       \ rS rSr\rSrg)AutoModelForPreTrainingi  r   N)r!  r"  r#  r$  MODEL_FOR_PRETRAINING_MAPPINGr&  r'  r   r(  r)  r:  r:    s    2Nr(  r:  pretraining)head_docc                       \ rS rSr\rSrg)_AutoModelWithLMHeadi  r   N)r!  r"  r#  r$  MODEL_WITH_LM_HEAD_MAPPINGr&  r'  r   r(  r)  r?  r?        /Nr(  r?  zlanguage modelingc                       \ rS rSr\rSrg)AutoModelForCausalLMi  r   N)r!  r"  r#  r$  MODEL_FOR_CAUSAL_LM_MAPPINGr&  r'  r   r(  r)  rC  rC        0Nr(  rC  zcausal language modelingc                       \ rS rSr\rSrg)AutoModelForMaskedLMi  r   N)r!  r"  r#  r$  MODEL_FOR_MASKED_LM_MAPPINGr&  r'  r   r(  r)  rG  rG    rE  r(  rG  zmasked language modelingc                       \ rS rSr\rSrg)AutoModelForSeq2SeqLMi  r   N)r!  r"  r#  r$  &MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGr&  r'  r   r(  r)  rJ  rJ        ;Nr(  rJ  z&sequence-to-sequence language modelingzgoogle-t5/t5-base)r=  checkpoint_for_examplec                       \ rS rSr\rSrg)"AutoModelForSequenceClassificationi  r   N)r!  r"  r#  r$  )MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  rO  rO    s    >Nr(  rO  zsequence classificationc                       \ rS rSr\rSrg)AutoModelForQuestionAnsweringi  r   N)r!  r"  r#  r$  $MODEL_FOR_QUESTION_ANSWERING_MAPPINGr&  r'  r   r(  r)  rR  rR    r.  r(  rR  zquestion answeringc                       \ rS rSr\rSrg)"AutoModelForTableQuestionAnsweringi  r   N)r!  r"  r#  r$  *MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGr&  r'  r   r(  r)  rU  rU        ?Nr(  rU  ztable question answeringzgoogle/tapas-base-finetuned-wtqc                       \ rS rSr\rSrg)#AutoModelForVisualQuestionAnsweringi	  r   N)r!  r"  r#  r$  +MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGr&  r'  r   r(  r)  rY  rY  	  s    @Nr(  rY  zvisual question answeringzdandelin/vilt-b32-finetuned-vqac                       \ rS rSr\rSrg)%AutoModelForDocumentQuestionAnsweringi  r   N)r!  r"  r#  r$  -MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGr&  r'  r   r(  r)  r\  r\    s    BNr(  r\  zdocument question answeringz/impira/layoutlm-document-qa", revision="52e01b3c                       \ rS rSr\rSrg)AutoModelForTokenClassificationi  r   N)r!  r"  r#  r$  &MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  r_  r_    rL  r(  r_  ztoken classificationc                       \ rS rSr\rSrg)AutoModelForMultipleChoicei&  r   N)r!  r"  r#  r$  !MODEL_FOR_MULTIPLE_CHOICE_MAPPINGr&  r'  r   r(  r)  rb  rb  &  r*  r(  rb  zmultiple choicec                       \ rS rSr\rSrg)"AutoModelForNextSentencePredictioni-  r   N)r!  r"  r#  r$  *MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGr&  r'  r   r(  r)  re  re  -  rW  r(  re  znext sentence predictionc                       \ rS rSr\rSrg)AutoModelForImageClassificationi6  r   N)r!  r"  r#  r$  &MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  rh  rh  6  rL  r(  rh  zimage classificationc                       \ rS rSr\rSrg)'AutoModelForZeroShotImageClassificationi=  r   N)r!  r"  r#  r$  0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  rk  rk  =  s    ENr(  rk  zzero-shot image classificationc                       \ rS rSr\rSrg)AutoModelForImageSegmentationiF  r   N)r!  r"  r#  r$  $MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGr&  r'  r   r(  r)  rn  rn  F  r.  r(  rn  zimage segmentationc                       \ rS rSr\rSrg) AutoModelForSemanticSegmentationiM  r   N)r!  r"  r#  r$  'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGr&  r'  r   r(  r)  rq  rq  M      <Nr(  rq  zsemantic segmentationc                       \ rS rSr\rSrg) AutoModelForTimeSeriesPredictioniV  r   N)r!  r"  r#  r$  (MODEL_FOR_TIME_SERIES_PREDICTION_MAPPINGr&  r'  r   r(  r)  ru  ru  V      =Nr(  ru  ztime-series predictionc                       \ rS rSr\rSrg)!AutoModelForUniversalSegmentationi_  r   N)r!  r"  r#  r$  (MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGr&  r'  r   r(  r)  ry  ry  _  rw  r(  ry  zuniversal image segmentationc                       \ rS rSr\rSrg) AutoModelForInstanceSegmentationih  r   N)r!  r"  r#  r$  'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGr&  r'  r   r(  r)  r|  r|  h  rs  r(  r|  zinstance segmentationc                       \ rS rSr\rSrg)AutoModelForObjectDetectioniq  r   N)r!  r"  r#  r$  "MODEL_FOR_OBJECT_DETECTION_MAPPINGr&  r'  r   r(  r)  r  r  q      7Nr(  r  zobject detectionc                       \ rS rSr\rSrg)#AutoModelForZeroShotObjectDetectionix  r   N)r!  r"  r#  r$  ,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGr&  r'  r   r(  r)  r  r  x      ANr(  r  zzero-shot object detectionc                       \ rS rSr\rSrg)AutoModelForDepthEstimationi  r   N)r!  r"  r#  r$  "MODEL_FOR_DEPTH_ESTIMATION_MAPPINGr&  r'  r   r(  r)  r  r    r  r(  r  zdepth estimationc                       \ rS rSr\rSrg)AutoModelForVideoClassificationi  r   N)r!  r"  r#  r$  &MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  r  r    rL  r(  r  zvideo classificationc                       \ rS rSr\rSrg)AutoModelForVision2Seqi  r   N)r!  r"  r#  r$  MODEL_FOR_VISION_2_SEQ_MAPPINGr&  r'  r   r(  r)  r  r    s    3Nr(  r  zvision-to-text modelingc                       \ rS rSr\rSrg)AutoModelForImageTextToTexti  r   N)r!  r"  r#  r$  $MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPINGr&  r'  r   r(  r)  r  r    r.  r(  r  zimage-text-to-text modelingc                       \ rS rSr\rSrg)AutoModelForAudioClassificationi  r   N)r!  r"  r#  r$  &MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  r  r    rL  r(  r  zaudio classificationc                       \ rS rSr\rSrg)AutoModelForCTCi  r   N)r!  r"  r#  r$  MODEL_FOR_CTC_MAPPINGr&  r'  r   r(  r)  r  r    s    *Nr(  r  z%connectionist temporal classificationc                       \ rS rSr\rSrg)AutoModelForSpeechSeq2Seqi  r   N)r!  r"  r#  r$  "MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGr&  r'  r   r(  r)  r  r    r  r(  r  z,sequence-to-sequence speech-to-text modelingc                       \ rS rSr\rSrg)$AutoModelForAudioFrameClassificationi  r   N)r!  r"  r#  r$  ,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGr&  r'  r   r(  r)  r  r    r  r(  r  z"audio frame (token) classificationc                       \ rS rSr\rSrg)AutoModelForAudioXVectori  r   N)r!  r"  r#  r$  MODEL_FOR_AUDIO_XVECTOR_MAPPINGr&  r'  r   r(  r)  r  r    r2  r(  r  c                       \ rS rSr\rSrg)AutoModelForTextToSpectrogrami  r   N)r!  r"  r#  r$  %MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGr&  r'  r   r(  r)  r  r    s    :Nr(  r  c                       \ rS rSr\rSrg)AutoModelForTextToWaveformi  r   N)r!  r"  r#  r$  "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPINGr&  r'  r   r(  r)  r  r    r  r(  r  c                       \ rS rSr\rSrg)AutoBackbonei  r   N)r!  r"  r#  r$  MODEL_FOR_BACKBONE_MAPPINGr&  r'  r   r(  r)  r  r    rA  r(  r  zaudio retrieval via x-vectorc                       \ rS rSr\rSrg)AutoModelForMaskedImageModelingi  r   N)r!  r"  r#  r$  'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGr&  r'  r   r(  r)  r  r    rs  r(  r  zmasked image modelingc                   H   ^  \ rS rSr\U 4S j5       r\U 4S j5       rSrU =r$ )AutoModelWithLMHeadi  c                 X   > [         R                  " S[        5        [        TU ]  U5      $ 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  AutoModelWithLMHead.from_config  s,    B 		
 w"6**r(  c                 b   > [         R                  " S[        5        [        TU ]  " U/UQ70 UD6$ r  )r  r  r  r  from_pretrained)r  pretrained_model_name_or_path
model_argskwargsr  s       r)  r  #AutoModelWithLMHead.from_pretrained  s8    B 		
 w&'D\z\U[\\r(  r   )	r!  r"  r#  r$  classmethodr  r  r'  __classcell__)r  s   @r)  r  r    s+    + + ] ]r(  r  )Tr  r  r  r  'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGrD  r  r]  r  ri  MODEL_FOR_IMAGE_MAPPINGro  r5  r-  r}  r  rH  r%  rc  rf  r  r;  rS  rr  rK  rP  r  rV  r1  r  r  rv  r`  rz  r  r  MODEL_FOR_RETRIEVAL_MAPPINGr  rZ  r8  r@  rl  r  ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGr7  r  r  r  r  rC  r  r  rh  rn  r4  r|  r,  r  r0  r  rG  rb  re  r  r:  rR  rq  rJ  rO  r  rU  r  r  r_  ry  r  r  rY  r\  r  rk  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_NAMESr8  r;  r@  rD  r  ri  rl  ro  rr  r}  rz  r  r  r  r  rZ  r]  rH  r  r  r  r  r  rK  rP  rS  rV  r`  rc  rf  r  r  r  r  r  r  r  r  r%  r-  r1  r  r  rv  r5  r  r,  r0  r4  r7  r:  r?  rC  rG  rJ  rO  rR  rU  rY  r\  r_  rb  re  rh  rk  rn  rq  ru  ry  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  __all__r   r(  r)  <module>r     sX     #   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(  