
    sg$                       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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(      Z/ eg d)      Z0 ed*d+g      Z1 eg d,      Z2 eg d-      Z3 eg d.      Z4 ed/g      Z5 ed0g      Z6 eg d1      Z7 ed2d3g      Z8 ed4d5g      Z9 ed6g      Z: e	ee      Z; e	ee      Z< e	ee      Z= e	ee      Z> e	ee      Z? e	ee      Z@ e	ee3      ZA e	ee      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e0      Z\ e	ee1      Z] e	ee2      Z^ e	ee4      Z_ e	ee5      Z` e	ee6      Za e	ee7      Zb e	ee8      Zc e	ee9      Zd e	ee:      Ze G d7 d8e      Zf G d9 d:e      Zg G d; d<e      Zh G d= d>e      Zi G d? d@e      Zj e
ej      Zj G dA dBe      Zk e
ekdCD      Zk G dE dFe      Zl e
eldGD      Zl G dH dIe      Zm e
emdJD      Zm G dK dLe      Zn e
endMD      Zn G dN dOe      Zo e
eodPdQR      Zo G dS dTe      Zp e
epdUD      Zp G dV dWe      Zq e
eqdXD      Zq G dY dZe      Zr e
erd[d\R      Zr G d] d^e      Zs e
esd_d`R      Zs G da dbe      Zt e
etdcddR      Zt G de dfe      Zu e
eudgD      Zu G dh die      Zv e
evdjD      Zv G dk dle      Zw e
ewdmD      Zw G dn doe      Zx e
exdpD      Zx G dq dre      Zy e
eydsD      Zy G dt due      Zz e
ezdvD      Zz G dw dxe      Z{ e
e{dyD      Z{ G dz d{e      Z| e
e|d|D      Z| G d} d~e      Z} e
e}dD      Z} G d de      Z~ e
e~dD      Z~ G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d de      Z G d de      Z G d de      Z G d de      Z e
edD      Z G d de      Z e
edD      Z G d del      Zy)zAuto Model class.    N)OrderedDict   )logging   )_BaseAutoBackboneClass_BaseAutoModelClass_LazyAutoMappingauto_class_update)CONFIG_MAPPING_NAMES(  albertAlbertModelalign
AlignModelaltclipAltCLIPModel)audio-spectrogram-transformerASTModel)
autoformerAutoformerModel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conditional_detrConditionalDetrModel)convbertConvBertModelconvnextConvNextModel
convnextv2ConvNextV2Model)cpmantCpmAntModel)ctrl	CTRLModel)cvtCvtModel)dacDacModel)data2vec-audioData2VecAudioModeldata2vec-textData2VecTextModeldata2vec-visionData2VecVisionModel)dbrx	DbrxModeldebertaDebertaModel
deberta-v2DebertaV2Model)decision_transformerDecisionTransformerModeldeformable_detrDeformableDetrModeldeit	DeiTModeldeta	DetaModeldetr	DetrModeldinat
DinatModeldinov2Dinov2Model
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)gitGitModel)glmGlmModelglpn	GLPNModel)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)
graphormerGraphormerModel)grounding-dinoGroundingDinoModel)groupvitGroupViTModelhiera
HieraModel)hubertHubertModelibert
IBertModel)ideficsIdeficsModel)idefics2Idefics2Model)idefics3Idefics3Model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)llamarZ   
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)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)phimoePhimoeModel)pixtralPixtralVisionModel)plbartPLBartModel
poolformerPoolFormerModel)
prophetnetProphetNetModelpvtPvtModel)pvt_v2
PvtV2Model)qdqbertQDQBertModel)qwen2
Qwen2Model)qwen2_audio_encoderQwen2AudioEncoder)	qwen2_moeQwen2MoeModel)qwen2_vlQwen2VLModel)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)rwkv	RwkvModelsamSamModel)seamless_m4tSeamlessM4TModel)seamless_m4t_v2SeamlessM4Tv2Model	segformerSegformerModel)seggptSegGptModel)sewSEWModel)sew-d	SEWDModelsiglipSiglipModelsiglip_vision_modelSiglipVisionModel)speech_to_textSpeech2TextModel)speecht5SpeechT5Model)splinterSplinterModelsqueezebertSqueezeBertModel)stablelmStableLmModel)
starcoder2Starcoder2ModelswiftformerSwiftFormerModelswin	SwinModelswin2srSwin2SRModelswinv2Swinv2Model)switch_transformersSwitchTransformersModel)t5T5Modeltable-transformerTableTransformerModel)tapas
TapasModel)time_series_transformerTimeSeriesTransformerModeltimesformerTimesformerModeltimm_backboneTimmBackbone)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)M)r   AlbertForPreTrainingr   BartForConditionalGeneration)r"   BertForPreTraining)r'   BigBirdForPreTrainingr9   BloomForCausalLMr?   CamembertForMaskedLMrl   CTRLLMHeadModelru   Data2VecTextForMaskedLMr}   DebertaForMaskedLMr   DebertaV2ForMaskedLMr   DistilBertForMaskedLM)r   ElectraForPreTraining)r   ErnieForPreTrainingr   FalconMambaForCausalLMr   FlaubertWithLMHeadModel)r   FlavaForPreTraining)r   FNetForPreTrainingr   FSMTForConditionalGeneration)r   FunnelForPreTraining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r  LukeForMaskedLM)r  LxmertForPreTrainingr!  MambaForCausalLMr#  Mamba2ForCausalLMr3  MegaForMaskedLM)r5  MegatronBertForPreTrainingmllamaMllamaForConditionalGeneration)r?  MobileBertForPreTrainingrO  MPNetForMaskedLMrQ  MptForCausalLMrS  MraForMaskedLMr[  MvpForConditionalGeneration)rb  NezhaForPreTrainingrd  NllbMoeForConditionalGenerationru  OpenAIGPTLMHeadModel	paligemma!PaliGemmaForConditionalGenerationqwen2_audio"Qwen2AudioForConditionalGenerationr  r  RobertaForMaskedLMr  RobertaPreLayerNormForMaskedLM)r  RoCBertForPreTrainingr  RwkvForCausalLM)r  SplinterForPreTrainingr  SqueezeBertForMaskedLMr  *SwitchTransformersForConditionalGenerationr  T5ForConditionalGenerationr  TapasForMaskedLMr
  TransfoXLLMHeadModel)r  TvltForPreTraining)r  UniSpeechForPreTraining)r  UniSpeechSatForPreTrainingvideo_llava"VideoLlavaForConditionalGeneration)r  VideoMAEForPreTrainingvipllava VipLlavaForConditionalGeneration)r#  VisualBertForPreTraining)r,  ViTMAEForPreTraining)r:  Wav2Vec2ForPreTraining)r>  Wav2Vec2ConformerForPreTrainingrI  XLMWithLMHeadModelrN  XLMRobertaForMaskedLMrQ  XLMRobertaXLForMaskedLMrS  XLNetLMHeadModelrU  XmodForMaskedLM)Pr   AlbertForMaskedLMr_  r"   BertForMaskedLMr'   BigBirdForMaskedLMr)   &BigBirdPegasusForConditionalGenerationr2   'BlenderbotSmallForConditionalGenerationrc  re  r[   CodeGenForCausalLMrb   ConvBertForMaskedLMrj   CpmAntForCausalLMrg  ri  rk  rm  ro  r   ElectraForMaskedLMzencoder-decoderEncoderDecoderModelr   ErnieForMaskedLMr   EsmForMaskedLMrs  ru  r   FNetForMaskedLMry  r   FunnelForMaskedLMr   GitForCausalLMr|  r~  r  r   GPTNeoForCausalLMr   GPTNeoXForCausalLMr   GPTNeoXJapaneseForCausalLMr   GPTJForCausalLMr   r  r  r  LEDForConditionalGenerationr  r  LongT5ForConditionalGenerationr  r  M2M100ForConditionalGenerationr  r  r%  MarianMTModelr  r5  MegatronBertForCausalLMr?  MobileBertForMaskedLMr  r  r  r  rb  NezhaForMaskedLMr  rg  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rB  WhisperForConditionalGenerationr  r  r  r  r  rZ  YosoForMaskedLM)[)r   BartForCausalLM)r"   BertLMHeadModel)r$   BertGenerationDecoder)r'   BigBirdForCausalLM)r)   BigBirdPegasusForCausalLM)r+   BioGptForCausalLM)r0   BlenderbotForCausalLM)r2   BlenderbotSmallForCausalLMrc  )r?   CamembertForCausalLM)rY   LlamaForCausalLMr  )r]   CohereForCausalLMr  rg  )ru   Data2VecTextForCausalLM)rz   DbrxForCausalLM)r   ElectraForCausalLM)r   ErnieForCausalLM)r   FalconForCausalLMrs  fuyuFuyuForCausalLM)r   GemmaForCausalLM)r   Gemma2ForCausalLMr  )r   GlmForCausalLMr|  r~  r  r  r  r  r	  )r   GraniteForCausalLM)r   GraniteMoeForCausalLM)r   JambaForCausalLM)r  JetMoeForCausalLM)r  r=  r  r  )r%  MarianForCausalLM)r/  MBartForCausalLM)r3  MegaForCausalLMr  )r:  MistralForCausalLM)r<  MixtralForCausalLM)r  MllamaForCausalLM)rM  MoshiForCausalLMr  )rW  MusicgenForCausalLM)rY  MusicgenMelodyForCausalLM)r[  MvpForCausalLM)r`  NemotronForCausalLM)ri  OlmoForCausalLM)rk  Olmo2ForCausalLM)rm  OlmoeForCausalLM)rs  OpenLlamaForCausalLMr  )rw  OPTForCausalLM)r  PegasusForCausalLM)r  PersimmonForCausalLM)r  PhiForCausalLM)r  Phi3ForCausalLM)r  PhimoeForCausalLM)r  PLBartForCausalLM)r  ProphetNetForCausalLM)r  QDQBertLMHeadModel)r  Qwen2ForCausalLM)r  Qwen2MoeForCausalLM)r  RecurrentGemmaForCausalLMr$  )r  RemBertForCausalLM)r  RobertaForCausalLM)r  RobertaPreLayerNormForCausalLM)r  RoCBertForCausalLM)r  RoFormerForCausalLMr  )speech_to_text_2Speech2Text2ForCausalLM)r  StableLmForCausalLM)r  Starcoder2ForCausalLMr  )trocrTrOCRForCausalLM)rB  WhisperForCausalLM)rF  XGLMForCausalLMr  )rK  XLMProphetNetForCausalLM)rN  XLMRobertaForCausalLM)rQ  XLMRobertaXLForCausalLMr  )rU  XmodForCausalLM)r\  ZambaForCausalLM)1r   r-   r_   rd   rg   rw   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  )r  MllamaVisionModelrA  rD  rG  rJ  r]  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r%  r(  r+  r.  r1  r7  rW  ))r   DeiTForMaskedImageModeling)r   FocalNetForMaskedImageModeling)r  SwinForMaskedImageModeling)r  Swinv2ForMaskedImageModeling)r&  ViTForMaskedImageModeling)r   ImageGPTForCausalImageModeling)%)r   BeitForImageClassification)r.   BitForImageClassification)rM   CLIPForImageClassification)re   ConvNextForImageClassification)rh    ConvNextV2ForImageClassification)rn   CvtForImageClassification)rx   $Data2VecVisionForImageClassification)r   )DeiTForImageClassification%DeiTForImageClassificationWithTeacher)r   DinatForImageClassification)r   Dinov2ForImageClassification)r   )%EfficientFormerForImageClassification0EfficientFormerForImageClassificationWithTeacher)r   "EfficientNetForImageClassification)r   FocalNetForImageClassification)r   HieraForImageClassification)r   IJepaForImageClassification)r   ImageGPTForImageClassification)r  )LevitForImageClassification&LevitForImageClassificationWithTeacher)rB  !MobileNetV1ForImageClassification)rE  !MobileNetV2ForImageClassification)rH  MobileViTForImageClassification)rK  !MobileViTV2ForImageClassification)r^  NatForImageClassification)r  )&PerceiverForImageClassificationLearned&PerceiverForImageClassificationFourier-PerceiverForImageClassificationConvProcessing)r   PoolFormerForImageClassification)r  PvtForImageClassification)r  PvtV2ForImageClassification)r  RegNetForImageClassification)r  ResNetForImageClassification)r  SegformerForImageClassification)r  SiglipForImageClassification)r  !SwiftFormerForImageClassification)r  SwinForImageClassification)r  Swinv2ForImageClassification)r  VanForImageClassification)r&  ViTForImageClassification)r)  ViTHybridForImageClassification)r/  ViTMSNForImageClassificationr   DetrForSegmentation))r   BeitForSemanticSegmentation)rx   %Data2VecVisionForSemanticSegmentation)r   DPTForSemanticSegmentation)rE  "MobileNetV2ForSemanticSegmentation)rH   MobileViTForSemanticSegmentation)rK  "MobileViTV2ForSemanticSegmentation)r   SegformerForSemanticSegmentation)upernetUperNetForSemanticSegmentationr+  !MaskFormerForInstanceSegmentation)r  )r)  #Mask2FormerForUniversalSegmentationr  )rq  !OneFormerForUniversalSegmentation))r  !TimesformerForVideoClassification)r  VideoMAEForVideoClassification)r8  VivitForVideoClassification)r5   BlipForConditionalGenerationr7   Blip2ForConditionalGenerationrC   !ChameleonForConditionalGenerationr  r  r  instructblip$InstructBlipForConditionalGeneration)instructblipvideo)InstructBlipVideoForConditionalGenerationr  Kosmos2ForConditionalGenerationr  r  r  r  r  r  
