Ë
    S^(hç8  ã                   ó’  — d Z ddlmZ ddlmZ ddl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dg«      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ddg«      Z edg«      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
e«      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
e«      Z) G d„ de«      Z* ee*«      Z* G d„ de«      Z+ ee+d¬«      Z+ G d„ de«      Z, ee,d¬«      Z, G d „ d!e«      Z- ee-d"¬«      Z- G d#„ d$e«      Z. ee.d%d&¬'«      Z. G d(„ d)e«      Z/ ee/d*¬«      Z/ G d+„ d,e«      Z0 ee0d-¬«      Z0 G d.„ d/e«      Z1 ee1d0¬«      Z1 G d1„ d2e«      Z2 ee2d3¬«      Z2 G d4„ d5e«      Z3 ee3d6¬«      Z3 G d7„ d8e«      Z4 ee4d9¬«      Z4 G d:„ d;e«      Z5 ee5d<¬«      Z5 G d=„ d>e«      Z6 ee6d?¬«      Z6y@)AzAuto Model class.é    )ÚOrderedDicté   )Úloggingé   )Ú_BaseAutoModelClassÚ_LazyAutoMappingÚauto_class_update)ÚCONFIG_MAPPING_NAMES)%)ÚalbertÚFlaxAlbertModel)ÚbartÚFlaxBartModel)ÚbeitÚFlaxBeitModel)ÚbertÚFlaxBertModel)Úbig_birdÚFlaxBigBirdModel)Ú
blenderbotÚFlaxBlenderbotModel)úblenderbot-smallÚFlaxBlenderbotSmallModel)ÚbloomÚFlaxBloomModel)ÚclipÚFlaxCLIPModel)Údinov2ÚFlaxDinov2Model)Ú
distilbertÚFlaxDistilBertModel)ÚelectraÚFlaxElectraModel)ÚgemmaÚFlaxGemmaModel)úgpt-sw3ÚFlaxGPT2Model)Úgpt2r&   )Úgpt_neoÚFlaxGPTNeoModel)ÚgptjÚFlaxGPTJModel)ÚllamaÚFlaxLlamaModel)Úlongt5ÚFlaxLongT5Model)ÚmarianÚFlaxMarianModel)ÚmbartÚFlaxMBartModel)ÚmistralÚFlaxMistralModel)Úmt5ÚFlaxMT5Model)ÚoptÚFlaxOPTModel)ÚpegasusÚFlaxPegasusModel)ÚregnetÚFlaxRegNetModel)ÚresnetÚFlaxResNetModel)ÚrobertaÚFlaxRobertaModel)úroberta-prelayernormÚFlaxRobertaPreLayerNormModel)ÚroformerÚFlaxRoFormerModel)Út5ÚFlaxT5Model)zvision-text-dual-encoderÚFlaxVisionTextDualEncoderModel)ÚvitÚFlaxViTModel)Úwav2vec2ÚFlaxWav2Vec2Model)ÚwhisperÚFlaxWhisperModel)ÚxglmÚFlaxXGLMModel)úxlm-robertaÚFlaxXLMRobertaModel))r   ÚFlaxAlbertForPreTraining©r   Ú FlaxBartForConditionalGeneration)r   ÚFlaxBertForPreTraining)r   ÚFlaxBigBirdForPreTraining)r!   ÚFlaxElectraForPreTraining©r.   Ú"FlaxLongT5ForConditionalGeneration©r2   Ú!FlaxMBartForConditionalGeneration©r6   ÚFlaxMT5ForConditionalGeneration©r@   ÚFlaxRobertaForMaskedLM©rB   Ú"FlaxRobertaPreLayerNormForMaskedLM©rD   ÚFlaxRoFormerForMaskedLM©rF   ÚFlaxT5ForConditionalGeneration)rK   ÚFlaxWav2Vec2ForPreTraining©rM   Ú#FlaxWhisperForConditionalGeneration©rQ   ÚFlaxXLMRobertaForMaskedLM))r   ÚFlaxAlbertForMaskedLMrT   )r   ÚFlaxBertForMaskedLM)r   ÚFlaxBigBirdForMaskedLM)r   ÚFlaxDistilBertForMaskedLM)r!   ÚFlaxElectraForMaskedLMr[   r_   ra   rc   rj   )
rT   )r   Ú&FlaxBlenderbotForConditionalGeneration)r   Ú+FlaxBlenderbotSmallForConditionalGeneration)zencoder-decoderÚFlaxEncoderDecoderModelrY   )r0   ÚFlaxMarianMTModelr[   r]   )r:   Ú#FlaxPegasusForConditionalGenerationre   ))r   ÚFlaxBeitForImageClassification)r   Ú FlaxDinov2ForImageClassification)r<   Ú FlaxRegNetForImageClassification)r>   Ú FlaxResNetForImageClassification)rI   ÚFlaxViTForImageClassification)zvision-encoder-decoderÚFlaxVisionEncoderDecoderModel))r   ÚFlaxBartForCausalLM)r   ÚFlaxBertForCausalLM)r   ÚFlaxBigBirdForCausalLM)r   ÚFlaxBloomForCausalLM)r!   ÚFlaxElectraForCausalLM)r#   ÚFlaxGemmaForCausalLM)r%   ÚFlaxGPT2LMHeadModel)r'   r‚   )r(   ÚFlaxGPTNeoForCausalLM)r*   ÚFlaxGPTJForCausalLM)r,   ÚFlaxLlamaForCausalLM)r4   ÚFlaxMistralForCausalLM)r8   ÚFlaxOPTForCausalLM)r@   ÚFlaxRobertaForCausalLM)rB   Ú"FlaxRobertaPreLayerNormForCausalLM)rO   ÚFlaxXGLMForCausalLM)rQ   ÚFlaxXLMRobertaForCausalLM))r   Ú#FlaxAlbertForSequenceClassification)r   Ú!FlaxBartForSequenceClassification)r   Ú!FlaxBertForSequenceClassification)r   Ú$FlaxBigBirdForSequenceClassification)r   Ú'FlaxDistilBertForSequenceClassification)r!   Ú$FlaxElectraForSequenceClassification)r2   Ú"FlaxMBartForSequenceClassification)r@   Ú$FlaxRobertaForSequenceClassification)rB   Ú0FlaxRobertaPreLayerNormForSequenceClassification)rD   Ú%FlaxRoFormerForSequenceClassification)rQ   Ú'FlaxXLMRobertaForSequenceClassification))r   ÚFlaxAlbertForQuestionAnswering)r   ÚFlaxBartForQuestionAnswering)r   ÚFlaxBertForQuestionAnswering)r   ÚFlaxBigBirdForQuestionAnswering)r   Ú"FlaxDistilBertForQuestionAnswering)r!   ÚFlaxElectraForQuestionAnswering)r2   ÚFlaxMBartForQuestionAnswering)r@   ÚFlaxRobertaForQuestionAnswering)rB   Ú+FlaxRobertaPreLayerNormForQuestionAnswering)rD   Ú FlaxRoFormerForQuestionAnswering)rQ   Ú"FlaxXLMRobertaForQuestionAnswering)	)r   Ú FlaxAlbertForTokenClassification)r   ÚFlaxBertForTokenClassification)r   Ú!FlaxBigBirdForTokenClassification)r   Ú$FlaxDistilBertForTokenClassification)r!   Ú!FlaxElectraForTokenClassification)r@   Ú!FlaxRobertaForTokenClassification)rB   Ú-FlaxRobertaPreLayerNormForTokenClassification)rD   Ú"FlaxRoFormerForTokenClassification)rQ   Ú$FlaxXLMRobertaForTokenClassification)	)r   ÚFlaxAlbertForMultipleChoice)r   ÚFlaxBertForMultipleChoice)r   ÚFlaxBigBirdForMultipleChoice)r   ÚFlaxDistilBertForMultipleChoice)r!   ÚFlaxElectraForMultipleChoice)r@   ÚFlaxRobertaForMultipleChoice)rB   Ú(FlaxRobertaPreLayerNormForMultipleChoice)rD   ÚFlaxRoFormerForMultipleChoice)rQ   ÚFlaxXLMRobertaForMultipleChoice)r   Ú!FlaxBertForNextSentencePrediction)zspeech-encoder-decoderÚFlaxSpeechEncoderDecoderModelrh   )rM   Ú!FlaxWhisperForAudioClassificationc                   ó   — e Zd ZeZy)ÚFlaxAutoModelN)Ú__name__Ú
__module__Ú__qualname__ÚFLAX_MODEL_MAPPINGÚ_model_mapping© ó    úi/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/transformers/models/auto/modeling_flax_auto.pyr¸   r¸     s   „ Ø'�Nr¿   r¸   c                   ó   — e Zd ZeZy)ÚFlaxAutoModelForPreTrainingN)r¹   rº   r»   Ú"FLAX_MODEL_FOR_PRETRAINING_MAPPINGr½   r¾   r¿   rÀ   rÂ   rÂ      s   „ Ø7�Nr¿   rÂ   Úpretraining)Úhead_docc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForCausalLMN)r¹   rº   r»   Ú FLAX_MODEL_FOR_CAUSAL_LM_MAPPINGr½   r¾   r¿   rÀ   rÇ   rÇ   '  ó   „ Ø5�Nr¿   rÇ   zcausal language modelingc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForMaskedLMN)r¹   rº   r»   Ú FLAX_MODEL_FOR_MASKED_LM_MAPPINGr½   r¾   r¿   rÀ   rË   rË   .  rÉ   r¿   rË   zmasked language modelingc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForSeq2SeqLMN)r¹   rº   r»   Ú+FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGr½   r¾   r¿   rÀ   rÎ   rÎ   5  ó   „ Ø@�Nr¿   rÎ   z&sequence-to-sequence language modelingzgoogle-t5/t5-base)rÅ   Úcheckpoint_for_examplec                   ó   — e Zd ZeZy)Ú&FlaxAutoModelForSequenceClassificationN)r¹   rº   r»   Ú.FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGr½   r¾   r¿   rÀ   rÓ   rÓ   @  s   „ ØC�Nr¿   rÓ   zsequence classificationc                   ó   — e Zd ZeZy)Ú!FlaxAutoModelForQuestionAnsweringN)r¹   rº   r»   Ú)FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPINGr½   r¾   r¿   rÀ   rÖ   rÖ   I  s   „ Ø>�Nr¿   rÖ   zquestion answeringc                   ó   — e Zd ZeZy)Ú#FlaxAutoModelForTokenClassificationN)r¹   rº   r»   Ú+FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGr½   r¾   r¿   rÀ   rÙ   rÙ   P  rÐ   r¿   rÙ   ztoken classificationc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForMultipleChoiceN)r¹   rº   r»   Ú&FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPINGr½   r¾   r¿   rÀ   rÜ   rÜ   Y  s   „ Ø;�Nr¿   rÜ   zmultiple choicec                   ó   — e Zd ZeZy)Ú&FlaxAutoModelForNextSentencePredictionN)r¹   rº   r»   Ú/FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGr½   r¾   r¿   rÀ   rß   rß   `  s   „ ØD�Nr¿   rß   znext sentence predictionc                   ó   — e Zd ZeZy)Ú#FlaxAutoModelForImageClassificationN)r¹   rº   r»   Ú+FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGr½   r¾   r¿   rÀ   râ   râ   i  rÐ   r¿   râ   zimage classificationc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForVision2SeqN)r¹   rº   r»   Ú#FLAX_MODEL_FOR_VISION_2_SEQ_MAPPINGr½   r¾   r¿   rÀ   rå   rå   r  s   „ Ø8�Nr¿   rå   zvision-to-text modelingc                   ó   — e Zd ZeZy)ÚFlaxAutoModelForSpeechSeq2SeqN)r¹   rº   r»   Ú'FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGr½   r¾   r¿   rÀ   rè   rè   y  s   „ Ø<�Nr¿   rè   z,sequence-to-sequence speech-to-text modelingN)7Ú__doc__Úcollectionsr   Úutilsr   Úauto_factoryr   r   r	   Úconfiguration_autor
   Ú
