U 4Af@s|dZddlZddlmZddlmZddlmZmZm Z m Z ddl m Z e eZed d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddgZedddddddddddddddddd d d d d]d ddddddddddddddddddd d!d"d#d$dd%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=gHZed>dd?d@dAdBdddCdDdEddddddFdGdHdIddJddKdLd d d dMdNdOdPd]dddQddRddSdddTddUdVdddd dWd"dXd#dYdZd[d\d]d^d%d&d_d`d(dad*d+d,d-d.dbdcd9d:d;d<d=ddgOZededfdgdhdidjdkdlddmdndCdodEddpdqdrdsdtdudvdwdLd d d dMdNdOdPdxdydzddd{d|d}dUd~ddddddddd#dddddddddddd]dddddd(dddd.dddd9dddd<dgQZeddd+d-d.d4d9d:d;ddBdCdDdOdUdadfdpdddddddddddddddddddddddddddg/ZedddddgZedgZeddddddddddddddddddddddddddddddÐdĐdŐdƐdǐdȐdg$ZedgZedːd̐d͐dΐdϐdАdѐdgZedgZedʐdԐdӐdgZed֐dאdgZedِdڐdېdLddܐdݐdސdddd$dߐd2d4dgZed>dd?d@ddDdddddFdHdIddJdKddddddddVddd dWdXdd\dd^d%d&d_d`d*d-dbd9d:d;d=ddg-ZedddddddgZedddgZ eddddgZ!eddAddBdGdd]dQdRdSdTddd d"ddYdZdddd+d,ddgZ"ed[ddddaddcgZ#eddddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVgXZ$edWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddddddddg@Z%edgZ&eddڐdgZ'eddpdqgZ(edddddddddddddddddddddddddddddddddddddddddÐdĐdŐdƐdǐdȐdɐdʐdːd̐d͐dΐdϐdАdѐdҐdӐdԐdՐd֐dאdؐdِdڐdېdܐdݐdސdߐddgGZ)eddddddddddddddddddddddddddddddddddddddg&Z*edd d d d d dgZ+eddddddddddddg Z,edddddd d!d"d#d$d%g Z-ed&d'd(d)d*d+gZ.ed,d-d.d/d0d1gZ/edKd2gZ0edd3d4d5d6d7dgZ1ed d dd!d$d&dgZ2ed8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdӐdGgZ3edgZ4edHgZ5ed ddd3d6d7d?dEdLdcdsddIddddddddƐdJdKdddgZ6edLdMgZ7edNdOgZ8edPgZ9e e eZ:e e eZ;e e eZe e eZ?e e e2Z@e e eZAe e eZBe e eZCe e eZDe e eZEe e eZFe e e'ZGe e e(ZHe e eZIe e eZJe e eZKe e eZLe e e ZMe e e!ZNe e e"ZOe e e$ZPe e e%ZQe e e&ZRe e e)ZSe e e*ZTe e e+ZUe e e,ZVe e e-ZWe e e#ZXe e e.ZYe e e/ZZe e e0Z[e e e1Z\e e e3Z]e e e4Z^e e e5Z_e e e6Z`e e e7Zae e e8Zbe e e9ZcGdQdRdReZdGdSdTdTeZeGdUdVdVeZfGdWdXdXeZgGdYdZdZeZhe ehZhGd[d\d\eZie eid]d^ZiGd_d`d`eZje ejdad^ZjGdbdcdceZke ekddd^ZkGdedfdfeZle eldgd^ZlGdhdidieZme emdjdkdlZmGdmdndneZne endod^ZnGdpdqdqeZoe eodrd^ZoGdsdtdteZpe epdudvdlZpGdwdxdxeZqe eqdydzdlZqGd{d|d|eZre erd}d~dlZrGdddeZse esdd^ZsGdddeZte etdd^ZtGdddeZue eudd^ZuGdddeZve evdd^ZvGdddeZwe ewdd^ZwGdddeZxe exdd^ZxGdddeZye eydd^ZyGdddeZze ezdd^ZzGdddeZ{e e{dd^Z{GdddeZ|e e|dd^Z|GdddeZ}e e}dd^Z}GdddeZ~e e~dd^Z~GdddeZe edd^ZGdddeZe edd^ZGdddeZe edd^ZGdddeZe edd^ZGdddeZe edd^ZGdddeZe edd^ZGdddeZGdddeZGdddeZGdddeZe edd^ZGdddeZe edd^ZGdd„dejZdS(zAuto Model class.N) OrderedDict)logging)_BaseAutoBackboneClass_BaseAutoModelClass_LazyAutoMappingauto_class_update)CONFIG_MAPPING_NAMES)albertZ AlbertModel)ZalignZ AlignModel)ZaltclipZ AltCLIPModel)audio-spectrogram-transformerZASTModel)Z autoformerZAutoformerModel)ZbarkZ BarkModel)bartZ BartModel)beitZ BeitModel)bertZ BertModel)bert-generationZBertGenerationEncoder)big_birdZ BigBirdModel)bigbird_pegasusZBigBirdPegasusModel)biogptZ BioGptModel)bitZBitModel) blenderbotZBlenderbotModel)blenderbot-smallZBlenderbotSmallModel)blipZ BlipModel)blip-2Z Blip2Model)bloomZ BloomModel)Z bridgetowerZBridgeTowerModel)brosZ BrosModel) camembertZCamembertModel)canineZ CanineModel) chameleonZChameleonModel)Z chinese_clipZChineseCLIPModel)Zchinese_clip_vision_modelZChineseCLIPVisionModel)ZclapZ ClapModel)clipZ CLIPModel)Zclip_vision_modelZCLIPVisionModel)ZclipsegZ CLIPSegModel)ZclvpZ!ClvpModelForConditionalGeneration) code_llama LlamaModel)codegenZ CodeGenModel)cohereZ CohereModel)conditional_detrZConditionalDetrModel)convbertZ ConvBertModel)convnextZ ConvNextModel) convnextv2ZConvNextV2Model)cpmantZ CpmAntModel)ctrlZ CTRLModel)cvtZCvtModel)data2vec-audioZData2VecAudioModel) data2vec-textZData2VecTextModel)data2vec-visionZData2VecVisionModel)dbrxZ DbrxModel)debertaZ DebertaModel) deberta-v2ZDebertaV2Model)Zdecision_transformerZDecisionTransformerModel)deformable_detrZDeformableDetrModel)deitZ DeiTModel)detaZ DetaModel)detrZ DetrModel)dinatZ DinatModel)dinov2Z Dinov2Model) distilbertZDistilBertModel)z donut-swinZDonutSwinModel)ZdprZDPRQuestionEncoder)dptZDPTModel)efficientformerZEfficientFormerModel) efficientnetZEfficientNetModel)electraZ ElectraModel)ZencodecZ EncodecModel)ernieZ ErnieModel)ernie_mZ ErnieMModel)esmZEsmModel)falconZ FalconModel)fastspeech2_conformerZFastSpeech2ConformerModel)flaubertZ FlaubertModel)flavaZ FlavaModel)fnetZ FNetModel)focalnetZ FocalNetModel)fsmtZ FSMTModel)funnel)Z FunnelModelZFunnelBaseModel)gemmaZ GemmaModel)gemma2Z Gemma2Model)gitZGitModel)glpnZ GLPNModel)gpt-sw3 GPT2Model)gpt2rK) gpt_bigcodeZGPTBigCodeModel)gpt_neoZ GPTNeoModel)gpt_neoxZ GPTNeoXModel)gpt_neox_japaneseZGPTNeoXJapaneseModel)gptjZ GPTJModel)zgptsan-japaneseZ&GPTSanJapaneseForConditionalGeneration)Z graphormerZGraphormerModel)grounding-dinoZGroundingDinoModel)ZgroupvitZ GroupViTModel)hieraZ HieraModel)hubertZ HubertModel)ibertZ IBertModel)ideficsZ IdeficsModel)idefics2Z Idefics2Model)imagegptZ ImageGPTModel)ZinformerZ InformerModel)jambaZ JambaModel)jetmoeZ JetMoeModel)ZjukeboxZ JukeboxModel)kosmos-2Z Kosmos2Model)layoutlmZ LayoutLMModel) layoutlmv2ZLayoutLMv2Model) layoutlmv3ZLayoutLMv3Model)ledZLEDModel)levitZ LevitModel)liltZ LiltModel)llamar ) longformerZLongformerModel)longt5Z LongT5Model)lukeZ LukeModel)lxmertZ LxmertModel)m2m_100Z M2M100Model)mambaZ MambaModel)mamba2Z Mamba2Model)marianZ MarianModel)markuplmZ MarkupLMModel) mask2formerZMask2FormerModel) maskformerZMaskFormerModel)maskformer-swinZMaskFormerSwinModel)mbartZ MBartModel)mctctZ MCTCTModel)megaZ MegaModel) megatron-bertZMegatronBertModel)zmgp-strZMgpstrForSceneTextRecognition)mistralZ MistralModel)mixtralZ MixtralModel) mobilebertZMobileBertModel) mobilenet_v1ZMobileNetV1Model) mobilenet_v2ZMobileNetV2Model) mobilevitZMobileViTModel) mobilevitv2ZMobileViTV2Model)mpnetZ MPNetModel)mptZMptModel)mraZMraModel)mt5ZMT5Model)musicgenZ MusicgenModel)musicgen_melodyZMusicgenMelodyModel)mvpZMvpModel)natZNatModel)nemotronZ NemotronModel)nezhaZ NezhaModel)nllb-moeZ NllbMoeModel) nystromformerZNystromformerModel)olmoZ OlmoModel) oneformerZOneFormerModel) open-llamaZOpenLlamaModel) openai-gptZOpenAIGPTModel)optZOPTModel)owlv2Z Owlv2Model)owlvitZ OwlViTModel) patchtsmixerZPatchTSMixerModel)patchtstZ PatchTSTModel)pegasusZ PegasusModel) pegasus_xZ PegasusXModel) perceiverZPerceiverModel) persimmonZPersimmonModel)phiZPhiModel)phi3Z Phi3Model)plbartZ PLBartModel) poolformerZPoolFormerModel) prophetnetZProphetNetModel)pvtZPvtModel)pvt_v2Z PvtV2Model)qdqbertZ QDQBertModel)qwen2Z Qwen2Model) qwen2_moeZ Qwen2MoeModel)recurrent_gemmaZRecurrentGemmaModel)reformerZ ReformerModel)regnetZ RegNetModel)rembertZ RemBertModel)resnetZ ResNetModel)Z retribertZRetriBertModel)robertaZ RobertaModel)roberta-prelayernormZRobertaPreLayerNormModel)roc_bertZ RoCBertModel)roformerZ RoFormerModel)rt_detrZ RTDetrModel)rwkvZ RwkvModel)ZsamZSamModel) seamless_m4tZSeamlessM4TModel)seamless_m4t_v2ZSeamlessM4Tv2Model) segformerZSegformerModel)ZseggptZ SegGptModel)sewZSEWModel)sew-dZ SEWDModel)siglipZ SiglipModel)Zsiglip_vision_modelZSiglipVisionModel)speech_to_textZSpeech2TextModel)speecht5Z SpeechT5Model)splinterZ SplinterModel) squeezebertZSqueezeBertModel)stablelmZ StableLmModel) starcoder2ZStarcoder2Model) swiftformerZSwiftFormerModel)swinZ SwinModel)swin2srZ Swin2SRModel)swinv2Z Swinv2Model)switch_transformersZSwitchTransformersModel)t5ZT5Model)table-transformerZTableTransformerModel)tapasZ TapasModel)Ztime_series_transformerZTimeSeriesTransformerModel) timesformerZTimesformerModel)Z timm_backboneZ TimmBackbone)Ztrajectory_transformerZTrajectoryTransformerModel) transfo-xlZTransfoXLModel)tvltZ TvltModel)ZtvpZTvpModel)ZudopZ UdopModel)umt5Z UMT5Model) unispeechZUniSpeechModel) unispeech-satZUniSpeechSatModel)ZunivnetZ UnivNetModel)vanZVanModel)videomaeZ VideoMAEModel)viltZ ViltModel)zvision-text-dual-encoderZVisionTextDualEncoderModel) visual_bertZVisualBertModel)vitZViTModel) vit_hybridZViTHybridModel)vit_maeZ ViTMAEModel)vit_msnZ ViTMSNModel)vitdetZ VitDetModel)ZvitsZ VitsModel)vivitZ VivitModel)wav2vec2Z Wav2Vec2Model) wav2vec2-bertZWav2Vec2BertModel)wav2vec2-conformerZWav2Vec2ConformerModel)wavlmZ WavLMModel)whisperZ WhisperModel)ZxclipZ XCLIPModel)xglmZ XGLMModel)xlmZXLMModel)xlm-prophetnetZXLMProphetNetModel) xlm-robertaZXLMRobertaModel)xlm-roberta-xlZXLMRobertaXLModel)xlnetZ XLNetModel)xmodZ XmodModel)yolosZ YolosModel)yosoZ YosoModel)r ZAlbertForPreTraining)r ZBartForConditionalGeneration)rZBertForPreTraining)rZBigBirdForPreTraining)rZBloomForCausalLM)rZCamembertForMaskedLM)r(ZCTRLLMHeadModel)r+ZData2VecTextForMaskedLM)r.ZDebertaForMaskedLM)r/ZDebertaV2ForMaskedLM)r6ZDistilBertForMaskedLM)r:ZElectraForPreTraining)r;ZErnieForPreTraining)r@ZFlaubertWithLMHeadModel)rAZFlavaForPreTraining)rBZFNetForPreTraining)rDZFSMTForConditionalGeneration)rEZFunnelForPreTraining)rJGPT2LMHeadModel)rLr)rMZGPTBigCodeForCausalLM)rSZHieraForPreTraining)rUZIBertForMaskedLM)rVZIdeficsForVisionText2Text)rWZ Idefics2ForConditionalGeneration)r\ZLayoutLMForMaskedLM)ZllavaZLlavaForConditionalGeneration)zllava-next-videoZ&LlavaNextVideoForConditionalGeneration)Z llava_nextZ!LlavaNextForConditionalGeneration)rcZLongformerForMaskedLM)reZLukeForMaskedLM)rfZLxmertForPreTraining)rhZMambaForCausalLM)riZMamba2ForCausalLM)rqZMegaForMaskedLM)rrZMegatronBertForPreTraining)ruZMobileBertForPreTraining)rzZMPNetForMaskedLM)r{ZMptForCausalLM)r|ZMraForMaskedLM)rZMvpForConditionalGeneration)rZNezhaForPreTraining)rZNllbMoeForConditionalGeneration)rZOpenAIGPTLMHeadModel)Z paligemmaZ!PaliGemmaForConditionalGeneration)rZRobertaForMaskedLM)rZRobertaPreLayerNormForMaskedLM)rZRoCBertForPreTraining)rZRwkvForCausalLM)rZSplinterForPreTraining)rZSqueezeBertForMaskedLM)rZ*SwitchTransformersForConditionalGeneration)rZT5ForConditionalGeneration)rZTapasForMaskedLM)rZTransfoXLLMHeadModel)rZTvltForPreTraining)rZUniSpeechForPreTraining)rZUniSpeechSatForPreTraining)Z video_llavaZ"VideoLlavaForConditionalGeneration)rZVideoMAEForPreTraining)ZvipllavaZ VipLlavaForConditionalGeneration)rZVisualBertForPreTraining)rZViTMAEForPreTraining)rZWav2Vec2ForPreTraining)rZWav2Vec2ConformerForPreTraining)rZXLMWithLMHeadModel)rZXLMRobertaForMaskedLM)rZXLMRobertaXLForMaskedLM)rZXLNetLMHeadModel)rZXmodForMaskedLM)r ZAlbertForMaskedLM)rZBertForMaskedLM)rZBigBirdForMaskedLM)rZ&BigBirdPegasusForConditionalGeneration)rZ'BlenderbotSmallForConditionalGeneration)r!ZCodeGenForCausalLM)r$ZConvBertForMaskedLM)r'ZCpmAntForCausalLM)r:ZElectraForMaskedLM)zencoder-decoderZEncoderDecoderModel)r;ZErnieForMaskedLM)r=ZEsmForMaskedLM)rBZFNetForMaskedLM)rEZFunnelForMaskedLM)rHZGitForCausalLM)rNZGPTNeoForCausalLM)rOZGPTNeoXForCausalLM)rPZGPTNeoXJapaneseForCausalLM)rQZGPTJForCausalLM)r_ZLEDForConditionalGeneration)rdZLongT5ForConditionalGeneration)rgZM2M100ForConditionalGeneration)rjZ MarianMTModel)rrZMegatronBertForCausalLM)ruZMobileBertForMaskedLM)rZNezhaForMaskedLM)rZNystromformerForMaskedLM)rZ PegasusXForConditionalGeneration)rZPLBartForConditionalGeneration)Z pop2pianoZ!Pop2PianoForConditionalGeneration)rZQDQBertForMaskedLM)rZReformerModelWithLMHead)rZRemBertForMaskedLM)rZRoCBertForMaskedLM)rZRoFormerForMaskedLM)rZ#Speech2TextForConditionalGeneration)rZWav2Vec2ForMaskedLM)rZWhisperForConditionalGeneration)rZYosoForMaskedLM)r ZBartForCausalLM)rZBertLMHeadModel)rZBertGenerationDecoder)rZBigBirdForCausalLM)rZBigBirdPegasusForCausalLM)rZBioGptForCausalLM)rZBlenderbotForCausalLM)rZBlenderbotSmallForCausalLM)rZCamembertForCausalLM)rLlamaForCausalLM)r"ZCohereForCausalLM)r+ZData2VecTextForCausalLM)r-ZDbrxForCausalLM)r:ZElectraForCausalLM)r;ZErnieForCausalLM)r>ZFalconForCausalLM)ZfuyuZFuyuForCausalLM)rFZGemmaForCausalLM)rGZGemma2ForCausalLM)rYZJambaForCausalLM)rZZJetMoeForCausalLM)rbr)rjZMarianForCausalLM)roZMBartForCausalLM)rqZMegaForCausalLM)rsZMistralForCausalLM)rtZMixtralForCausalLM)r~ZMusicgenForCausalLM)rZMusicgenMelodyForCausalLM)rZMvpForCausalLM)rZNemotronForCausalLM)rZOlmoForCausalLM)rZOpenLlamaForCausalLM)rZOPTForCausalLM)rZPegasusForCausalLM)rZPersimmonForCausalLM)rZPhiForCausalLM)rZPhi3ForCausalLM)rZPLBartForCausalLM)rZProphetNetForCausalLM)rZQDQBertLMHeadModel)rZQwen2ForCausalLM)rZQwen2MoeForCausalLM)rZRecurrentGemmaForCausalLM)rZRemBertForCausalLM)rZRobertaForCausalLM)rZRobertaPreLayerNormForCausalLM)rZRoCBertForCausalLM)rZRoFormerForCausalLM)Zspeech_to_text_2ZSpeech2Text2ForCausalLM)rZStableLmForCausalLM)rZStarcoder2ForCausalLM)ZtrocrZTrOCRForCausalLM)rZWhisperForCausalLM)rZXGLMForCausalLM)rZXLMProphetNetForCausalLM)rZXLMRobertaForCausalLM)rZXLMRobertaXLForCausalLM)rZXmodForCausalLM)r1ZDeiTForMaskedImageModeling)rCZFocalNetForMaskedImageModeling)rZSwinForMaskedImageModeling)rZSwinv2ForMaskedImageModeling)rZViTForMaskedImageModeling)rXZImageGPTForCausalImageModeling)rZBeitForImageClassification)rZBitForImageClassification)rZCLIPForImageClassification)r%ZConvNextForImageClassification)r&Z ConvNextV2ForImageClassification)r)ZCvtForImageClassification)r,Z$Data2VecVisionForImageClassification)r1)ZDeiTForImageClassificationZ%DeiTForImageClassificationWithTeacher)r4ZDinatForImageClassification)r5ZDinov2ForImageClassification)r8)Z%EfficientFormerForImageClassificationZ0EfficientFormerForImageClassificationWithTeacher)r9Z"EfficientNetForImageClassification)rCZFocalNetForImageClassification)rSZHieraForImageClassification)rXZImageGPTForImageClassification)r`)ZLevitForImageClassificationZ&LevitForImageClassificationWithTeacher)rvZ!MobileNetV1ForImageClassification)rwZ!MobileNetV2ForImageClassification)rxZMobileViTForImageClassification)ryZ!MobileViTV2ForImageClassification)rZNatForImageClassification)r)Z&PerceiverForImageClassificationLearnedZ&PerceiverForImageClassificationFourierZ-PerceiverForImageClassificationConvProcessing)rZ PoolFormerForImageClassification)rZPvtForImageClassification)rZPvtV2ForImageClassification)rZRegNetForImageClassification)rZResNetForImageClassification)rZSegformerForImageClassification)rZSiglipForImageClassification)rZ!SwiftFormerForImageClassification)rZSwinForImageClassification)rZSwinv2ForImageClassification)rZVanForImageClassification)rZViTForImageClassification)rZViTHybridForImageClassification)rZViTMSNForImageClassification)r3ZDetrForSegmentation)rZBeitForSemanticSegmentation)r,Z%Data2VecVisionForSemanticSegmentation)r7ZDPTForSemanticSegmentation)rwZ"MobileNetV2ForSemanticSegmentation)rxZ MobileViTForSemanticSegmentation)ryZ"MobileViTV2ForSemanticSegmentation)rZ SegformerForSemanticSegmentation)ZupernetZUperNetForSemanticSegmentation)rmZ!MaskFormerForInstanceSegmentation)rlZ#Mask2FormerForUniversalSegmentation)rZ!OneFormerForUniversalSegmentation)rZ!TimesformerForVideoClassification)rZVideoMAEForVideoClassification)rZVivitForVideoClassification)rZBlipForConditionalGeneration)rZBlip2ForConditionalGeneration)rZ!ChameleonForConditionalGeneration)Z instructblipZ$InstructBlipForConditionalGeneration)ZinstructblipvideoZ)InstructBlipVideoForConditionalGeneration)r[ZKosmos2ForConditionalGeneration)Z pix2structZ"Pix2StructForConditionalGeneration)zvision-encoder-decoderZVisionEncoderDecoderModel)roZMBartForConditionalGeneration)rrZMegatronBertForMaskedLM)rZPerceiverForMaskedLM)rZReformerForMaskedLM)r#Z!ConditionalDetrForObjectDetection)r0Z DeformableDetrForObjectDetection)r2ZDetaForObjectDetection)r3ZDetrForObjectDetection)rZRTDetrForObjectDetection)rZ"TableTransformerForObjectDetection)rZYolosForObjectDetection)rRZGroundingDinoForObjectDetection)rZOwlv2ForObjectDetection)rZOwlViTForObjectDetection)Zdepth_anythingZDepthAnythingForDepthEstimation)r7ZDPTForDepthEstimation)rIZGLPNForDepthEstimation)ZzoedepthZZoeDepthForDepthEstimation)rZ"BlenderbotForConditionalGeneration)r}ZMT5ForConditionalGeneration)rZPegasusForConditionalGeneration)rZ"ProphetNetForConditionalGeneration)rZSeamlessM4TForTextToText)rZSeamlessM4Tv2ForTextToText)rZUMT5ForConditionalGeneration)rZ%XLMProphetNetForConditionalGeneration)rZSeamlessM4TForSpeechToText)rZSeamlessM4Tv2ForSpeechToText)zspeech-encoder-decoderZSpeechEncoderDecoderModel)rZSpeechT5ForSpeechToText)r ZAlbertForSequenceClassification)r ZBartForSequenceClassification)rZBertForSequenceClassification)rZ BigBirdForSequenceClassification)rZ'BigBirdPegasusForSequenceClassification)rZBioGptForSequenceClassification)rZBloomForSequenceClassification)rZ"CamembertForSequenceClassification)rZCanineForSequenceClassification)rLlamaForSequenceClassification)r$Z!ConvBertForSequenceClassification)r(ZCTRLForSequenceClassification)r+Z%Data2VecTextForSequenceClassification)r.Z DebertaForSequenceClassification)r/Z"DebertaV2ForSequenceClassification)r6Z#DistilBertForSequenceClassification)r:Z ElectraForSequenceClassification)r;ZErnieForSequenceClassification)r<ZErnieMForSequenceClassification)r=ZEsmForSequenceClassification)r>ZFalconForSequenceClassification)r@Z!FlaubertForSequenceClassification)rBZFNetForSequenceClassification)rEZFunnelForSequenceClassification)rFZGemmaForSequenceClassification)rGZGemma2ForSequenceClassification)rJGPT2ForSequenceClassification)rLr)rMZ#GPTBigCodeForSequenceClassification)rNZGPTNeoForSequenceClassification)rOZ GPTNeoXForSequenceClassification)rQZGPTJForSequenceClassification)rUZIBertForSequenceClassification)rYZJambaForSequenceClassification)rZZJetMoeForSequenceClassification)r\Z!LayoutLMForSequenceClassification)r]Z#LayoutLMv2ForSequenceClassification)r^Z#LayoutLMv3ForSequenceClassification)r_ZLEDForSequenceClassification)raZLiltForSequenceClassification)rbr)rcZ#LongformerForSequenceClassification)reZLukeForSequenceClassification)rkZ!MarkupLMForSequenceClassification)roZMBartForSequenceClassification)rqZMegaForSequenceClassification)rrZ%MegatronBertForSequenceClassification)rsZ MistralForSequenceClassification)rtZ MixtralForSequenceClassification)ruZ#MobileBertForSequenceClassification)rzZMPNetForSequenceClassification)r{ZMptForSequenceClassification)r|ZMraForSequenceClassification)r}ZMT5ForSequenceClassification)rZMvpForSequenceClassification)rZ!NemotronForSequenceClassification)rZNezhaForSequenceClassification)rZ&NystromformerForSequenceClassification)rZ"OpenLlamaForSequenceClassification)rZ"OpenAIGPTForSequenceClassification)rZOPTForSequenceClassification)rZ"PerceiverForSequenceClassification)rZ"PersimmonForSequenceClassification)rZPhiForSequenceClassification)rZPhi3ForSequenceClassification)rZPLBartForSequenceClassification)rZ QDQBertForSequenceClassification)rZQwen2ForSequenceClassification)rZ!Qwen2MoeForSequenceClassification)rZ!ReformerForSequenceClassification)rZ RemBertForSequenceClassification)rZ RobertaForSequenceClassification)rZ,RobertaPreLayerNormForSequenceClassification)rZ RoCBertForSequenceClassification)rZ!RoFormerForSequenceClassification)rZ$SqueezeBertForSequenceClassification)rZ!StableLmForSequenceClassification)rZ#Starcoder2ForSequenceClassification)rZT5ForSequenceClassification)rZTapasForSequenceClassification)rZ"TransfoXLForSequenceClassification)rZUMT5ForSequenceClassification)rZXLMForSequenceClassification)rZ#XLMRobertaForSequenceClassification)rZ%XLMRobertaXLForSequenceClassification)rZXLNetForSequenceClassification)rZXmodForSequenceClassification)rZYosoForSequenceClassification)r ZAlbertForQuestionAnswering)r ZBartForQuestionAnswering)rZBertForQuestionAnswering)rZBigBirdForQuestionAnswering)rZ"BigBirdPegasusForQuestionAnswering)rZBloomForQuestionAnswering)rZCamembertForQuestionAnswering)rZCanineForQuestionAnswering)r$ZConvBertForQuestionAnswering)r+Z Data2VecTextForQuestionAnswering)r.ZDebertaForQuestionAnswering)r/ZDebertaV2ForQuestionAnswering)r6ZDistilBertForQuestionAnswering)r:ZElectraForQuestionAnswering)r;ZErnieForQuestionAnswering)r<ZErnieMForQuestionAnswering)r>ZFalconForQuestionAnswering)r@Z"FlaubertForQuestionAnsweringSimple)rBZFNetForQuestionAnswering)rEZFunnelForQuestionAnswering)rLZGPT2ForQuestionAnswering)rNZGPTNeoForQuestionAnswering)rOZGPTNeoXForQuestionAnswering)rQZGPTJForQuestionAnswering)rUZIBertForQuestionAnswering)r]ZLayoutLMv2ForQuestionAnswering)r^ZLayoutLMv3ForQuestionAnswering)r_ZLEDForQuestionAnswering)raZLiltForQuestionAnswering)rbZLlamaForQuestionAnswering)rcZLongformerForQuestionAnswering)reZLukeForQuestionAnswering)rfZLxmertForQuestionAnswering)rkZMarkupLMForQuestionAnswering)roZMBartForQuestionAnswering)rqZMegaForQuestionAnswering)rrZ MegatronBertForQuestionAnswering)ruZMobileBertForQuestionAnswering)rzZMPNetForQuestionAnswering)r{ZMptForQuestionAnswering)r|ZMraForQuestionAnswering)r}ZMT5ForQuestionAnswering)rZMvpForQuestionAnswering)rZNemotronForQuestionAnswering)rZNezhaForQuestionAnswering)rZ!NystromformerForQuestionAnswering)rZOPTForQuestionAnswering)rZQDQBertForQuestionAnswering)rZReformerForQuestionAnswering)rZRemBertForQuestionAnswering)rZRobertaForQuestionAnswering)rZ'RobertaPreLayerNormForQuestionAnswering)rZRoCBertForQuestionAnswering)rZRoFormerForQuestionAnswering)rZSplinterForQuestionAnswering)rZSqueezeBertForQuestionAnswering)rZT5ForQuestionAnswering)rZUMT5ForQuestionAnswering)rZXLMForQuestionAnsweringSimple)rZXLMRobertaForQuestionAnswering)rZ XLMRobertaXLForQuestionAnswering)rZXLNetForQuestionAnsweringSimple)rZXmodForQuestionAnswering)rZYosoForQuestionAnswering)rZTapasForQuestionAnswering)rZBlipForQuestionAnswering)rZViltForQuestionAnswering)r\ZLayoutLMForQuestionAnswering)r ZAlbertForTokenClassification)rZBertForTokenClassification)rZBigBirdForTokenClassification)rZBioGptForTokenClassification)rZBloomForTokenClassification)rZBrosForTokenClassification)rZCamembertForTokenClassification)rZCanineForTokenClassification)r$ZConvBertForTokenClassification)r+Z"Data2VecTextForTokenClassification)r.ZDebertaForTokenClassification)r/ZDebertaV2ForTokenClassification)r6Z DistilBertForTokenClassification)r:ZElectraForTokenClassification)r;ZErnieForTokenClassification)r<ZErnieMForTokenClassification)r=ZEsmForTokenClassification)r>ZFalconForTokenClassification)r@ZFlaubertForTokenClassification)rBZFNetForTokenClassification)rEZFunnelForTokenClassification)rFZGemmaForTokenClassification)rGZGemma2ForTokenClassification)rJGPT2ForTokenClassification)rLr)rMZ GPTBigCodeForTokenClassification)rNZGPTNeoForTokenClassification)rOZGPTNeoXForTokenClassification)rUZIBertForTokenClassification)r\ZLayoutLMForTokenClassification)r]Z LayoutLMv2ForTokenClassification)r^Z LayoutLMv3ForTokenClassification)raZLiltForTokenClassification)rbZLlamaForTokenClassification)rcZ LongformerForTokenClassification)reZLukeForTokenClassification)rkZMarkupLMForTokenClassification)rqZMegaForTokenClassification)rrZ"MegatronBertForTokenClassification)rsZMistralForTokenClassification)rtZMixtralForTokenClassification)ruZ MobileBertForTokenClassification)rzZMPNetForTokenClassification)r{ZMptForTokenClassification)r|ZMraForTokenClassification)r}ZMT5ForTokenClassification)rZNemotronForTokenClassification)rZNezhaForTokenClassification)rZ#NystromformerForTokenClassification)rZPersimmonForTokenClassification)rZPhiForTokenClassification)rZPhi3ForTokenClassification)rZQDQBertForTokenClassification)rZQwen2ForTokenClassification)rZQwen2MoeForTokenClassification)rZRemBertForTokenClassification)rZRobertaForTokenClassification)rZ)RobertaPreLayerNormForTokenClassification)rZRoCBertForTokenClassification)rZRoFormerForTokenClassification)rZ!SqueezeBertForTokenClassification)rZStableLmForTokenClassification)rZ Starcoder2ForTokenClassification)rZT5ForTokenClassification)rZUMT5ForTokenClassification)rZXLMForTokenClassification)rZ XLMRobertaForTokenClassification)rZ"XLMRobertaXLForTokenClassification)rZXLNetForTokenClassification)rZXmodForTokenClassification)rZYosoForTokenClassification)r ZAlbertForMultipleChoice)rZBertForMultipleChoice)rZBigBirdForMultipleChoice)rZCamembertForMultipleChoice)rZCanineForMultipleChoice)r$ZConvBertForMultipleChoice)r+ZData2VecTextForMultipleChoice)r/ZDebertaV2ForMultipleChoice)r6ZDistilBertForMultipleChoice)r:ZElectraForMultipleChoice)r;ZErnieForMultipleChoice)r<ZErnieMForMultipleChoice)r@ZFlaubertForMultipleChoice)rBZFNetForMultipleChoice)rEZFunnelForMultipleChoice)rUZIBertForMultipleChoice)rcZLongformerForMultipleChoice)reZLukeForMultipleChoice)rqZMegaForMultipleChoice)rrZMegatronBertForMultipleChoice)ruZMobileBertForMultipleChoice)rzZMPNetForMultipleChoice)r|ZMraForMultipleChoice)rZNezhaForMultipleChoice)rZNystromformerForMultipleChoice)rZQDQBertForMultipleChoice)rZRemBertForMultipleChoice)rZRobertaForMultipleChoice)rZ$RobertaPreLayerNormForMultipleChoice)rZRoCBertForMultipleChoice)rZRoFormerForMultipleChoice)rZSqueezeBertForMultipleChoice)rZXLMForMultipleChoice)rZXLMRobertaForMultipleChoice)rZXLMRobertaXLForMultipleChoice)rZXLNetForMultipleChoice)rZXmodForMultipleChoice)rZYosoForMultipleChoice)rZBertForNextSentencePrediction)r;ZErnieForNextSentencePrediction)rBZFNetForNextSentencePrediction)rrZ%MegatronBertForNextSentencePrediction)ruZ#MobileBertForNextSentencePrediction)rZNezhaForNextSentencePrediction)rZ QDQBertForNextSentencePrediction)r ZASTForAudioClassification)r*Z&Data2VecAudioForSequenceClassification)rTZHubertForSequenceClassification)rZSEWForSequenceClassification)rZSEWDForSequenceClassification)rZ"UniSpeechForSequenceClassification)rZ%UniSpeechSatForSequenceClassification)rZ!Wav2Vec2ForSequenceClassification)rZ%Wav2Vec2BertForSequenceClassification)rZ*Wav2Vec2ConformerForSequenceClassification)rZWavLMForSequenceClassification)rZWhisperForAudioClassification)r*ZData2VecAudioForCTC)rTZ HubertForCTC)rpZ MCTCTForCTC)rZ SEWForCTC)rZ SEWDForCTC)rZUniSpeechForCTC)rZUniSpeechSatForCTC)rZWav2Vec2ForCTC)rZWav2Vec2BertForCTC)rZWav2Vec2ConformerForCTC)rZ WavLMForCTC)r*Z(Data2VecAudioForAudioFrameClassification)rZ'UniSpeechSatForAudioFrameClassification)rZ#Wav2Vec2ForAudioFrameClassification)rZ'Wav2Vec2BertForAudioFrameClassification)rZ,Wav2Vec2ConformerForAudioFrameClassification)rZ WavLMForAudioFrameClassification)r*ZData2VecAudioForXVector)rZUniSpeechSatForXVector)rZWav2Vec2ForXVector)rZWav2Vec2BertForXVector)rZWav2Vec2ConformerForXVector)rZWavLMForXVector)rZSpeechT5ForTextToSpeech)r?ZFastSpeech2ConformerWithHifiGan)r~Z MusicgenForConditionalGeneration)rZ&MusicgenMelodyForConditionalGeneration)rZSeamlessM4TForTextToSpeech)rZSeamlessM4Tv2ForTextToSpeech)rZ BeitBackbone)rZ BitBackbone)r%ZConvNextBackbone)r&ZConvNextV2Backbone)r4Z DinatBackbone)r5ZDinov2Backbone)rCZFocalNetBackbone)rSZ HieraBackbone)rnZMaskFormerSwinBackbone)rZ NatBackbone)rZ PvtV2Backbone)rZResNetBackbone)Zrt_detr_resnetZRTDetrResNetBackbone)rZ SwinBackbone)rZSwinv2Backbone)rZVitDetBackbone)Z superpointZSuperPointForKeypointDetection)r}ZMT5EncoderModel)rZT5EncoderModel)rZUMT5EncoderModel)rZ'PatchTSMixerForTimeSeriesClassification)rZPatchTSTForClassification)rZPatchTSMixerForRegression)rZPatchTSTForRegression)rZSwin2SRForImageSuperResolutionc@seZdZeZdS)AutoModelForMaskGenerationN)__name__ __module__ __qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mappingrrJ/tmp/pip-unpacked-wheel-zw5xktn0/transformers/models/auto/modeling_auto.pyrsrc@seZdZeZdS)AutoModelForKeypointDetectionN)rrr$MODEL_FOR_KEYPOINT_DETECTION_MAPPINGrrrrrrsrc@seZdZeZdS)AutoModelForTextEncodingN)rrrMODEL_FOR_TEXT_ENCODING_MAPPINGrrrrrrsrc@seZdZeZdS)AutoModelForImageToImageN)rrr MODEL_FOR_IMAGE_TO_IMAGE_MAPPINGrrrrrrsrc@seZdZeZdS) AutoModelN)rrr MODEL_MAPPINGrrrrrrsrc@seZdZeZdS)AutoModelForPreTrainingN)rrrMODEL_FOR_PRETRAINING_MAPPINGrrrrrrsrZ pretraining)head_docc@seZdZeZdS)_AutoModelWithLMHeadN)rrrMODEL_WITH_LM_HEAD_MAPPINGrrrrrrsrzlanguage modelingc@seZdZeZdS)AutoModelForCausalLMN)rrrMODEL_FOR_CAUSAL_LM_MAPPINGrrrrrrsrzcausal language modelingc@seZdZeZdS)AutoModelForMaskedLMN)rrrMODEL_FOR_MASKED_LM_MAPPINGrrrrrrsrzmasked language modelingc@seZdZeZdS)AutoModelForSeq2SeqLMN)rrr&MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGrrrrrrsrz&sequence-to-sequence language modelingzgoogle-t5/t5-base)rZcheckpoint_for_examplec@seZdZeZdS)"AutoModelForSequenceClassificationN)rrr)MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGrrrrrrsrzsequence classificationc@seZdZeZdS)AutoModelForQuestionAnsweringN)rrr$MODEL_FOR_QUESTION_ANSWERING_MAPPINGrrrrrrsrzquestion answeringc@seZdZeZdS)"AutoModelForTableQuestionAnsweringN)rrr*MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGrrrrrrsrztable question answeringzgoogle/tapas-base-finetuned-wtqc@seZdZeZdS)#AutoModelForVisualQuestionAnsweringN)rrr+MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGrrrrrr srzvisual question answeringzdandelin/vilt-b32-finetuned-vqac@seZdZeZdS)%AutoModelForDocumentQuestionAnsweringN)rrr-MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGrrrrrrsrzdocument question answeringz/impira/layoutlm-document-qa", revision="52e01b3c@seZdZeZdS)AutoModelForTokenClassificationN)rrr&MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGrrrrrrsrztoken classificationc@seZdZeZdS)AutoModelForMultipleChoiceN)rrr!MODEL_FOR_MULTIPLE_CHOICE_MAPPINGrrrrrr&srzmultiple choicec@seZdZeZdS)"AutoModelForNextSentencePredictionN)rrr*MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGrrrrrr-srznext sentence predictionc@seZdZeZdS)AutoModelForImageClassificationN)rrr&MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGrrrrrr 6sr zimage classificationc@seZdZeZdS)'AutoModelForZeroShotImageClassificationN)rrr0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGrrrrrr =sr zzero-shot image classificationc@seZdZeZdS)AutoModelForImageSegmentationN)rrr$MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGrrrrrrFsrzimage segmentationc@seZdZeZdS) AutoModelForSemanticSegmentationN)rrr'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGrrrrrrMsrzsemantic segmentationc@seZdZeZdS)!AutoModelForUniversalSegmentationN)rrr(MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGrrrrrrVsrzuniversal image segmentationc@seZdZeZdS) AutoModelForInstanceSegmentationN)rrr'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGrrrrrr_srzinstance segmentationc@seZdZeZdS)AutoModelForObjectDetectionN)rrr"MODEL_FOR_OBJECT_DETECTION_MAPPINGrrrrrrhsrzobject detectionc@seZdZeZdS)#AutoModelForZeroShotObjectDetectionN)rrr,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGrrrrrrosrzzero-shot object detectionc@seZdZeZdS)AutoModelForDepthEstimationN)rrr"MODEL_FOR_DEPTH_ESTIMATION_MAPPINGrrrrrrxsrzdepth estimationc@seZdZeZdS)AutoModelForVideoClassificationN)rrr&MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGrrrrrrsrzvideo classificationc@seZdZeZdS)AutoModelForVision2SeqN)rrrMODEL_FOR_VISION_2_SEQ_MAPPINGrrrrrrsrzvision-to-text modelingc@seZdZeZdS)AutoModelForAudioClassificationN)rrr&MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGrrrrrr sr zaudio classificationc@seZdZeZdS)AutoModelForCTCN)rrrMODEL_FOR_CTC_MAPPINGrrrrrr"sr"z%connectionist temporal classificationc@seZdZeZdS)AutoModelForSpeechSeq2SeqN)rrr"MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGrrrrrr$sr$z,sequence-to-sequence speech-to-text modelingc@seZdZeZdS)$AutoModelForAudioFrameClassificationN)rrr,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGrrrrrr&sr&z"audio frame (token) classificationc@seZdZeZdS)AutoModelForAudioXVectorN)rrrMODEL_FOR_AUDIO_XVECTOR_MAPPINGrrrrrr(sr(c@seZdZeZdS)AutoModelForTextToSpectrogramN)rrr%MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGrrrrrr*sr*c@seZdZeZdS)AutoModelForTextToWaveformN)rrr"MODEL_FOR_TEXT_TO_WAVEFORM_MAPPINGrrrrrr,sr,c@seZdZeZdS) AutoBackboneN)rrrMODEL_FOR_BACKBONE_MAPPINGrrrrrr.sr.zaudio retrieval via x-vectorc@seZdZeZdS)AutoModelForMaskedImageModelingN)rrr'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGrrrrrr0sr0zmasked image modelingcs0eZdZefddZefddZZS)AutoModelWithLMHeadcstdtt|SNzThe 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.)warningswarn FutureWarningsuper from_config)clsconfig __class__rrr8s zAutoModelWithLMHead.from_configcs tdttj|f||Sr3)r4r5r6r7from_pretrained)r9Zpretrained_model_name_or_pathZ model_argskwargsr;rrr=s z#AutoModelWithLMHead.from_pretrained)rrr classmethodr8r= __classcell__rrr;rr2s r2)__doc__r4 collectionsrutilsrZ auto_factoryrrrr Zconfiguration_autor Z get_loggerrloggerZMODEL_MAPPING_NAMESZ#MODEL_FOR_PRETRAINING_MAPPING_NAMESZ MODEL_WITH_LM_HEAD_MAPPING_NAMESZ!MODEL_FOR_CAUSAL_LM_MAPPING_NAMESZMODEL_FOR_IMAGE_MAPPING_NAMESZ-MODEL_FOR_MASKED_IMAGE_MODELING_MAPPING_NAMESZ-MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPING_NAMESZ,MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMESZ*MODEL_FOR_IMAGE_SEGMENTATION_MAPPING_NAMESZ-MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING_NAMESZ-MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING_NAMESZ.MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPING_NAMESZ,MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING_NAMESZ$MODEL_FOR_VISION_2_SEQ_MAPPING_NAMESZ!MODEL_FOR_MASKED_LM_MAPPING_NAMESZ(MODEL_FOR_OBJECT_DETECTION_MAPPING_NAMESZ2MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING_NAMESZ(MODEL_FOR_DEPTH_ESTIMATION_MAPPING_NAMESZ,MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMESZ(MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMESZ/MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMESZ*MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMESZ0MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPING_NAMESZ1MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING_NAMESZ3MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING_NAMESZ,MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMESZ'MODEL_FOR_MULTIPLE_CHOICE_MAPPING_NAMESZ0MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING_NAMESZ,MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMESZMODEL_FOR_CTC_MAPPING_NAMESZ2MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPING_NAMESZ%MODEL_FOR_AUDIO_XVECTOR_MAPPING_NAMESZ+MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING_NAMESZ(MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING_NAMESZ6MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING_NAMESZ MODEL_FOR_BACKBONE_MAPPING_NAMESZ'MODEL_FOR_MASK_GENERATION_MAPPING_NAMESZ*MODEL_FOR_KEYPOINT_DETECTION_MAPPING_NAMESZ%MODEL_FOR_TEXT_ENCODING_MAPPING_NAMESZ2MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING_NAMESZ.MODEL_FOR_TIME_SERIES_REGRESSION_MAPPING_NAMESZ&MODEL_FOR_IMAGE_TO_IMAGE_MAPPING_NAMESrrrrZ'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGr r rrrrrrrrrZMODEL_FOR_IMAGE_MAPPINGr1rrrrrrrrrr r!r#r%r'r)r+r-r/rrrZ,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPINGZ(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPINGrrrrrrrrrrrrrrrrrrrr r rrrrrrrrrr r"r$r&r(r*r,r.r0r2rrrrs     vNUW5 = 3    ^FM,