pix2struct"Pix2StructForConditionalGenerationr  Qwen2VLForConditionalGenerationr  r  zvision-encoder-decoderVisionEncoderDecoderModel)r  r  r  rD  r  r  r  r  r  r  r  r  r  r  r  r  )r  r  r  )r  UdopForConditionalGenerationr  r  )-r  r_  r  r  re  r  ri  rk  rm  ro  r  r  r  ru  r  r  r  r  r  r  r/  MBartForConditionalGenerationr  )r5  MegatronBertForMaskedLMr  r  r  r  r  r  )r  PerceiverForMaskedLMr"  )r  ReformerForMaskedLMr&  r  r  r(  r*  r  r  r.  r  r  r  r  r2  ))r`   !ConditionalDetrForObjectDetection)r    DeformableDetrForObjectDetection)r   DetaForObjectDetection)r   DetrForObjectDetection)r  RTDetrForObjectDetection)r  "TableTransformerForObjectDetection)rX  YolosForObjectDetection))r   GroundingDinoForObjectDetectionro  )ry  Owlv2ForObjectDetection)r{  OwlViTForObjectDetection))depth_anythingDepthAnythingForDepthEstimation)r   DPTForDepthEstimation)r   GLPNForDepthEstimation)zoedepthZoeDepthForDepthEstimation)r_  r  )r0   "BlenderbotForConditionalGenerationr  r  ry  r   r  r  r  r  r  )rU  MT5ForConditionalGenerationr  r  )r  PegasusForConditionalGenerationr  r  )r  "ProphetNetForConditionalGenerationr  )r  SeamlessM4TForTextToText)r  SeamlessM4Tv2ForTextToTextr  r  )r  UMT5ForConditionalGeneration)rK  %XLMProphetNetForConditionalGeneration)r  )r  SeamlessM4TForSpeechToText)r  SeamlessM4Tv2ForSpeechToText)zspeech-encoder-decoderSpeechEncoderDecoderModelr,  )r  SpeechT5ForSpeechToTextr0  )[)r   AlbertForSequenceClassification)r   BartForSequenceClassification)r"   BertForSequenceClassification)r'    BigBirdForSequenceClassification)r)   'BigBirdPegasusForSequenceClassification)r+   BioGptForSequenceClassification)r9   BloomForSequenceClassification)r?   "CamembertForSequenceClassification)rA   CanineForSequenceClassification)rY   LlamaForSequenceClassification)rb   !ConvBertForSequenceClassification)rl   CTRLForSequenceClassification)ru   %Data2VecTextForSequenceClassification)r}    DebertaForSequenceClassification)r   "DebertaV2ForSequenceClassification)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   GPT2ForSequenceClassification)r   r  )r   #GPTBigCodeForSequenceClassification)r   GPTNeoForSequenceClassification)r    GPTNeoXForSequenceClassification)r   GPTJForSequenceClassification)r   IBertForSequenceClassification)r   JambaForSequenceClassification)r  JetMoeForSequenceClassification)r  !LayoutLMForSequenceClassification)r
  #LayoutLMv2ForSequenceClassification)r  #LayoutLMv3ForSequenceClassification)r  LEDForSequenceClassification)r  LiltForSequenceClassification)r  r  )r  #LongformerForSequenceClassification)r  LukeForSequenceClassification)r'  !MarkupLMForSequenceClassification)r/  MBartForSequenceClassification)r3  MegaForSequenceClassification)r5  %MegatronBertForSequenceClassification)r:   MistralForSequenceClassification)r<   MixtralForSequenceClassification)r?  #MobileBertForSequenceClassification)rO  MPNetForSequenceClassification)rQ  MptForSequenceClassification)rS  MraForSequenceClassification)rU  MT5ForSequenceClassification)r[  MvpForSequenceClassification)r`  !NemotronForSequenceClassification)rb  NezhaForSequenceClassification)rg  &NystromformerForSequenceClassification)rs  "OpenLlamaForSequenceClassification)ru  "OpenAIGPTForSequenceClassification)rw  OPTForSequenceClassification)r  "PerceiverForSequenceClassification)r  "PersimmonForSequenceClassification)r  PhiForSequenceClassification)r  Phi3ForSequenceClassification)r  PhimoeForSequenceClassification)r  PLBartForSequenceClassification)r   QDQBertForSequenceClassification)r  Qwen2ForSequenceClassification)r  !Qwen2MoeForSequenceClassification)r  !ReformerForSequenceClassification)r   RemBertForSequenceClassification)r   RobertaForSequenceClassification)r  ,RobertaPreLayerNormForSequenceClassification)r   RoCBertForSequenceClassification)r  !RoFormerForSequenceClassification)r  $SqueezeBertForSequenceClassification)r  !StableLmForSequenceClassification)r  #Starcoder2ForSequenceClassification)r  T5ForSequenceClassification)r  TapasForSequenceClassification)r
  "TransfoXLForSequenceClassification)r  UMT5ForSequenceClassification)rI  XLMForSequenceClassification)rN  #XLMRobertaForSequenceClassification)rQ  %XLMRobertaXLForSequenceClassification)rS  XLNetForSequenceClassification)rU  XmodForSequenceClassification)rZ  YosoForSequenceClassification)r\  ZambaForSequenceClassification)D)r   AlbertForQuestionAnswering)r   BartForQuestionAnswering)r"   BertForQuestionAnswering)r'   BigBirdForQuestionAnswering)r)   "BigBirdPegasusForQuestionAnswering)r9   BloomForQuestionAnswering)r?   CamembertForQuestionAnswering)rA   CanineForQuestionAnswering)rb   ConvBertForQuestionAnswering)ru    Data2VecTextForQuestionAnswering)r}   DebertaForQuestionAnswering)r   DebertaV2ForQuestionAnswering)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)r  LEDForQuestionAnswering)r  LiltForQuestionAnswering)r  LlamaForQuestionAnswering)r  LongformerForQuestionAnswering)r  LukeForQuestionAnswering)r  LxmertForQuestionAnswering)r'  MarkupLMForQuestionAnswering)r/  MBartForQuestionAnswering)r3  MegaForQuestionAnswering)r5   MegatronBertForQuestionAnswering)r:  MistralForQuestionAnswering)r<  MixtralForQuestionAnswering)r?  MobileBertForQuestionAnswering)rO  MPNetForQuestionAnswering)rQ  MptForQuestionAnswering)rS  MraForQuestionAnswering)rU  MT5ForQuestionAnswering)r[  MvpForQuestionAnswering)r`  NemotronForQuestionAnswering)rb  NezhaForQuestionAnswering)rg  !NystromformerForQuestionAnswering)rw  OPTForQuestionAnswering)r  QDQBertForQuestionAnswering)r  Qwen2ForQuestionAnswering)r  Qwen2MoeForQuestionAnswering)r  ReformerForQuestionAnswering)r  RemBertForQuestionAnswering)r  RobertaForQuestionAnswering)r  'RobertaPreLayerNormForQuestionAnswering)r  RoCBertForQuestionAnswering)r  RoFormerForQuestionAnswering)r  SplinterForQuestionAnswering)r  SqueezeBertForQuestionAnswering)r  T5ForQuestionAnswering)r  UMT5ForQuestionAnswering)rI  XLMForQuestionAnsweringSimple)rN  XLMRobertaForQuestionAnswering)rQ   XLMRobertaXLForQuestionAnswering)rS  XLNetForQuestionAnsweringSimple)rU  XmodForQuestionAnswering)rZ  YosoForQuestionAnswering)r  TapasForQuestionAnswering))r5   BlipForQuestionAnsweringr  )r   ViltForQuestionAnswering))r  LayoutLMForQuestionAnsweringrf  rh  )H)r   AlbertForTokenClassification)r"   BertForTokenClassification)r'   BigBirdForTokenClassification)r+   BioGptForTokenClassification)r9   BloomForTokenClassification)r=   BrosForTokenClassification)r?   CamembertForTokenClassification)rA   CanineForTokenClassification)rb   ConvBertForTokenClassification)ru   "Data2VecTextForTokenClassification)r}   DebertaForTokenClassification)r   DebertaV2ForTokenClassification)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   GPT2ForTokenClassification)r   r  )r    GPTBigCodeForTokenClassification)r   GPTNeoForTokenClassification)r   GPTNeoXForTokenClassification)r   IBertForTokenClassification)r  LayoutLMForTokenClassification)r
   LayoutLMv2ForTokenClassification)r   LayoutLMv3ForTokenClassification)r  LiltForTokenClassification)r  LlamaForTokenClassification)r   LongformerForTokenClassification)r  LukeForTokenClassification)r'  MarkupLMForTokenClassification)r3  MegaForTokenClassification)r5  "MegatronBertForTokenClassification)r:  MistralForTokenClassification)r<  MixtralForTokenClassification)r?   MobileBertForTokenClassification)rO  MPNetForTokenClassification)rQ  MptForTokenClassification)rS  MraForTokenClassification)rU  MT5ForTokenClassification)r`  NemotronForTokenClassification)rb  NezhaForTokenClassification)rg  #NystromformerForTokenClassification)r  PersimmonForTokenClassification)r  PhiForTokenClassification)r  Phi3ForTokenClassification)r  QDQBertForTokenClassification)r  Qwen2ForTokenClassification)r  Qwen2MoeForTokenClassification)r  RemBertForTokenClassification)r  RobertaForTokenClassification)r  )RobertaPreLayerNormForTokenClassification)r  RoCBertForTokenClassification)r  RoFormerForTokenClassification)r  !SqueezeBertForTokenClassification)r  StableLmForTokenClassification)r   Starcoder2ForTokenClassification)r  T5ForTokenClassification)r  UMT5ForTokenClassification)rI  XLMForTokenClassification)rN   XLMRobertaForTokenClassification)rQ  "XLMRobertaXLForTokenClassification)rS  XLNetForTokenClassification)rU  XmodForTokenClassification)rZ  YosoForTokenClassification)&)r   AlbertForMultipleChoice)r"   BertForMultipleChoice)r'   BigBirdForMultipleChoice)r?   CamembertForMultipleChoice)rA   CanineForMultipleChoice)rb   ConvBertForMultipleChoice)ru   Data2VecTextForMultipleChoice)r   DebertaV2ForMultipleChoice)r   DistilBertForMultipleChoice)r   ElectraForMultipleChoice)r   ErnieForMultipleChoice)r   ErnieMForMultipleChoice)r   FlaubertForMultipleChoice)r   FNetForMultipleChoice)r   FunnelForMultipleChoice)r   IBertForMultipleChoice)r  LongformerForMultipleChoice)r  LukeForMultipleChoice)r3  MegaForMultipleChoice)r5  MegatronBertForMultipleChoice)r?  MobileBertForMultipleChoice)rO  MPNetForMultipleChoice)rS  MraForMultipleChoice)rb  NezhaForMultipleChoice)rg  NystromformerForMultipleChoice)r  QDQBertForMultipleChoice)r  RemBertForMultipleChoice)r  RobertaForMultipleChoice)r  $RobertaPreLayerNormForMultipleChoice)r  RoCBertForMultipleChoice)r  RoFormerForMultipleChoice)r  SqueezeBertForMultipleChoice)rI  XLMForMultipleChoice)rN  XLMRobertaForMultipleChoice)rQ  XLMRobertaXLForMultipleChoice)rS  XLNetForMultipleChoice)rU  XmodForMultipleChoice)rZ  YosoForMultipleChoice))r"   BertForNextSentencePrediction)r   ErnieForNextSentencePrediction)r   FNetForNextSentencePrediction)r5  %MegatronBertForNextSentencePrediction)r?  #MobileBertForNextSentencePrediction)rb  NezhaForNextSentencePrediction)r   QDQBertForNextSentencePrediction))r   ASTForAudioClassification)rr   &Data2VecAudioForSequenceClassification)r   HubertForSequenceClassification)r  SEWForSequenceClassification)r  SEWDForSequenceClassification)r  "UniSpeechForSequenceClassification)r  %UniSpeechSatForSequenceClassification)r:  !Wav2Vec2ForSequenceClassification)r<  %Wav2Vec2BertForSequenceClassification)r>  *Wav2Vec2ConformerForSequenceClassification)r@  WavLMForSequenceClassification)rB  WhisperForAudioClassification))rr   Data2VecAudioForCTC)r   HubertForCTC)r1  MCTCTForCTC)r  	SEWForCTC)r  