get_loggerr¹   ÚloggerÚFLAX_MODEL_MAPPING_NAMESÚ(FLAX_MODEL_FOR_PRETRAINING_MAPPING_NAMESÚ&FLAX_MODEL_FOR_MASKED_LM_MAPPING_NAMESÚ1FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING_NAMESÚ1FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING_NAMESÚ)FLAX_MODEL_FOR_VISION_2_SEQ_MAPPING_NAMESÚ&FLAX_MODEL_FOR_CAUSAL_LM_MAPPING_NAMESÚ4FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING_NAMESÚ/FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING_NAMESÚ1FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING_NAMESÚ,FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING_NAMESÚ5FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING_NAMESÚ-FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING_NAMESÚ1FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMESr¼   rÃ   rÌ   rÏ   rã   ræ   rÈ   rÔ   r×   rÚ   rÝ   rà   ré   Ú+FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGr¸   rÂ   rÇ   rË   rÎ   rÓ   rÖ   rÙ   rÜ   rß   râ   rå   rè   r¾   r¿   rÀ   ú<module>r      sÐ  ðñ å #å ß RÑ RÝ 4ð 
ˆ×	Ñ	˜HÓ	%€ñ 'ò'ó)Ð ñV ,7òó,Ð (ñ* *5òó*Ð &ñ" 5@òó5Ð 1ñ  5@òó	5Ð 1ñ -8àCðó-Ð )ñ *5òó*Ð &ñ. 8Còó8Ð 4ñ" 3>òó3Ð /ñ" 5@òó5Ð 1ñ 0;òó0Ð ,ñ 9Dà5ðó9Ð 5ñ 1<àCØ:ðó1Ð -ñ 5@à8ðó5Ð 1ñ &Ð&:Ð<TÓUÐ Ù%5Ð6JÐLtÓ%uÐ "Ù#3Ð4HÐJpÓ#qÐ  Ù.>ØÐKó/Ð +ñ /?ØÐKó/Ð +ñ '7Ð7KÐMvÓ&wÐ #Ù#3Ð4HÐJpÓ#qÐ  Ù1AØÐNó2Ð .ñ -=ØÐIó-Ð )ñ /?ØÐKó/Ð +ñ *:ØÐFó*Ð &ñ 3CØÐOó3Ð /ñ +;ØÐGó+Ð 'ñ /?ØÐKó/Ð +ô
(Ð'ô (ñ " -Ó0€ô8Ð"5ô 8ñ 0Ð0KÐVcÔdÐ ô6Ð2ô 6ñ -Ð-EÐPjÔkÐ ô6Ð2ô 6ñ -Ð-EÐPjÔkÐ ôAÐ 3ô Añ .ØØ5Ø.ôÐ ôDÐ-@ô Dñ *;Ø*Ð5Nô*Ð &ô
?Ð(;ô ?ñ %6Ð6WÐbvÔ$wÐ !ôAÐ*=ô Añ '8Ø'Ð2Hô'Ð #ô
<Ð%8ô <ñ "3Ð3QÐ\mÔ!nÐ ôEÐ-@ô Eñ *;Ø*Ð5Oô*Ð &ô
AÐ*=ô Añ '8Ø'Ð2Hô'Ð #ô
9Ð!4ô 9ñ /Ð/IÐTmÔnÐ ô=Ð$7ô =ñ !2Ø!Ð,Zô!Ñ r¿   