
    g#                    j   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          ZdS )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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)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   rx  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r0  MegaForMaskedLM)r2  MegatronBertForPreTrainingmllamaMllamaForConditionalGeneration)r<  MobileBertForPreTrainingrL  MPNetForMaskedLMrN  MptForCausalLMrP  MraForMaskedLMrX  MvpForConditionalGeneration)r_  NezhaForPreTrainingra  NllbMoeForConditionalGenerationrp  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)r5  Wav2Vec2ForPreTraining)r9  Wav2Vec2ConformerForPreTrainingrD  XLMWithLMHeadModelrI  XLMRobertaForMaskedLMrL  XLMRobertaXLForMaskedLMrN  XLNetLMHeadModelrP  XmodForMaskedLM)Pr   AlbertForMaskedLMrZ  r"   BertForMaskedLMr'   BigBirdForMaskedLMr)   &BigBirdPegasusForConditionalGenerationr2   'BlenderbotSmallForConditionalGenerationr^  r`  r[   CodeGenForCausalLMrb   ConvBertForMaskedLMrj   CpmAntForCausalLMrb  rd  rf  rh  rj  r   ElectraForMaskedLMzencoder-decoderEncoderDecoderModelr   ErnieForMaskedLMr   EsmForMaskedLMrn  rp  r   FNetForMaskedLMrt  r   FunnelForMaskedLMr   GitForCausalLMrw  ry  rz  r   GPTNeoForCausalLMr   GPTNeoXForCausalLMr   GPTNeoXJapaneseForCausalLMr   GPTJForCausalLMr   r}  r  r  LEDForConditionalGenerationr  r  LongT5ForConditionalGenerationr  r  M2M100ForConditionalGenerationr  r  r"  MarianMTModelr  r2  MegatronBertForCausalLMr<  MobileBertForMaskedLMr  r  r  r  r_  NezhaForMaskedLMr  rd  NystromformerForMaskedLMr  r~   PegasusXForConditionalGenerationr  PLBartForConditionalGeneration	pop2piano!Pop2PianoForConditionalGenerationr  QDQBertForMaskedLMr  ReformerModelWithLMHeadr  RemBertForMaskedLMr  r  r  RoCBertForMaskedLMr  RoFormerForMaskedLMr  r  #Speech2TextForConditionalGenerationr  r  r  r  r  r5  Wav2Vec2ForMaskedLMr=  WhisperForConditionalGenerationr  r  r  r  r  rU  YosoForMaskedLM)Z)r   BartForCausalLM)r"   BertLMHeadModel)r$   BertGenerationDecoder)r'   BigBirdForCausalLM)r)   BigBirdPegasusForCausalLM)r+   BioGptForCausalLM)r0   BlenderbotForCausalLM)r2   BlenderbotSmallForCausalLMr^  )r?   CamembertForCausalLM)rY   LlamaForCausalLMr  )r]   CohereForCausalLMr  rb  )ru   Data2VecTextForCausalLM)rz   DbrxForCausalLM)r   ElectraForCausalLM)r   ErnieForCausalLM)r   FalconForCausalLMrn  fuyuFuyuForCausalLM)r   GemmaForCausalLM)r   Gemma2ForCausalLMr  )r   GlmForCausalLMrw  ry  rz  r  r   r  r  )r   GraniteForCausalLM)r   GraniteMoeForCausalLM)r   JambaForCausalLM)r   JetMoeForCausalLM)r  r8  r  r  )r"  MarianForCausalLM)r,  MBartForCausalLM)r0  MegaForCausalLMr  )r7  MistralForCausalLM)r9  MixtralForCausalLM)r  MllamaForCausalLM)rJ  MoshiForCausalLMr  )rT  MusicgenForCausalLM)rV  MusicgenMelodyForCausalLM)rX  MvpForCausalLM)r]  NemotronForCausalLM)rf  OlmoForCausalLM)rh  OlmoeForCausalLM)rn  OpenLlamaForCausalLMr  )rr  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)r=  WhisperForCausalLM)rA  XGLMForCausalLMr  )rF  XLMProphetNetForCausalLM)rI  XLMRobertaForCausalLM)rL  XLMRobertaXLForCausalLMr  )rP  XmodForCausalLM)rW  ZambaForCausalLM)0r   r-   r_   rd   rg   rw   r   r   r   r   r   r   r   r   r   r   r   r   r   r  )r  MllamaVisionModelr>  rA  rD  rG  rZ  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r   r#  r&  r)  r,  r2  rR  ))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   ImageGPTForImageClassification)r  )LevitForImageClassification&LevitForImageClassificationWithTeacher)r?  !MobileNetV1ForImageClassification)rB  !MobileNetV2ForImageClassification)rE  MobileViTForImageClassification)rH  !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)rB  "MobileNetV2ForSemanticSegmentation)rE   MobileViTForSemanticSegmentation)rH  "MobileViTV2ForSemanticSegmentation)r   SegformerForSemanticSegmentation)upernetUperNetForSemanticSegmentationr(  !MaskFormerForInstanceSegmentation)r  )r&  #Mask2FormerForUniversalSegmentationr  )rl  !OneFormerForUniversalSegmentation))r  !TimesformerForVideoClassification)r  VideoMAEForVideoClassification)r3  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  r?  r  r  r  r  r  r  r  r  r  r  r  r  )r  r  r  )r  UdopForConditionalGenerationr  r  )-r  rZ  r  r  r`  r  rd  rf  rh  rj  r  r  r  rp  r  r  r}  r  r  r  r,  MBartForConditionalGenerationr  )r2  MegatronBertForMaskedLMr  r  r  r  r  r  )r  PerceiverForMaskedLMr  )r  ReformerForMaskedLMr!  r  r  r#  r%  r  r  r)  r  r  r  r  r-  ))r`   !ConditionalDetrForObjectDetection)r    DeformableDetrForObjectDetection)r   DetaForObjectDetection)r   DetrForObjectDetection)r  RTDetrForObjectDetection)r  "TableTransformerForObjectDetection)rS  YolosForObjectDetection))r   GroundingDinoForObjectDetectionrj  )rt  Owlv2ForObjectDetection)rv  OwlViTForObjectDetection))depth_anythingDepthAnythingForDepthEstimation)r   DPTForDepthEstimation)r   GLPNForDepthEstimation)zoedepthZoeDepthForDepthEstimation)rZ  r  )r0   "BlenderbotForConditionalGenerationr  r  rt  r   r  r  r
  r  r  )rR  MT5ForConditionalGenerationr  r  )r|  PegasusForConditionalGenerationr  r  )r  "ProphetNetForConditionalGenerationr  )r  SeamlessM4TForTextToText)r  SeamlessM4Tv2ForTextToTextr  r  )r  UMT5ForConditionalGeneration)rF  %XLMProphetNetForConditionalGeneration)r  )r  SeamlessM4TForSpeechToText)r  SeamlessM4Tv2ForSpeechToText)zspeech-encoder-decoderSpeechEncoderDecoderModelr'  )r  SpeechT5ForSpeechToTextr+  )[)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)r0  MegaForSequenceClassification)r2  %MegatronBertForSequenceClassification)r7   MistralForSequenceClassification)r9   MixtralForSequenceClassification)r<  #MobileBertForSequenceClassification)rL  MPNetForSequenceClassification)rN  MptForSequenceClassification)rP  MraForSequenceClassification)rR  MT5ForSequenceClassification)rX  MvpForSequenceClassification)r]  !NemotronForSequenceClassification)r_  NezhaForSequenceClassification)rd  &NystromformerForSequenceClassification)rn  "OpenLlamaForSequenceClassification)rp  "OpenAIGPTForSequenceClassification)rr  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)rD  XLMForSequenceClassification)rI  #XLMRobertaForSequenceClassification)rL  %XLMRobertaXLForSequenceClassification)rN  XLNetForSequenceClassification)rP  XmodForSequenceClassification)rU  YosoForSequenceClassification)rW  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)r0  MegaForQuestionAnswering)r2   MegatronBertForQuestionAnswering)r7  MistralForQuestionAnswering)r9  MixtralForQuestionAnswering)r<  MobileBertForQuestionAnswering)rL  MPNetForQuestionAnswering)rN  MptForQuestionAnswering)rP  MraForQuestionAnswering)rR  MT5ForQuestionAnswering)rX  MvpForQuestionAnswering)r]  NemotronForQuestionAnswering)r_  NezhaForQuestionAnswering)rd  !NystromformerForQuestionAnswering)rr  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)rD  XLMForQuestionAnsweringSimple)rI  XLMRobertaForQuestionAnswering)rL   XLMRobertaXLForQuestionAnswering)rN  XLNetForQuestionAnsweringSimple)rP  XmodForQuestionAnswering)rU  YosoForQuestionAnswering)r  TapasForQuestionAnswering))r5   BlipForQuestionAnsweringr  )r  ViltForQuestionAnswering))r  LayoutLMForQuestionAnsweringr_  ra  )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)r0  MegaForTokenClassification)r2  "MegatronBertForTokenClassification)r7  MistralForTokenClassification)r9  MixtralForTokenClassification)r<   MobileBertForTokenClassification)rL  MPNetForTokenClassification)rN  MptForTokenClassification)rP  MraForTokenClassification)rR  MT5ForTokenClassification)r]  NemotronForTokenClassification)r_  NezhaForTokenClassification)rd  #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)rD  XLMForTokenClassification)rI   XLMRobertaForTokenClassification)rL  "XLMRobertaXLForTokenClassification)rN  XLNetForTokenClassification)rP  XmodForTokenClassification)rU  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)r0  MegaForMultipleChoice)r2  MegatronBertForMultipleChoice)r<  MobileBertForMultipleChoice)rL  MPNetForMultipleChoice)rP  MraForMultipleChoice)r_  NezhaForMultipleChoice)rd  NystromformerForMultipleChoice)r  QDQBertForMultipleChoice)r  RemBertForMultipleChoice)r  RobertaForMultipleChoice)r  $RobertaPreLayerNormForMultipleChoice)r  RoCBertForMultipleChoice)r  RoFormerForMultipleChoice)r  SqueezeBertForMultipleChoice)rD  XLMForMultipleChoice)rI  XLMRobertaForMultipleChoice)rL  XLMRobertaXLForMultipleChoice)rN  XLNetForMultipleChoice)rP  XmodForMultipleChoice)rU  YosoForMultipleChoice))r"   BertForNextSentencePrediction)r   ErnieForNextSentencePrediction)r   FNetForNextSentencePrediction)r2  %MegatronBertForNextSentencePrediction)r<  #MobileBertForNextSentencePrediction)r_  NezhaForNextSentencePrediction)r   QDQBertForNextSentencePrediction))r   ASTForAudioClassification)rr   &Data2VecAudioForSequenceClassification)r   HubertForSequenceClassification)r  SEWForSequenceClassification)r  SEWDForSequenceClassification)r  "UniSpeechForSequenceClassification)r  %UniSpeechSatForSequenceClassification)r5  !Wav2Vec2ForSequenceClassification)r7  %Wav2Vec2BertForSequenceClassification)r9  *Wav2Vec2ConformerForSequenceClassification)r;  WavLMForSequenceClassification)r=  WhisperForAudioClassification))rr   Data2VecAudioForCTC)r   HubertForCTC)r.  MCTCTForCTC)r  	SEWForCTC)r  