SEWDForCTC)r  UniSpeechForCTC)r  UniSpeechSatForCTC)r:  Wav2Vec2ForCTC)r<  Wav2Vec2BertForCTC)r>  Wav2Vec2ConformerForCTC)r@  WavLMForCTC))rr   (Data2VecAudioForAudioFrameClassification)r  'UniSpeechSatForAudioFrameClassification)r:  #Wav2Vec2ForAudioFrameClassification)r<  'Wav2Vec2BertForAudioFrameClassification)r>  ,Wav2Vec2ConformerForAudioFrameClassification)r@   WavLMForAudioFrameClassification))rr   Data2VecAudioForXVector)r  UniSpeechSatForXVector)r:  Wav2Vec2ForXVector)r<  Wav2Vec2BertForXVector)r>  Wav2Vec2ConformerForXVector)r@  WavLMForXVectorr   )r  SpeechT5ForTextToSpeech)r   )r   FastSpeech2ConformerWithHifiGan)rW   MusicgenForConditionalGeneration)rY  &MusicgenMelodyForConditionalGeneration)r  SeamlessM4TForTextToSpeech)r  SeamlessM4Tv2ForTextToSpeechr4  )r   r   r4   )r7   Blip2ForImageTextRetrievalrE   rL   rT   r  ))r   BeitBackbone)r.   BitBackbone)re   ConvNextBackbone)rh   ConvNextV2Backbone)r   DinatBackbone)r   Dinov2Backbone)r   FocalNetBackbone)r   HieraBackbone)r-  MaskFormerSwinBackbone)r^  NatBackbone)r  PvtV2Backbone)r  ResNetBackbone)rt_detr_resnetRTDetrResNetBackbone)r  SwinBackbone)r  Swinv2Backboner  )r2  VitDetBackboner  )
superpointSuperPointForKeypointDetection)r   r!   r&   rO   rt   r|   r   r   r   r   r   r  )r  MllamaTextModelr>  )rU  MT5EncoderModelrf  r  r  r  r  r  r  r  )r  T5EncoderModel)r  UMT5EncoderModelrH  rM  rP  )r}  'PatchTSMixerForTimeSeriesClassification)r  PatchTSTForClassification)r}  PatchTSMixerForRegression)r  PatchTSTForRegression)r  Swin2SRForImageSuperResolutionc                       e Zd ZeZy)AutoModelForMaskGenerationN)__name__
__module____qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mapping     Y/var/www/html/venv/lib/python3.12/site-packages/transformers/models/auto/modeling_auto.pyrR  rR        6NrY  rR  c                       e Zd ZeZy)AutoModelForKeypointDetectionN)rS  rT  rU  $MODEL_FOR_KEYPOINT_DETECTION_MAPPINGrW  rX  rY  rZ  r]  r]        9NrY  r]  c                       e Zd ZeZy)AutoModelForTextEncodingN)rS  rT  rU  MODEL_FOR_TEXT_ENCODING_MAPPINGrW  rX  rY  rZ  ra  ra  
      4NrY  ra  c                       e Zd ZeZy)AutoModelForImageToImageN)rS  rT  rU   MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGrW  rX  rY  rZ  re  re    s    5NrY  re  c                       e Zd ZeZy)	AutoModelN)rS  rT  rU  MODEL_MAPPINGrW  rX  rY  rZ  rh  rh    s    "NrY  rh  c                       e Zd ZeZy)AutoModelForPreTrainingN)rS  rT  rU  MODEL_FOR_PRETRAINING_MAPPINGrW  rX  rY  rZ  rk  rk    s    2NrY  rk  pretraining)head_docc                       e Zd ZeZy)_AutoModelWithLMHeadN)rS  rT  rU  MODEL_WITH_LM_HEAD_MAPPINGrW  rX  rY  rZ  rp  rp  !      /NrY  rp  zlanguage modelingc                       e Zd ZeZy)AutoModelForCausalLMN)rS  rT  rU  MODEL_FOR_CAUSAL_LM_MAPPINGrW  rX  rY  rZ  rt  rt  (      0NrY  rt  zcausal language modelingc                       e Zd ZeZy)AutoModelForMaskedLMN)rS  rT  rU  MODEL_FOR_MASKED_LM_MAPPINGrW  rX  rY  rZ  rx  rx  /  rv  rY  rx  zmasked language modelingc                       e Zd ZeZy)AutoModelForSeq2SeqLMN)rS  rT  rU  &MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGrW  rX  rY  rZ  r{  r{  6      ;NrY  r{  z&sequence-to-sequence language modelingzgoogle-t5/t5-base)rn  checkpoint_for_examplec                       e Zd ZeZy)"AutoModelForSequenceClassificationN)rS  rT  rU  )MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r  A  s    >NrY  r  zsequence classificationc                       e Zd ZeZy)AutoModelForQuestionAnsweringN)rS  rT  rU  $MODEL_FOR_QUESTION_ANSWERING_MAPPINGrW  rX  rY  rZ  r  r  J  r_  rY  r  zquestion answeringc                       e Zd ZeZy)"AutoModelForTableQuestionAnsweringN)rS  rT  rU  *MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGrW  rX  rY  rZ  r  r  Q      ?NrY  r  ztable question answeringzgoogle/tapas-base-finetuned-wtqc                       e Zd ZeZy)#AutoModelForVisualQuestionAnsweringN)rS  rT  rU  +MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGrW  rX  rY  rZ  r  r  \  s    @NrY  r  zvisual question answeringzdandelin/vilt-b32-finetuned-vqac                       e Zd ZeZy)%AutoModelForDocumentQuestionAnsweringN)rS  rT  rU  -MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGrW  rX  rY  rZ  r  r  g  s    BNrY  r  zdocument question answeringz/impira/layoutlm-document-qa", revision="52e01b3c                       e Zd ZeZy)AutoModelForTokenClassificationN)rS  rT  rU  &MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r  r  r}  rY  r  ztoken classificationc                       e Zd ZeZy)AutoModelForMultipleChoiceN)rS  rT  rU  !MODEL_FOR_MULTIPLE_CHOICE_MAPPINGrW  rX  rY  rZ  r  r  y  r[  rY  r  zmultiple choicec                       e Zd ZeZy)"AutoModelForNextSentencePredictionN)rS  rT  rU  *MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  znext sentence predictionc                       e Zd ZeZy)AutoModelForImageClassificationN)rS  rT  rU  &MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r    r}  rY  r  zimage classificationc                       e Zd ZeZy)'AutoModelForZeroShotImageClassificationN)rS  rT  rU  0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r    s    ENrY  r  zzero-shot image classificationc                       e Zd ZeZy)AutoModelForImageSegmentationN)rS  rT  rU  $MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGrW  rX  rY  rZ  r  r    r_  rY  r  zimage segmentationc                       e Zd ZeZy) AutoModelForSemanticSegmentationN)rS  rT  rU  'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGrW  rX  rY  rZ  r  r        <NrY  r  zsemantic segmentationc                       e Zd ZeZy)!AutoModelForUniversalSegmentationN)rS  rT  rU  (MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGrW  rX  rY  rZ  r  r    s    =NrY  r  zuniversal image segmentationc                       e Zd ZeZy) AutoModelForInstanceSegmentationN)rS  rT  rU  'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  zinstance segmentationc                       e Zd ZeZy)AutoModelForObjectDetectionN)rS  rT  rU  "MODEL_FOR_OBJECT_DETECTION_MAPPINGrW  rX  rY  rZ  r  r        7NrY  r  zobject detectionc                       e Zd ZeZy)#AutoModelForZeroShotObjectDetectionN)rS  rT  rU  ,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGrW  rX  rY  rZ  r  r        ANrY  r  zzero-shot object detectionc                       e Zd ZeZy)AutoModelForDepthEstimationN)rS  rT  rU  "MODEL_FOR_DEPTH_ESTIMATION_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  zdepth estimationc                       e Zd ZeZy)AutoModelForVideoClassificationN)rS  rT  rU  &MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r    r}  rY  r  zvideo classificationc                       e Zd ZeZy)AutoModelForVision2SeqN)rS  rT  rU  MODEL_FOR_VISION_2_SEQ_MAPPINGrW  rX  rY  rZ  r  r    s    3NrY  r  zvision-to-text modelingc                       e Zd ZeZy)AutoModelForImageTextToTextN)rS  rT  rU  $MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPINGrW  rX  rY  rZ  r  r    r_  rY  r  zimage-text-to-text modelingc                       e Zd ZeZy)AutoModelForAudioClassificationN)rS  rT  rU  &MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r    r}  rY  r  zaudio classificationc                       e Zd ZeZy)AutoModelForCTCN)rS  rT  rU  MODEL_FOR_CTC_MAPPINGrW  rX  rY  rZ  r  r    s    *NrY  r  z%connectionist temporal classificationc                       e Zd ZeZy)AutoModelForSpeechSeq2SeqN)rS  rT  rU  "MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  z,sequence-to-sequence speech-to-text modelingc                       e Zd ZeZy)$AutoModelForAudioFrameClassificationN)rS  rT  rU  ,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  z"audio frame (token) classificationc                       e Zd ZeZy)AutoModelForAudioXVectorN)rS  rT  rU  MODEL_FOR_AUDIO_XVECTOR_MAPPINGrW  rX  rY  rZ  r  r    rc  rY  r  c                       e Zd ZeZy)AutoModelForTextToSpectrogramN)rS  rT  rU  %MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGrW  rX  rY  rZ  r  r    s    :NrY  r  c                       e Zd ZeZy)AutoModelForTextToWaveformN)rS  rT  rU  "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  c                       e Zd ZeZy)AutoBackboneN)rS  rT  rU  MODEL_FOR_BACKBONE_MAPPINGrW  rX  rY  rZ  r  r    rr  rY  r  zaudio retrieval via x-vectorc                       e Zd ZeZy)AutoModelForMaskedImageModelingN)rS  rT  rU  'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGrW  rX  rY  rZ  r  r    r  rY  r  zmasked image modelingc                   <     e Zd Ze fd       Ze fd       Z xZS )AutoModelWithLMHeadc                 V    t        j                  dt               t        |   |      S NzThe class `AutoModelWithLMHead` is deprecated and will be removed in a future version. Please use `AutoModelForCausalLM` for causal language models, `AutoModelForMaskedLM` for masked language models and `AutoModelForSeq2SeqLM` for encoder-decoder models.)warningswarnFutureWarningsuperfrom_config)clsconfig	__class__s     rZ  r  zAutoModelWithLMHead.from_config"  s,    B 		
 w"6**rY  c                 ^    t        j                  dt               t        |   |g|i |S r  )r  r  r  r  from_pretrained)r  pretrained_model_name_or_path
