§
     `ƒi]=  ã                   óÜ  — 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?¬¦  «        Z6g d@¢Z7dAS )Bz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dS )ÚFlaxAutoModelN)Ú__name__Ú
__module__Ú__qualname__ÚFLAX_MODEL_MAPPINGÚ_model_mapping© ó    ú/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/transformers/models/auto/modeling_flax_auto.pyr¸   r¸     s   € € € € € Ø'€N€N€Nr¿   r¸   c                   ó   — e Zd ZeZdS )ÚFlaxAutoModelForPreTrainingN)r¹   rº   r»   Ú"FLAX_MODEL_FOR_PRETRAINING_MAPPINGr½   r¾   r¿   rÀ   rÂ   rÂ      s   € € € € € Ø7€N€N€Nr¿   rÂ   Úpretraining)Úhead_docc                   ó   — e Zd ZeZdS )ÚFlaxAutoModelForCausalLMN)r¹   rº   r»   Ú FLAX_MODEL_FOR_CAUSAL_LM_MAPPINGr½   r¾   r¿   rÀ   rÇ   rÇ   '  ó   € € € € € Ø5€N€N€Nr¿   rÇ   zcausal language modelingc                   ó   — e Zd ZeZdS )Ú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dS )ÚFlaxAutoModelForSeq2SeqLMN)r¹   rº   r»   Ú+FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGr½   r¾   r¿   rÀ   rÎ   rÎ   5  ó   € € € € € Ø@€N€N€Nr¿   rÎ   z&sequence-to-sequence language modelingzgoogle-t5/t5-base)rÅ   Úcheckpoint_for_examplec                   ó   — e Zd ZeZdS )Ú&FlaxAutoModelForSequenceClassificationN)r¹   rº   r»   Ú.FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGr½   r¾   r¿   rÀ   rÓ   rÓ   @  s   € € € € € ØC€N€N€Nr¿   rÓ   zsequence classificationc                   ó   — e Zd ZeZdS )Ú!FlaxAutoModelForQuestionAnsweringN)r¹   rº   r»   Ú)FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPINGr½   r¾   r¿   rÀ   rÖ   rÖ   I  s   € € € € € Ø>€N€N€Nr¿   rÖ   zquestion answeringc                   ó   — e Zd ZeZdS )Ú#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dS )ÚFlaxAutoModelForMultipleChoiceN)r¹   rº   r»   Ú&FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPINGr½   r¾   r¿   rÀ   rÜ   rÜ   Y  s   € € € € € Ø;€N€N€Nr¿   rÜ   zmultiple choicec                   ó   — e Zd ZeZdS )Ú&FlaxAutoModelForNextSentencePredictionN)r¹   rº   r»   Ú/FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGr½   r¾   r¿   rÀ   rß   rß   `  s   € € € € € ØD€N€N€Nr¿   rß   znext sentence predictionc                   ó   — e Zd ZeZdS )Ú#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dS )ÚFlaxAutoModelForVision2SeqN)r¹   rº   r»   Ú#FLAX_MODEL_FOR_VISION_2_SEQ_MAPPINGr½   r¾   r¿   rÀ   rå   rå   r  s   € € € € € Ø8€N€N€Nr¿   rå   zvision-to-text modelingc                   ó   — e Zd ZeZdS )ÚFlaxAutoModelForSpeechSeq2SeqN)r¹   rº   r»   Ú'FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGr½   r¾   r¿   rÀ   rè   rè   y  s   € € € € € Ø<€N€N€Nr¿   rè   z,sequence-to-sequence speech-to-text modeling)Ú+FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGrÈ   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å   N)8Ú__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é   rê   r¸   rÂ   rÇ   rË   rÎ   rÓ   rÖ   rÙ   rÜ   rß   râ   rå   rè   Ú__all__r¾   r¿   rÀ   ú<module>r     s‡  ðð Ð à #Ð #Ð #Ð #Ð #Ð #à Ð Ð Ð Ð Ð Ø RÐ RÐ RÐ RÐ RÐ RÐ RÐ RÐ RÐ RØ 4Ð 4Ð 4Ð 4Ð 4Ð 4ð 
ˆÔ	˜HÑ	%Ô	%€ð '˜;ð'ð 'ð 'ñ)ô )Ð ðV ,7¨;ðð ð ñ,ô ,Ð (ð* *5¨ðð ð ñ*ô *Ð &ð" 5@°Kðð ð ñ5ô 5Ð 1ð  5@°Kðð ð ñ	5ô 	5Ð 1ð -8¨KàCðñ-ô -Ð )ð *5¨ðð ð ñ*ô *Ð &ð. 8C°{ðð ð ñ8ô 8Ð 4ð" 3>°+ðð ð ñ3ô 3Ð /ð" 5@°Kðð ð ñ5ô 5Ð 1ð 0;¨{ðð ð ñ0ô 0Ð ,ð 9D¸à5ðñ9ô 9Ð 5ð 1<°àCØ:ðñ1ô 1Ð -ð 5@°Kà8ðñ5ô 5Ð 1ð &Ð%Ð&:Ð<TÑUÔUÐ Ø%5Ð%5Ð6JÐLtÑ%uÔ%uÐ "Ø#3Ð#3Ð4HÐJpÑ#qÔ#qÐ  Ø.>Ð.>ØÐKñ/ô /Ð +ð /?Ð.>ØÐKñ/ô /Ð +ð '7Ð&6Ð7KÐMvÑ&wÔ&wÐ #Ø#3Ð#3Ð4HÐJpÑ#qÔ#qÐ  Ø1AÐ1AØÐNñ2ô 2Ð .ð -=Ð,<ØÐIñ-ô -Ð )ð /?Ð.>ØÐKñ/ô /Ð +ð *:Ð)9ØÐFñ*ô *Ð &ð 3CÐ2BØÐOñ3ô 3Ð /ð +;Ð*:ØÐGñ+ô +Ð 'ð /?Ð.>ØÐKñ/ô /Ð +ð
(ð (ð (ð (ð (Ð'ñ (ô (ð (ð "Ð! -Ñ0Ô0€ð8ð 8ð 8ð 8ð 8Ð"5ñ 8ô 8ð 8ð 0Ð/Ð0KÐVcÐdÑdÔdÐ ð6ð 6ð 6ð 6ð 6Ð2ñ 6ô 6ð 6ð -Ð,Ð-EÐPjÐkÑkÔkÐ ð6ð 6ð 6ð 6ð 6Ð2ñ 6ô 6ð 6ð -Ð,Ð-EÐPjÐkÑkÔkÐ ðAð Að Að Að AÐ 3ñ Aô Að Að .Ð-ØØ5Ø.ðñ ô Ð ðDð Dð Dð Dð DÐ-@ñ Dô Dð Dð *;Ð):Ø*Ð5Nð*ñ *ô *Ð &ð
?ð ?ð ?ð ?ð ?Ð(;ñ ?ô ?ð ?ð %6Ð$5Ð6WÐbvÐ$wÑ$wÔ$wÐ !ðAð Að Að Að AÐ*=ñ Aô Að Að '8Ð&7Ø'Ð2Hð'ñ 'ô 'Ð #ð
<ð <ð <ð <ð <Ð%8ñ <ô <ð <ð "3Ð!2Ð3QÐ\mÐ!nÑ!nÔ!nÐ ðEð Eð Eð Eð EÐ-@ñ Eô Eð Eð *;Ð):Ø*Ð5Oð*ñ *ô *Ð &ð
Að Að Að Að AÐ*=ñ Aô Að Að '8Ð&7Ø'Ð2Hð'ñ 'ô 'Ð #ð
9ð 9ð 9ð 9ð 9Ð!4ñ 9ô 9ð 9ð /Ð.Ð/IÐTmÐnÑnÔnÐ ð=ð =ð =ð =ð =Ð$7ñ =ô =ð =ð !2Ð 1Ø!Ð,Zð!ñ !ô !Ð ðð ð €€€r¿   