U ,-e@sdZddlZddlmZddlmZddlmZmZm 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~ddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddgăZeddddddddddddddddddddddTddddddddddddddddddddddddddddddddddddddddg>Zed dΐd d d d ddҐddddddddאdddddڐddݐddddddddddTdddddddddd d!ddddd"dd#dd$d%d&d'd(d)ddd*d+dd,dddddd-d.dddddd/gMZed0d1d2d3d4d5d6d7dѐd8d9dddӐd:d;d<d=ddddddddd>d?d@dAd dBddCdDdEddFdGdHdIdJdKd(dLdMdNdOdPddQddRdSddTdUdVddWg<ZedXdYdZd[d\gZed]gZed^d_d`dadbdcdddedfdgdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}g Zed~gZeddddddddgZedgZed~dddgZedddgZeddddddgZed dΐd d dҐdddddאddddڐdddddddddd!dddd"d#dd'dd)ddd*d+ddd-ddddd/g-ZeddddddgZedgZeddgZ edΐd dd dddTdddddddddd$d%dddddgZ!ed&dd,dd.gZ"edddddddddddddddddddddddddddd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ߐdddddddddddddddddddgLZ#eddddddddddddddddddddddd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0g>Z$ed1gZ%edd2gZ&ed3d d gZ'ed4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKdLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjg7Z(edkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddg&Z)edddddddgZ*edddddddddddg Z+eddddddddddg Z,edddddgZ-edddddgZ.edgZ/edddgZ0ed d dd d"d#gZ1eddddddddddddg Z2edgZ3ed ddd.d0d1d9d?dEdXdddtddddddddddŐddddgZ4edgZ5e e eZ6e e eZ7e e eZ8e e eZ9e e eZ:e e eZ;e e e1Ze e eZ?e e eZ@e e eZAe e eZBe e e&ZCe e e'ZDe e eZEe e eZFe e eZGe e eZHe e e ZIe e e!ZJe e e#ZKe e e$ZLe 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 e0ZWe e e2ZXe e e3ZYe e e4ZZe e e5Z[GdȐdɄdeZ\Gdʐd˄deZ]Gd̐d̈́deZ^GdΐdτdeZ_e e_Z_GdАdфdeZ`e e`dҐdӍZ`GdԐdՄdeZae ead֐dӍZaGdאd؄deZbe ebdِdӍZbGdڐdۄdeZce ecdܐdӍZcGdݐdބdeZde eddߐddZdGdddeZee eeddӍZeGdddeZfe efddӍZfGdddeZge egdddZgGdddeZhe ehdddZhGdddeZie eidddZiGdddeZje ejddӍZjGdddeZke ekddӍZkGdddeZle elddӍZlGdddeZme emddӍZmGdddeZne enddӍZnGdddeZoe eoddӍZoGdddeZpe epddӍZpGd d d eZqe eqd dӍZqGd d d eZre erddӍZrGdddeZse esddӍZsGdddeZte etddӍZtGdddeZue euddӍZuGdddeZve evddӍZvGdddeZwe ewddӍZwGdddeZxe exd dӍZxGd!d"d"eZye eyd#dӍZyGd$d%d%eZze ezd&dӍZzGd'd(d(eZ{e e{d)dӍZ{Gd*d+d+eZ|Gd,d-d-eZ}Gd.d/d/eZ~Gd0d1d1eZe e|d2dӍZ|Gd3d4d4eZe ed5dӍZGd6d7d7eaZdS(8z Auto Model class.N) OrderedDict)logging)_BaseAutoBackboneClass_BaseAutoModelClass_LazyAutoMappingauto_class_update)CONFIG_MAPPING_NAMES)albertZ AlbertModel)alignZ 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)Z chinese_clipZChineseCLIPModel)clapZ ClapModel)ZclipZ CLIPModel)ZclipsegZ CLIPSegModel) code_llama LlamaModel)codegenZ CodeGenModel)conditional_detrZConditionalDetrModel)convbertZ ConvBertModel)convnextZ ConvNextModel) convnextv2ZConvNextV2Model)cpmantZ CpmAntModel)ctrlZ CTRLModel)cvtZCvtModel)data2vec-audioZData2VecAudioModel) data2vec-textZData2VecTextModel)data2vec-visionZData2VecVisionModel)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)flaubertZ FlaubertModel)flavaZ FlavaModel)fnetZ FNetModel)focalnetZ FocalNetModel)fsmtZ FSMTModel)funnel)Z FunnelModelZFunnelBaseModel)gitZGitModel)glpnZ GLPNModel)gpt-sw3 GPT2Model)gpt2rF) gpt_bigcodeZGPTBigCodeModel)gpt_neoZ GPTNeoModel)gpt_neoxZ GPTNeoXModel)gpt_neox_japaneseZGPTNeoXJapaneseModel)gptjZ GPTJModel)zgptsan-japaneseZ&GPTSanJapaneseForConditionalGeneration)Z graphormerZGraphormerModel)ZgroupvitZ GroupViTModel)hubertZ HubertModel)ibertZ IBertModel)ideficsZ IdeficsModel)imagegptZ ImageGPTModel)ZinformerZ InformerModel)ZjukeboxZ JukeboxModel)layoutlmZ LayoutLMModel) layoutlmv2ZLayoutLMv2Model) layoutlmv3ZLayoutLMv3Model)ledZLEDModel)levitZ LevitModel)liltZ LiltModel)llamar ) longformerZLongformerModel)longt5Z LongT5Model)lukeZ LukeModel)lxmertZ LxmertModel)m2m_100Z M2M100Model)marianZ MarianModel)markuplmZ MarkupLMModel) mask2formerZMask2FormerModel) maskformerZMaskFormerModel)maskformer-swinZMaskFormerSwinModel)mbartZ MBartModel)mctctZ MCTCTModel)megaZ MegaModel) megatron-bertZMegatronBertModel)zmgp-strZMgpstrForSceneTextRecognition)mistralZ MistralModel) mobilebertZMobileBertModel) mobilenet_v1ZMobileNetV1Model) mobilenet_v2ZMobileNetV2Model) mobilevitZMobileViTModel) mobilevitv2ZMobileViTV2Model)mpnetZ MPNetModel)mptZMptModel)mraZMraModel)mt5ZMT5Model)mvpZMvpModel)natZNatModel)nezhaZ NezhaModel)nllb-moeZ NllbMoeModel) nystromformerZNystromformerModel) oneformerZOneFormerModel) open-llamaZOpenLlamaModel) openai-gptZOpenAIGPTModel)optZOPTModel)owlvitZ OwlViTModel)pegasusZ PegasusModel) pegasus_xZ PegasusXModel) perceiverZPerceiverModel) persimmonZPersimmonModel)plbartZ PLBartModel) poolformerZPoolFormerModel) prophetnetZProphetNetModel)pvtZPvtModel)qdqbertZ QDQBertModel)reformerZ ReformerModel)regnetZ RegNetModel)rembertZ RemBertModel)resnetZ ResNetModel)Z retribertZRetriBertModel)robertaZ RobertaModel)roberta-prelayernormZRobertaPreLayerNormModel)roc_bertZ RoCBertModel)roformerZ RoFormerModel)rwkvZ RwkvModel)ZsamZSamModel) segformerZSegformerModel)sewZSEWModel)sew-dZ SEWDModel)speech_to_textZSpeech2TextModel)speecht5Z SpeechT5Model)splinterZ SplinterModel) squeezebertZSqueezeBertModel) 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)umt5Z UMT5Model) unispeechZUniSpeechModel) unispeech-satZUniSpeechSatModel)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-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)rZBartForConditionalGeneration)rZBertForPreTraining)rZBigBirdForPreTraining)rZBloomForCausalLM)rZCamembertForMaskedLM)r'ZCTRLLMHeadModel)r*ZData2VecTextForMaskedLM)r,ZDebertaForMaskedLM)r-ZDebertaV2ForMaskedLM)r4ZDistilBertForMaskedLM)r8ZElectraForPreTraining)r9ZErnieForPreTraining)r=ZFlaubertWithLMHeadModel)r>ZFlavaForPreTraining)r?ZFNetForPreTraining)rAZFSMTForConditionalGeneration)rBZFunnelForPreTraining)rEGPT2LMHeadModel)rGr)rHZGPTBigCodeForCausalLM)rNZIBertForMaskedLM)rOZIdeficsForVisionText2Text)rQZLayoutLMForMaskedLM)rXZLongformerForMaskedLM)rZZLukeForMaskedLM)r[ZLxmertForPreTraining)rdZMegaForMaskedLM)reZMegatronBertForPreTraining)rgZMobileBertForPreTraining)rlZMPNetForMaskedLM)rmZMptForCausalLM)rnZMraForMaskedLM)rpZMvpForConditionalGeneration)rrZNezhaForPreTraining)rsZNllbMoeForConditionalGeneration)rwZOpenAIGPTLMHeadModel)rZRobertaForMaskedLM)rZRobertaPreLayerNormForMaskedLM)rZRoCBertForPreTraining)rZRwkvForCausalLM)rZSplinterForPreTraining)rZSqueezeBertForMaskedLM)rZ*SwitchTransformersForConditionalGeneration)rZT5ForConditionalGeneration)rZTapasForMaskedLM)rZTransfoXLLMHeadModel)rZTvltForPreTraining)rZUniSpeechForPreTraining)rZUniSpeechSatForPreTraining)rZVideoMAEForPreTraining)rZVisualBertForPreTraining)rZViTMAEForPreTraining)rZWav2Vec2ForPreTraining)rZWav2Vec2ConformerForPreTraining)rZXLMWithLMHeadModel)rZXLMRobertaForMaskedLM)rZXLMRobertaXLForMaskedLM)rZXLNetLMHeadModel)rZXmodForMaskedLM)r ZAlbertForMaskedLM)rZBertForMaskedLM)rZBigBirdForMaskedLM)rZ&BigBirdPegasusForConditionalGeneration)rZ'BlenderbotSmallForConditionalGeneration)r!ZCodeGenForCausalLM)r#ZConvBertForMaskedLM)r&ZCpmAntForCausalLM)r8ZElectraForMaskedLM)zencoder-decoderZEncoderDecoderModel)r9ZErnieForMaskedLM)r;ZEsmForMaskedLM)r?ZFNetForMaskedLM)rBZFunnelForMaskedLM)rCZGitForCausalLM)rIZGPTNeoForCausalLM)rJZGPTNeoXForCausalLM)rKZGPTNeoXJapaneseForCausalLM)rLZGPTJForCausalLM)rTZLEDForConditionalGeneration)rYZLongT5ForConditionalGeneration)r\ZM2M100ForConditionalGeneration)r]Z MarianMTModel)reZMegatronBertForCausalLM)rgZMobileBertForMaskedLM)rrZNezhaForMaskedLM)rtZNystromformerForMaskedLM)r{Z PegasusXForConditionalGeneration)r~ZPLBartForConditionalGeneration)Z pop2pianoZ!Pop2PianoForConditionalGeneration)rZQDQBertForMaskedLM)rZReformerModelWithLMHead)rZRemBertForMaskedLM)rZRoCBertForMaskedLM)rZRoFormerForMaskedLM)rZ#Speech2TextForConditionalGeneration)rZWav2Vec2ForMaskedLM)rZWhisperForConditionalGeneration)rZYosoForMaskedLM)rZBartForCausalLM)rZBertLMHeadModel)rZBertGenerationDecoder)rZBigBirdForCausalLM)rZBigBirdPegasusForCausalLM)rZBioGptForCausalLM)rZBlenderbotForCausalLM)rZBlenderbotSmallForCausalLM)rZCamembertForCausalLM)rLlamaForCausalLM)r*ZData2VecTextForCausalLM)r8ZElectraForCausalLM)r9ZErnieForCausalLM)r<ZFalconForCausalLM)rWr)r]ZMarianForCausalLM)rbZMBartForCausalLM)rdZMegaForCausalLM)rfZMistralForCausalLM)musicgenZMusicgenForCausalLM)rpZMvpForCausalLM)rvZOpenLlamaForCausalLM)rxZOPTForCausalLM)rzZPegasusForCausalLM)r}ZPersimmonForCausalLM)r~ZPLBartForCausalLM)rZProphetNetForCausalLM)rZQDQBertLMHeadModel)rZRemBertForCausalLM)rZRobertaForCausalLM)rZRobertaPreLayerNormForCausalLM)rZRoCBertForCausalLM)rZRoFormerForCausalLM)Zspeech_to_text_2ZSpeech2Text2ForCausalLM)ZtrocrZTrOCRForCausalLM)rZXGLMForCausalLM)rZXLMProphetNetForCausalLM)rZXLMRobertaForCausalLM)rZXLMRobertaXLForCausalLM)rZXmodForCausalLM)r/ZDeiTForMaskedImageModeling)r@ZFocalNetForMaskedImageModeling)rZSwinForMaskedImageModeling)rZSwinv2ForMaskedImageModeling)rZViTForMaskedImageModeling)rPZImageGPTForCausalImageModeling)rZBeitForImageClassification)rZBitForImageClassification)r$ZConvNextForImageClassification)r%Z ConvNextV2ForImageClassification)r(ZCvtForImageClassification)r+Z$Data2VecVisionForImageClassification)r/)ZDeiTForImageClassificationZ%DeiTForImageClassificationWithTeacher)r2ZDinatForImageClassification)r3ZDinov2ForImageClassification)r6)Z%EfficientFormerForImageClassificationZ0EfficientFormerForImageClassificationWithTeacher)r7Z"EfficientNetForImageClassification)r@ZFocalNetForImageClassification)rPZImageGPTForImageClassification)rU)ZLevitForImageClassificationZ&LevitForImageClassificationWithTeacher)rhZ!MobileNetV1ForImageClassification)riZ!MobileNetV2ForImageClassification)rjZMobileViTForImageClassification)rkZ!MobileViTV2ForImageClassification)rqZNatForImageClassification)r|)Z&PerceiverForImageClassificationLearnedZ&PerceiverForImageClassificationFourierZ-PerceiverForImageClassificationConvProcessing)rZ PoolFormerForImageClassification)rZPvtForImageClassification)rZRegNetForImageClassification)rZResNetForImageClassification)rZSegformerForImageClassification)rZ!SwiftFormerForImageClassification)rZSwinForImageClassification)rZSwinv2ForImageClassification)rZVanForImageClassification)rZViTForImageClassification)rZViTHybridForImageClassification)rZViTMSNForImageClassification)r1ZDetrForSegmentation)rZBeitForSemanticSegmentation)r+Z%Data2VecVisionForSemanticSegmentation)r5ZDPTForSemanticSegmentation)riZ"MobileNetV2ForSemanticSegmentation)rjZ MobileViTForSemanticSegmentation)rkZ"MobileViTV2ForSemanticSegmentation)rZ SegformerForSemanticSegmentation)ZupernetZUperNetForSemanticSegmentation)r`Z!MaskFormerForInstanceSegmentation)r_Z#Mask2FormerForUniversalSegmentation)ruZ!OneFormerForUniversalSegmentation)rZ!TimesformerForVideoClassification)rZVideoMAEForVideoClassification)rZVivitForVideoClassification)rZBlipForConditionalGeneration)rZBlip2ForConditionalGeneration)Z instructblipZ$InstructBlipForConditionalGeneration)Z pix2structZ"Pix2StructForConditionalGeneration)zvision-encoder-decoderZVisionEncoderDecoderModel)rbZMBartForConditionalGeneration)reZMegatronBertForMaskedLM)r|ZPerceiverForMaskedLM)rZReformerForMaskedLM)r"Z!ConditionalDetrForObjectDetection)r.Z DeformableDetrForObjectDetection)r0ZDetaForObjectDetection)r1ZDetrForObjectDetection)rZ"TableTransformerForObjectDetection)rZYolosForObjectDetection)ryZOwlViTForObjectDetection)r5ZDPTForDepthEstimation)rDZGLPNForDepthEstimation)rZ"BlenderbotForConditionalGeneration)roZMT5ForConditionalGeneration)rzZPegasusForConditionalGeneration)rZ"ProphetNetForConditionalGeneration)rZUMT5ForConditionalGeneration)rZ%XLMProphetNetForConditionalGeneration)zspeech-encoder-decoderZSpeechEncoderDecoderModel)rZSpeechT5ForSpeechToText)r ZAlbertForSequenceClassification)rZBartForSequenceClassification)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)r4Z#DistilBertForSequenceClassification)r8Z ElectraForSequenceClassification)r9ZErnieForSequenceClassification)r:ZErnieMForSequenceClassification)r;ZEsmForSequenceClassification)r<ZFalconForSequenceClassification)r=Z!FlaubertForSequenceClassification)r?ZFNetForSequenceClassification)rBZFunnelForSequenceClassification)rEGPT2ForSequenceClassification)rGr)rHZ#GPTBigCodeForSequenceClassification)rIZGPTNeoForSequenceClassification)rJZ GPTNeoXForSequenceClassification)rLZGPTJForSequenceClassification)rNZIBertForSequenceClassification)rQZ!LayoutLMForSequenceClassification)rRZ#LayoutLMv2ForSequenceClassification)rSZ#LayoutLMv3ForSequenceClassification)rTZLEDForSequenceClassification)rVZLiltForSequenceClassification)rWr)rXZ#LongformerForSequenceClassification)rZZLukeForSequenceClassification)r^Z!MarkupLMForSequenceClassification)rbZMBartForSequenceClassification)rdZMegaForSequenceClassification)reZ%MegatronBertForSequenceClassification)rfZ MistralForSequenceClassification)rgZ#MobileBertForSequenceClassification)rlZMPNetForSequenceClassification)rmZMptForSequenceClassification)rnZMraForSequenceClassification)roZMT5ForSequenceClassification)rpZMvpForSequenceClassification)rrZNezhaForSequenceClassification)rtZ&NystromformerForSequenceClassification)rvZ"OpenLlamaForSequenceClassification)rwZ"OpenAIGPTForSequenceClassification)rxZOPTForSequenceClassification)r|Z"PerceiverForSequenceClassification)r}Z"PersimmonForSequenceClassification)r~ZPLBartForSequenceClassification)rZ QDQBertForSequenceClassification)rZ!ReformerForSequenceClassification)rZ RemBertForSequenceClassification)rZ RobertaForSequenceClassification)rZ,RobertaPreLayerNormForSequenceClassification)rZ RoCBertForSequenceClassification)rZ!RoFormerForSequenceClassification)rZ$SqueezeBertForSequenceClassification)rZT5ForSequenceClassification)rZTapasForSequenceClassification)rZ"TransfoXLForSequenceClassification)rZUMT5ForSequenceClassification)rZXLMForSequenceClassification)rZ#XLMRobertaForSequenceClassification)rZ%XLMRobertaXLForSequenceClassification)rZXLNetForSequenceClassification)rZXmodForSequenceClassification)rZYosoForSequenceClassification)r ZAlbertForQuestionAnswering)rZBartForQuestionAnswering)rZBertForQuestionAnswering)rZBigBirdForQuestionAnswering)rZ"BigBirdPegasusForQuestionAnswering)rZBloomForQuestionAnswering)rZCamembertForQuestionAnswering)rZCanineForQuestionAnswering)r#ZConvBertForQuestionAnswering)r*Z Data2VecTextForQuestionAnswering)r,ZDebertaForQuestionAnswering)r-ZDebertaV2ForQuestionAnswering)r4ZDistilBertForQuestionAnswering)r8ZElectraForQuestionAnswering)r9ZErnieForQuestionAnswering)r:ZErnieMForQuestionAnswering)r<ZFalconForQuestionAnswering)r=Z"FlaubertForQuestionAnsweringSimple)r?ZFNetForQuestionAnswering)rBZFunnelForQuestionAnswering)rGZGPT2ForQuestionAnswering)rIZGPTNeoForQuestionAnswering)rJZGPTNeoXForQuestionAnswering)rLZGPTJForQuestionAnswering)rNZIBertForQuestionAnswering)rRZLayoutLMv2ForQuestionAnswering)rSZLayoutLMv3ForQuestionAnswering)rTZLEDForQuestionAnswering)rVZLiltForQuestionAnswering)rXZLongformerForQuestionAnswering)rZZLukeForQuestionAnswering)r[ZLxmertForQuestionAnswering)r^ZMarkupLMForQuestionAnswering)rbZMBartForQuestionAnswering)rdZMegaForQuestionAnswering)reZ MegatronBertForQuestionAnswering)rgZMobileBertForQuestionAnswering)rlZMPNetForQuestionAnswering)rmZMptForQuestionAnswering)rnZMraForQuestionAnswering)roZMT5ForQuestionAnswering)rpZMvpForQuestionAnswering)rrZNezhaForQuestionAnswering)rtZ!NystromformerForQuestionAnswering)rxZOPTForQuestionAnswering)rZQDQBertForQuestionAnswering)rZReformerForQuestionAnswering)rZRemBertForQuestionAnswering)rZRobertaForQuestionAnswering)rZ'RobertaPreLayerNormForQuestionAnswering)rZRoCBertForQuestionAnswering)rZRoFormerForQuestionAnswering)rZSplinterForQuestionAnswering)rZSqueezeBertForQuestionAnswering)rZT5ForQuestionAnswering)rZUMT5ForQuestionAnswering)rZXLMForQuestionAnsweringSimple)rZXLMRobertaForQuestionAnswering)rZ XLMRobertaXLForQuestionAnswering)rZXLNetForQuestionAnsweringSimple)rZXmodForQuestionAnswering)rZYosoForQuestionAnswering)rZTapasForQuestionAnswering)rZViltForQuestionAnswering)rQZLayoutLMForQuestionAnswering)r ZAlbertForTokenClassification)rZBertForTokenClassification)rZBigBirdForTokenClassification)rZBioGptForTokenClassification)rZBloomForTokenClassification)rZBrosForTokenClassification)rZCamembertForTokenClassification)rZCanineForTokenClassification)r#ZConvBertForTokenClassification)r*Z"Data2VecTextForTokenClassification)r,ZDebertaForTokenClassification)r-ZDebertaV2ForTokenClassification)r4Z DistilBertForTokenClassification)r8ZElectraForTokenClassification)r9ZErnieForTokenClassification)r:ZErnieMForTokenClassification)r;ZEsmForTokenClassification)r<ZFalconForTokenClassification)r=ZFlaubertForTokenClassification)r?ZFNetForTokenClassification)rBZFunnelForTokenClassification)rEGPT2ForTokenClassification)rGr)rHZ GPTBigCodeForTokenClassification)rIZGPTNeoForTokenClassification)rJZGPTNeoXForTokenClassification)rNZIBertForTokenClassification)rQZLayoutLMForTokenClassification)rRZ LayoutLMv2ForTokenClassification)rSZ LayoutLMv3ForTokenClassification)rVZLiltForTokenClassification)rXZ LongformerForTokenClassification)rZZLukeForTokenClassification)r^ZMarkupLMForTokenClassification)rdZMegaForTokenClassification)reZ"MegatronBertForTokenClassification)rgZ MobileBertForTokenClassification)rlZMPNetForTokenClassification)rmZMptForTokenClassification)rnZMraForTokenClassification)rrZNezhaForTokenClassification)rtZ#NystromformerForTokenClassification)rZQDQBertForTokenClassification)rZRemBertForTokenClassification)rZRobertaForTokenClassification)rZ)RobertaPreLayerNormForTokenClassification)rZRoCBertForTokenClassification)rZRoFormerForTokenClassification)rZ!SqueezeBertForTokenClassification)rZXLMForTokenClassification)rZ XLMRobertaForTokenClassification)rZ"XLMRobertaXLForTokenClassification)rZXLNetForTokenClassification)rZXmodForTokenClassification)rZYosoForTokenClassification)r ZAlbertForMultipleChoice)rZBertForMultipleChoice)rZBigBirdForMultipleChoice)rZCamembertForMultipleChoice)rZCanineForMultipleChoice)r#ZConvBertForMultipleChoice)r*ZData2VecTextForMultipleChoice)r-ZDebertaV2ForMultipleChoice)r4ZDistilBertForMultipleChoice)r8ZElectraForMultipleChoice)r9ZErnieForMultipleChoice)r:ZErnieMForMultipleChoice)r=ZFlaubertForMultipleChoice)r?ZFNetForMultipleChoice)rBZFunnelForMultipleChoice)rNZIBertForMultipleChoice)rXZLongformerForMultipleChoice)rZZLukeForMultipleChoice)rdZMegaForMultipleChoice)reZMegatronBertForMultipleChoice)rgZMobileBertForMultipleChoice)rlZMPNetForMultipleChoice)rnZMraForMultipleChoice)rrZNezhaForMultipleChoice)rtZNystromformerForMultipleChoice)rZQDQBertForMultipleChoice)rZRemBertForMultipleChoice)rZRobertaForMultipleChoice)rZ$RobertaPreLayerNormForMultipleChoice)rZRoCBertForMultipleChoice)rZRoFormerForMultipleChoice)rZSqueezeBertForMultipleChoice)rZXLMForMultipleChoice)rZXLMRobertaForMultipleChoice)rZXLMRobertaXLForMultipleChoice)rZXLNetForMultipleChoice)rZXmodForMultipleChoice)rZYosoForMultipleChoice)rZBertForNextSentencePrediction)r9ZErnieForNextSentencePrediction)r?ZFNetForNextSentencePrediction)reZ%MegatronBertForNextSentencePrediction)rgZ#MobileBertForNextSentencePrediction)rrZNezhaForNextSentencePrediction)rZ QDQBertForNextSentencePrediction)r ZASTForAudioClassification)r)Z&Data2VecAudioForSequenceClassification)rMZHubertForSequenceClassification)rZSEWForSequenceClassification)rZSEWDForSequenceClassification)rZ"UniSpeechForSequenceClassification)rZ%UniSpeechSatForSequenceClassification)rZ!Wav2Vec2ForSequenceClassification)rZ*Wav2Vec2ConformerForSequenceClassification)rZWavLMForSequenceClassification)rZWhisperForAudioClassification)r)ZData2VecAudioForCTC)rMZ HubertForCTC)rcZ MCTCTForCTC)rZ SEWForCTC)rZ SEWDForCTC)rZUniSpeechForCTC)rZUniSpeechSatForCTC)rZWav2Vec2ForCTC)rZWav2Vec2ConformerForCTC)rZ WavLMForCTC)r)Z(Data2VecAudioForAudioFrameClassification)rZ'UniSpeechSatForAudioFrameClassification)rZ#Wav2Vec2ForAudioFrameClassification)rZ,Wav2Vec2ConformerForAudioFrameClassification)rZ WavLMForAudioFrameClassification)r)ZData2VecAudioForXVector)rZUniSpeechSatForXVector)rZWav2Vec2ForXVector)rZWav2Vec2ConformerForXVector)rZWavLMForXVector)rZSpeechT5ForTextToSpeech)rZ MusicgenForConditionalGeneration)rZ BitBackbone)r$ZConvNextBackbone)r%ZConvNextV2Backbone)r2Z DinatBackbone)r3ZDinov2Backbone)r@ZFocalNetBackbone)raZMaskFormerSwinBackbone)rqZ NatBackbone)rZResNetBackbone)rZ SwinBackbone)rZVitDetBackbone)roZMT5EncoderModel)rZT5EncoderModel)rZUMT5EncoderModel)rZSwin2SRForImageSuperResolutionc@seZdZeZdS)AutoModelForMaskGenerationN)__name__ __module__ __qualname__!MODEL_FOR_MASK_GENERATION_MAPPING_model_mappingrrg/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/transformers/models/auto/modeling_auto.pyrsrc@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 modelingzt5-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_MAPPINGrrrrrrsrzvisual 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_MAPPINGrrrrrr&srztoken classificationc@seZdZeZdS)AutoModelForMultipleChoiceN)rrr!MODEL_FOR_MULTIPLE_CHOICE_MAPPINGrrrrrr-srzmultiple choicec@seZdZeZdS)"AutoModelForNextSentencePredictionN)rrr*MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGrrrrrr4srznext sentence predictionc@seZdZeZdS)AutoModelForImageClassificationN)rrr&MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGrrrrrr=srzimage classificationc@seZdZeZdS)'AutoModelForZeroShotImageClassificationN)rrr0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGrrrrrrDsrzzero-shot image classificationc@seZdZeZdS)AutoModelForImageSegmentationN)rrr$MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGrrrrrrMsrzimage segmentationc@seZdZeZdS) AutoModelForSemanticSegmentationN)rrr'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGrrrrrrTsrzsemantic segmentationc@seZdZeZdS)!AutoModelForUniversalSegmentationN)rrr(MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGrrrrrr]srzuniversal image segmentationc@seZdZeZdS) AutoModelForInstanceSegmentationN)rrr'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGrrrrrrfsrzinstance segmentationc@seZdZeZdS)AutoModelForObjectDetectionN)rrr"MODEL_FOR_OBJECT_DETECTION_MAPPINGrrrrrrosrzobject detectionc@seZdZeZdS)#AutoModelForZeroShotObjectDetectionN)rrr,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGrrrrrrvsrzzero-shot object detectionc@seZdZeZdS)AutoModelForDepthEstimationN)rrr"MODEL_FOR_DEPTH_ESTIMATION_MAPPINGrrrrrrsrzdepth 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_MAPPINGrrrrrrsrzaudio classificationc@seZdZeZdS)AutoModelForCTCN)rrrMODEL_FOR_CTC_MAPPINGrrrrrrsrz%connectionist temporal classificationc@seZdZeZdS)AutoModelForSpeechSeq2SeqN)rrr"MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGrrrrrrsrz,sequence-to-sequence speech-to-text modelingc@seZdZeZdS)$AutoModelForAudioFrameClassificationN)rrr,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGrrrrrrsrz"audio frame (token) classificationc@seZdZeZdS)AutoModelForAudioXVectorN)rrrMODEL_FOR_AUDIO_XVECTOR_MAPPINGrrrrrrsrc@seZdZeZdS)AutoModelForTextToSpectrogramN)rrr%MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPINGrrrrrrsrc@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_MAPPINGrrrrrr sr zmasked 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__rrrs zAutoModelWithLMHead.from_configcs tdttj|f||Sr)rrrrfrom_pretrained)rZpretrained_model_name_or_pathZ model_argskwargsrrrrs z#AutoModelWithLMHead.from_pretrained)rrr classmethodrr __classcell__rrrrrs r)__doc__r 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_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_TEXT_ENCODING_MAPPING_NAMESZ&MODEL_FOR_IMAGE_TO_IMAGE_MAPPING_NAMESrrrrZ'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr r r rrrrrs\     KDSB 3  3  RD=,