SEWDForCTC)r  UniSpeechForCTC)r  UniSpeechSatForCTC)r5  Wav2Vec2ForCTC)r7  Wav2Vec2BertForCTC)r9  Wav2Vec2ConformerForCTC)r;  WavLMForCTC))rr   (Data2VecAudioForAudioFrameClassification)r  'UniSpeechSatForAudioFrameClassification)r5  #Wav2Vec2ForAudioFrameClassification)r7  'Wav2Vec2BertForAudioFrameClassification)r9  ,Wav2Vec2ConformerForAudioFrameClassification)r;   WavLMForAudioFrameClassification))rr   Data2VecAudioForXVector)r  UniSpeechSatForXVector)r5  Wav2Vec2ForXVector)r7  Wav2Vec2BertForXVector)r9  Wav2Vec2ConformerForXVector)r;  WavLMForXVectorr   )r  SpeechT5ForTextToSpeech)r   )r   FastSpeech2ConformerWithHifiGan)rT   MusicgenForConditionalGeneration)rV  &MusicgenMelodyForConditionalGeneration)r  SeamlessM4TForTextToSpeech)r  SeamlessM4Tv2ForTextToSpeechr/  )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   )r-  VitDetBackboner  )
superpointSuperPointForKeypointDetection)r   r!   r&   rO   rt   r|   r   r   r   r   r   r  )r  MllamaTextModelr;  )rR  MT5EncoderModelrc  r  r  r  r  r  r  r  )r  T5EncoderModel)r  UMT5EncoderModelrC  rH  rK  )rx  'PatchTSMixerForTimeSeriesClassification)rz  PatchTSTForClassification)rx  PatchTSMixerForRegression)rz  PatchTSTForRegression)r  Swin2SRForImageSuperResolutionc                       e Zd ZeZdS )AutoModelForMaskGenerationN)__name__
__module____qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mapping     b/var/www/html/ai-engine/env/lib/python3.11/site-packages/transformers/models/auto/modeling_auto.pyrK  rK            6NNNrR  rK  c                       e Zd ZeZdS )AutoModelForKeypointDetectionN)rL  rM  rN  $MODEL_FOR_KEYPOINT_DETECTION_MAPPINGrP  rQ  rR  rS  rV  rV            9NNNrR  rV  c                       e Zd ZeZdS )AutoModelForTextEncodingN)rL  rM  rN  MODEL_FOR_TEXT_ENCODING_MAPPINGrP  rQ  rR  rS  rZ  rZ            4NNNrR  rZ  c                       e Zd ZeZdS )AutoModelForImageToImageN)rL  rM  rN   MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGrP  rQ  rR  rS  r^  r^  	  s        5NNNrR  r^  c                       e Zd ZeZdS )	AutoModelN)rL  rM  rN  MODEL_MAPPINGrP  rQ  rR  rS  ra  ra    s        "NNNrR  ra  c                       e Zd ZeZdS )AutoModelForPreTrainingN)rL  rM  rN  MODEL_FOR_PRETRAINING_MAPPINGrP  rQ  rR  rS  rd  rd    s        2NNNrR  rd  pretraining)head_docc                       e Zd ZeZdS )_AutoModelWithLMHeadN)rL  rM  rN  MODEL_WITH_LM_HEAD_MAPPINGrP  rQ  rR  rS  ri  ri            /NNNrR  ri  zlanguage modelingc                       e Zd ZeZdS )AutoModelForCausalLMN)rL  rM  rN  MODEL_FOR_CAUSAL_LM_MAPPINGrP  rQ  rR  rS  rm  rm  #          0NNNrR  rm  zcausal language modelingc                       e Zd ZeZdS )AutoModelForMaskedLMN)rL  rM  rN  MODEL_FOR_MASKED_LM_MAPPINGrP  rQ  rR  rS  rq  rq  *  ro  rR  rq  zmasked language modelingc                       e Zd ZeZdS )AutoModelForSeq2SeqLMN)rL  rM  rN  &MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGrP  rQ  rR  rS  rt  rt  1          ;NNNrR  rt  z&sequence-to-sequence language modelingzgoogle-t5/t5-base)rg  checkpoint_for_examplec                       e Zd ZeZdS )"AutoModelForSequenceClassificationN)rL  rM  rN  )MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  ry  ry  <  s        >NNNrR  ry  zsequence classificationc                       e Zd ZeZdS )AutoModelForQuestionAnsweringN)rL  rM  rN  $MODEL_FOR_QUESTION_ANSWERING_MAPPINGrP  rQ  rR  rS  r|  r|  E  rX  rR  r|  zquestion answeringc                       e Zd ZeZdS )"AutoModelForTableQuestionAnsweringN)rL  rM  rN  *MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGrP  rQ  rR  rS  r  r  L          ?NNNrR  r  ztable question answeringzgoogle/tapas-base-finetuned-wtqc                       e Zd ZeZdS )#AutoModelForVisualQuestionAnsweringN)rL  rM  rN  +MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGrP  rQ  rR  rS  r  r  W  s        @NNNrR  r  zvisual question answeringzdandelin/vilt-b32-finetuned-vqac                       e Zd ZeZdS )%AutoModelForDocumentQuestionAnsweringN)rL  rM  rN  -MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGrP  rQ  rR  rS  r  r  b  s        BNNNrR  r  zdocument question answeringz/impira/layoutlm-document-qa", revision="52e01b3c                       e Zd ZeZdS )AutoModelForTokenClassificationN)rL  rM  rN  &MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r  m  rv  rR  r  ztoken classificationc                       e Zd ZeZdS )AutoModelForMultipleChoiceN)rL  rM  rN  !MODEL_FOR_MULTIPLE_CHOICE_MAPPINGrP  rQ  rR  rS  r  r  t  rT  rR  r  zmultiple choicec                       e Zd ZeZdS )"AutoModelForNextSentencePredictionN)rL  rM  rN  *MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGrP  rQ  rR  rS  r  r  {  r  rR  r  znext sentence predictionc                       e Zd ZeZdS )AutoModelForImageClassificationN)rL  rM  rN  &MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r    rv  rR  r  zimage classificationc                       e Zd ZeZdS )'AutoModelForZeroShotImageClassificationN)rL  rM  rN  0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r    s        ENNNrR  r  zzero-shot image classificationc                       e Zd ZeZdS )AutoModelForImageSegmentationN)rL  rM  rN  $MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGrP  rQ  rR  rS  r  r    rX  rR  r  zimage segmentationc                       e Zd ZeZdS ) AutoModelForSemanticSegmentationN)rL  rM  rN  'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGrP  rQ  rR  rS  r  r            <NNNrR  r  zsemantic segmentationc                       e Zd ZeZdS )!AutoModelForUniversalSegmentationN)rL  rM  rN  (MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGrP  rQ  rR  rS  r  r    s        =NNNrR  r  zuniversal image segmentationc                       e Zd ZeZdS ) AutoModelForInstanceSegmentationN)rL  rM  rN  'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGrP  rQ  rR  rS  r  r    r  rR  r  zinstance segmentationc                       e Zd ZeZdS )AutoModelForObjectDetectionN)rL  rM  rN  "MODEL_FOR_OBJECT_DETECTION_MAPPINGrP  rQ  rR  rS  r  r            7NNNrR  r  zobject detectionc                       e Zd ZeZdS )#AutoModelForZeroShotObjectDetectionN)rL  rM  rN  ,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGrP  rQ  rR  rS  r  r            ANNNrR  r  zzero-shot object detectionc                       e Zd ZeZdS )AutoModelForDepthEstimationN)rL  rM  rN  "MODEL_FOR_DEPTH_ESTIMATION_MAPPINGrP  rQ  rR  rS  r  r    r  rR  r  zdepth estimationc                       e Zd ZeZdS )AutoModelForVideoClassificationN)rL  rM  rN  &MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r    rv  rR  r  zvideo classificationc                       e Zd ZeZdS )AutoModelForVision2SeqN)rL  rM  rN  MODEL_FOR_VISION_2_SEQ_MAPPINGrP  rQ  rR  rS  r  r    s        3NNNrR  r  zvision-to-text modelingc                       e Zd ZeZdS )AutoModelForImageTextToTextN)rL  rM  rN  $MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPINGrP  rQ  rR  rS  r  r    rX  rR  r  zimage-text-to-text modelingc                       e Zd ZeZdS )AutoModelForAudioClassificationN)rL  rM  rN  &MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r    rv  rR  r  zaudio classificationc                       e Zd ZeZdS )AutoModelForCTCN)rL  rM  rN  MODEL_FOR_CTC_MAPPINGrP  rQ  rR  rS  r  r    s        *NNNrR  r  z%connectionist temporal classificationc                       e Zd ZeZdS )AutoModelForSpeechSeq2SeqN)rL  rM  rN  "MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGrP  rQ  rR  rS  r  r    r  rR  r  z,sequence-to-sequence speech-to-text modelingc                       e Zd ZeZdS )$AutoModelForAudioFrameClassificationN)rL  rM  rN  ,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGrP  rQ  rR  rS  r  r    r  rR  r  z"audio frame (token) classificationc                       e Zd ZeZdS )AutoModelForAudioXVectorN)rL  rM  rN  MODEL_FOR_AUDIO_XVECTOR_MAPPINGrP  rQ  rR  rS  r  r    r\  rR  r  c                       e Zd ZeZdS )AutoModelForTextToSpectrogramN)rL  rM  rN  %MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGrP  rQ  rR  rS  r  r    s        :NNNrR  r  c                       e Zd ZeZdS )AutoModelForTextToWaveformN)rL  rM  rN  "MODEL_FOR_TEXT_TO_WAVEFORM_MAPPINGrP  rQ  rR  rS  r  r  
  r  rR  r  c                       e Zd ZeZdS )AutoBackboneN)rL  rM  rN  MODEL_FOR_BACKBONE_MAPPINGrP  rQ  rR  rS  r  r    rk  rR  r  zaudio retrieval via x-vectorc                       e Zd ZeZdS )AutoModelForMaskedImageModelingN)rL  rM  rN  'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGrP  rQ  rR  rS  r  r    r  rR  r  zmasked image modelingc                   H     e Zd Ze fd            Ze fd            Z xZS )AutoModelWithLMHeadc                 z    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     rS  r  zAutoModelWithLMHead.from_config  s<    B 		
 	
 	
 ww""6***rR  c                 r    t          j        dt                      t                      j        |g|R i |S r  )r  r  r  r  from_pretrained)r  pretrained_model_name_or_path