model_argskwargsr  s       rZ  r  z#AutoModelWithLMHead.from_pretrained,  s8    B 		
 w&'D\z\U[\\rY  )rS  rT  rU  classmethodr  r  __classcell__)r  s   @rZ  r  r  !  s+    + + ] ]rY  r  )__doc__r  collectionsr   utilsr   auto_factoryr   r   r	   r
   configuration_autor   
get_loggerrS  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_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_NAMESri  rl  rq  ru  'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGr  r  r  r  r  r  r  r  r  r  r  ry  MODEL_FOR_IMAGE_MAPPINGr  r  r  r  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rV  r^  rb  ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGrf  rR  r]  ra  re  rh  rk  rp  rt  rx  r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rX  rY  rZ  <module>r#     s
     #   5 
		H	%!CE N '2OQ' #f $/RT$  l %0]_% !B !,35! n 1<1 - 1< 	71 - 0;:<0 ,| .9 	(. * 1<
1 - 1< 	<1 - 2=2 . 0;0 , (3( $2 .9. *4 %0/1% !f ,7	, ( 6A6 2 ,7, ( 0;0 ,@ ,7
, ( 3>]_3 /B .9FH. *T 4? 	/4 0 5@5 1 7B7 3 0;JL0 ,\ +6(*+ 'X 4?
4 0 0;0 ,$ * " 6A
6 2 )4
) % /: 	?// + ,7	, ( :E
: 6 $/$  . +6+ ' .98. * )4) %B 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( $ /?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 ]. ]rY  