model_argskwargsr  s       rS  r  z#AutoModelWithLMHead.from_pretrained'  sM    B 		
 	
 	
 'uww&'D\z\\\U[\\\rR  )rL  rM  rN  classmethodr  r  __classcell__)r  s   @rS  r  r    sp        + + + + [+ ] ] ] ] [] ] ] ] ]rR  r  )__doc__r  collectionsr   utilsr   auto_factoryr   r   r	   r
   configuration_autor   
get_loggerrL  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_NAMESrb  re  rj  rn  'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGr  r  r  r  r  r  r  r  r  r  r  rr  MODEL_FOR_IMAGE_MAPPINGr  r  r  r  ru  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  rO  rW  r[  ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGr_  rK  rV  rZ  r^  ra  rd  ri  rm  rq  rt  ry  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  rQ  rR  rS  <module>r     s3      # # # # # #                  5 4 4 4 4 4 
	H	%	%!kA A AC C J '2kO O OQ' Q' #f $/;R R RT$ T$  l %0K\ \ \^% ^% !@ !,2 2 24! 4! l 1<  1 1 - 1< 	71 1 - 0;{9 9 9;0 ;0 ,z .9[ 	(. . * 1<
 
 
1 1 - 1< 	<1 1 - 2=  2 2 . 0;{  0 0 , (3{  ( ( $2 .9[  . . *4 %0K/ / /1% 1% !f ,7;	 	 	, , ( 6A[  6 6 2 ,7;  , , ( 0;{  0 0 ,@ ,7;  
, 
, ( 3>+] ] ]_3 _3 /B .9[F F FH. H. *T 4?; 	/4 4 0 5@K  5 5 1 7Bk  7 7 3 0;{J J JL0 L0 ,\ +6+( ( (*+ *+ 'X 4?;  
4 
4 0 0;{  0 0 ,$ *k    " 6A[  
6 
6 2 )4  
) 
) % /:k 	?// / + ,7;	 	 	, , ( :E
 
 
: : 6 $/;  $ $  . +6++ + ' .9[8. . * )4  ) ) %B 6A[C16 6 2 2=5-2 2 . *55* * & ! !57JKK 0 01EGj k k --.BDdee ../CEfgg *:*:G+ + ' *:)9F* * & 4D3CP4 4 0 (8'7D( ( $ +;*:G+ + ' +;*:G+ + ' ,<+;H, , ( *:)9F* * & "2!12FHl!m!m '7'7D( ( $ /?.>K/ / + 1A0@M1 1 - /./CEfgg **+?A^__ *:*:G+ + ' &6%56JLt%u%u "/?/?L0 0 , &6%56JLt%u%u ")9)9F* * & -=,<I- - ) (8'7D( ( $ .>-=J. . * *:)9F* * & %5$45IKr$s$s !-=-=J. . * *:)9F* * & )()=?Z[[ %5%56JLt%u%u "/?/?L0 0 , #3"23GIn"o"o (8(8E) ) % &6%56JLt%u%u "--.BDdee $4$45IKr$s$s !'7'7D( ( $ #3"23GIn"o"o /?/?L0 0 , ,<+;H, , ( $4#34HJp#q#q  7 7 7 7 7!4 7 7 7: : : : :$7 : : :5 5 5 5 52 5 5 56 6 6 6 62 6 6 6# # # # ## # # # i((	3 3 3 3 31 3 3 3 ,+,Cm\\\ 0 0 0 0 0. 0 0 0 )()=H[\\\ 1 1 1 1 1. 1 1 1 )()=Hbccc 1 1 1 1 1. 1 1 1 )()=Hbccc < < < < </ < < < *)5.   ? ? ? ? ?)< ? ? ? &7%6&1J& & & "
: : : : :$7 : : : !2 12OZn o o o @ @ @ @ @)< @ @ @ &7%6&'<& & & "A A A A A*= A A A '8&7'(<' ' ' #C C C C C,? C C C ):(9)*L) ) ) %< < < < <&9 < < < #4"34S^t"u"u"u 7 7 7 7 7!4 7 7 7 /./ITefff @ @ @ @ @)< @ @ @ &7%6&1K& & & "
< < < < <&9 < < < #4"34S^t"u"u"u F F F F F.A F F F +<*;+6V+ + + '
: : : : :$7 : : : !2 12OZn o o o = = = = =': = = = $5#4$/F$ $ $  
> > > > >(; > > > %6$5%0N% % % !
= = = = =': = = = $5#4$/F$ $ $  
8 8 8 8 8"5 8 8 8 0/0KVhiii B B B B B*= B B B '8&7'2N' ' ' #
8 8 8 8 8"5 8 8 8 0/0KVhiii < < < < <&9 < < < #4"34S^t"u"u"u 4 4 4 4 40 4 4 4 +*+ALefff : : : : :"5 : : : 0/0KVsttt < < < < <&9 < < < #4"34S^t"u"u"u + + + + +) + + + $#O>efff8 8 8 8 8 3 8 8 8 .-(V   
B B B B B+> B B B (9'8(3W( ( ( $
5 5 5 5 52 5 5 5; ; ; ; ;$7 ; ; ;8 8 8 8 8!4 8 8 80 0 0 0 0) 0 0 0 -,-EPnooo = = = = =&9 = = = #4"34S^u"v"v"v ] ] ] ] ]. ] ] ] ] ]rR  