
    sgӄ                        d dl mZmZ  G d de      Z G d de      Z G d de      Z G d	 d
e      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d! d"e      Z G d# d$e      Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Zd1Zd1Zd1Zd1Zd1Zd1Zd1Z d1Z!d1Z"d1Z#d1Z$d1Z%d1Z&d1Z' G d2 d3e      Z( G d4 d5e      Z) G d6 d7e      Z* G d8 d9e      Z+ G d: d;e      Z, G d< d=e      Z- G d> d?e      Z. G d@ dAe      Z/ G dB dCe      Z0 G dD dEe      Z1 G dF dGe      Z2 G dH dIe      Z3 G dJ dKe      Z4 G dL dMe      Z5 G dN dOe      Z6 G dP dQe      Z7 G dR dSe      Z8 G dT dUe      Z9 G dV dWe      Z: G dX dYe      Z; G dZ d[e      Z< G d\ d]e      Z= G d^ d_e      Z> G d` dae      Z? G db dce      Z@ G dd dee      ZA G df dge      ZB G dh die      ZC G dj dke      ZD G dl dme      ZE G dn doe      ZF G dp dqe      ZG G dr dse      ZH G dt due      ZI G dv dwe      ZJ G dx dye      ZK G dz d{e      ZL G d| d}e      ZM G d~ de      ZN G d de      ZO G d de      ZP G d de      ZQ G d de      ZR G d de      ZS G d de      ZT G d de      ZU G d de      ZV G d de      ZW G d de      ZX G d de      ZY G d de      ZZ G d de      Z[ G d de      Z\ G d de      Z] G d de      Z^ G d de      Z_ G d de      Z` G d de      Za G d de      Zb G d de      Zc G d de      Zd G d de      Ze G d de      Zf G d de      Zg G d de      Zh G d de      Zi G d de      Zj G d de      Zk G d de      Zl G d de      Zm G d de      Zn G d de      Zo G d de      Zp G dĄ de      Zq G dƄ de      Zr G dȄ de      Zs G dʄ de      Zt G d̄ de      Zu G d΄ de      Zv G dЄ de      Zw G d҄ de      Zx G dԄ de      Zy G dք de      Zz G d؄ de      Z{ G dڄ de      Z| G d܄ de      Z} G dބ de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  de      Z G d de      Z G d de      Z G d de      Z G d d	e      Z G d
 de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  d!e      Z G d" d#e      Z G d$ d%e      Z G d& d'e      Z G d( d)e      Z G d* d+e      Z G d, d-e      Z G d. d/e      Z G d0 d1e      Z G d2 d3e      Z G d4 d5e      Z G d6 d7e      Z G d8 d9e      Z G d: d;e      Z G d< d=e      Z G d> d?e      Z G d@ dAe      Z G dB dCe      Z G dD dEe      Z G dF dGe      Z G dH dIe      Z G dJ dKe      Z G dL dMe      Z G dN dOe      Z G dP dQe      Z G dR dSe      Z G dT dUe      Z G dV dWe      Z G dX dYe      Z G dZ d[e      Z G d\ d]e      Z G d^ d_e      Z G d` dae      Z G db dce      Z G dd dee      Z G df dge      Z G dh die      Z G dj dke      Z G dl dme      Z G dn doe      Z G dp dqe      Z G dr dse      Z G dt due      Z G dv dwe      Z G dx dye      Z G dz d{e      Z G d| d}e      Z G d~ de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zy1(     )DummyObjectrequires_backendsc                       e Zd ZdgZd Zy)!FlaxForcedBOSTokenLogitsProcessorflaxc                     t        | dg       y Nr   r   selfargskwargss      X/var/www/html/venv/lib/python3.12/site-packages/transformers/utils/dummy_flax_objects.py__init__z*FlaxForcedBOSTokenLogitsProcessor.__init__       $)    N__name__
__module____qualname__	_backendsr    r   r   r   r          I*r   r   )	metaclassc                       e Zd ZdgZd Zy)!FlaxForcedEOSTokenLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxForcedEOSTokenLogitsProcessor.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)FlaxForceTokensLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxForceTokensLogitsProcessor.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)FlaxGenerationMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGenerationMixin.__init__   r   r   Nr   r   r   r   r"   r"      r   r   r"   c                       e Zd ZdgZd Zy)FlaxLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxLogitsProcessor.__init__$   r   r   Nr   r   r   r   r%   r%   !   r   r   r%   c                       e Zd ZdgZd Zy)FlaxLogitsProcessorListr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxLogitsProcessorList.__init__+   r   r   Nr   r   r   r   r(   r(   (   r   r   r(   c                       e Zd ZdgZd Zy)FlaxLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxLogitsWarper.__init__2   r   r   Nr   r   r   r   r+   r+   /   r   r   r+   c                       e Zd ZdgZd Zy)FlaxMinLengthLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxMinLengthLogitsProcessor.__init__9   r   r   Nr   r   r   r   r.   r.   6   r   r   r.   c                       e Zd ZdgZd Zy)(FlaxSuppressTokensAtBeginLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z1FlaxSuppressTokensAtBeginLogitsProcessor.__init__@   r   r   Nr   r   r   r   r1   r1   =   r   r   r1   c                       e Zd ZdgZd Zy)!FlaxSuppressTokensLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxSuppressTokensLogitsProcessor.__init__G   r   r   Nr   r   r   r   r4   r4   D   r   r   r4   c                       e Zd ZdgZd Zy)FlaxTemperatureLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxTemperatureLogitsWarper.__init__N   r   r   Nr   r   r   r   r7   r7   K   r   r   r7   c                       e Zd ZdgZd Zy)FlaxTopKLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxTopKLogitsWarper.__init__U   r   r   Nr   r   r   r   r:   r:   R   r   r   r:   c                       e Zd ZdgZd Zy)FlaxTopPLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxTopPLogitsWarper.__init__\   r   r   Nr   r   r   r   r=   r=   Y   r   r   r=   c                       e Zd ZdgZd Zy)#FlaxWhisperTimeStampLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxWhisperTimeStampLogitsProcessor.__init__c   r   r   Nr   r   r   r   r@   r@   `   r   r   r@   c                       e Zd ZdgZd Zy)FlaxPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxPreTrainedModel.__init__j   r   r   Nr   r   r   r   rC   rC   g   r   r   rC   c                       e Zd ZdgZd Zy)FlaxAlbertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxAlbertForMaskedLM.__init__q   r   r   Nr   r   r   r   rF   rF   n   r   r   rF   c                       e Zd ZdgZd Zy)FlaxAlbertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxAlbertForMultipleChoice.__init__x   r   r   Nr   r   r   r   rI   rI   u   r   r   rI   c                       e Zd ZdgZd Zy)FlaxAlbertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxAlbertForPreTraining.__init__   r   r   Nr   r   r   r   rL   rL   |   r   r   rL   c                       e Zd ZdgZd Zy)FlaxAlbertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxAlbertForQuestionAnswering.__init__   r   r   Nr   r   r   r   rO   rO      r   r   rO   c                       e Zd ZdgZd Zy)#FlaxAlbertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxAlbertForSequenceClassification.__init__   r   r   Nr   r   r   r   rR   rR      r   r   rR   c                       e Zd ZdgZd Zy) FlaxAlbertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxAlbertForTokenClassification.__init__   r   r   Nr   r   r   r   rU   rU      r   r   rU   c                       e Zd ZdgZd Zy)FlaxAlbertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxAlbertModel.__init__   r   r   Nr   r   r   r   rX   rX      r   r   rX   c                       e Zd ZdgZd Zy)FlaxAlbertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxAlbertPreTrainedModel.__init__   r   r   Nr   r   r   r   r[   r[      r   r   r[   Nc                       e Zd ZdgZd Zy)FlaxAutoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxAutoModel.__init__   r   r   Nr   r   r   r   r^   r^      r   r   r^   c                       e Zd ZdgZd Zy)FlaxAutoModelForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxAutoModelForCausalLM.__init__   r   r   Nr   r   r   r   ra   ra      r   r   ra   c                       e Zd ZdgZd Zy)#FlaxAutoModelForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxAutoModelForImageClassification.__init__   r   r   Nr   r   r   r   rd   rd      r   r   rd   c                       e Zd ZdgZd Zy)FlaxAutoModelForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxAutoModelForMaskedLM.__init__   r   r   Nr   r   r   r   rg   rg      r   r   rg   c                       e Zd ZdgZd Zy)FlaxAutoModelForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxAutoModelForMultipleChoice.__init__   r   r   Nr   r   r   r   rj   rj      r   r   rj   c                       e Zd ZdgZd Zy)&FlaxAutoModelForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z/FlaxAutoModelForNextSentencePrediction.__init__   r   r   Nr   r   r   r   rm   rm      r   r   rm   c                       e Zd ZdgZd Zy)FlaxAutoModelForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxAutoModelForPreTraining.__init__   r   r   Nr   r   r   r   rp   rp      r   r   rp   c                       e Zd ZdgZd Zy)!FlaxAutoModelForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxAutoModelForQuestionAnswering.__init__  r   r   Nr   r   r   r   rs   rs     r   r   rs   c                       e Zd ZdgZd Zy)FlaxAutoModelForSeq2SeqLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxAutoModelForSeq2SeqLM.__init__  r   r   Nr   r   r   r   rv   rv     r   r   rv   c                       e Zd ZdgZd Zy)&FlaxAutoModelForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z/FlaxAutoModelForSequenceClassification.__init__  r   r   Nr   r   r   r   ry   ry     r   r   ry   c                       e Zd ZdgZd Zy)FlaxAutoModelForSpeechSeq2Seqr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxAutoModelForSpeechSeq2Seq.__init__  r   r   Nr   r   r   r   r|   r|     r   r   r|   c                       e Zd ZdgZd Zy)#FlaxAutoModelForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxAutoModelForTokenClassification.__init__   r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxAutoModelForVision2Seqr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxAutoModelForVision2Seq.__init__'  r   r   Nr   r   r   r   r   r   $  r   r   r   c                       e Zd ZdgZd Zy)FlaxBartDecoderPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxBartDecoderPreTrainedModel.__init__.  r   r   Nr   r   r   r   r   r   +  r   r   r   c                       e Zd ZdgZd Zy)FlaxBartForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBartForCausalLM.__init__5  r   r   Nr   r   r   r   r   r   2  r   r   r   c                       e Zd ZdgZd Zy) FlaxBartForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxBartForConditionalGeneration.__init__<  r   r   Nr   r   r   r   r   r   9  r   r   r   c                       e Zd ZdgZd Zy)FlaxBartForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxBartForQuestionAnswering.__init__C  r   r   Nr   r   r   r   r   r   @  r   r   r   c                       e Zd ZdgZd Zy)!FlaxBartForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxBartForSequenceClassification.__init__J  r   r   Nr   r   r   r   r   r   G  r   r   r   c                       e Zd ZdgZd Zy)FlaxBartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBartModel.__init__Q  r   r   Nr   r   r   r   r   r   N  r   r   r   c                       e Zd ZdgZd Zy)FlaxBartPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxBartPreTrainedModel.__init__X  r   r   Nr   r   r   r   r   r   U  r   r   r   c                       e Zd ZdgZd Zy)FlaxBeitForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxBeitForImageClassification.__init___  r   r   Nr   r   r   r   r   r   \  r   r   r   c                       e Zd ZdgZd Zy)FlaxBeitForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxBeitForMaskedImageModeling.__init__f  r   r   Nr   r   r   r   r   r   c  r   r   r   c                       e Zd ZdgZd Zy)FlaxBeitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBeitModel.__init__m  r   r   Nr   r   r   r   r   r   j  r   r   r   c                       e Zd ZdgZd Zy)FlaxBeitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxBeitPreTrainedModel.__init__t  r   r   Nr   r   r   r   r   r   q  r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBertForCausalLM.__init__{  r   r   Nr   r   r   r   r   r   x  r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBertForMaskedLM.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)!FlaxBertForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxBertForNextSentencePrediction.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBertForPreTraining.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)!FlaxBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxBertForTokenClassification.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBertModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBigBirdForCausalLM.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBigBirdForMaskedLM.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxBigBirdForMultipleChoice.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxBigBirdForPreTraining.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxBigBirdForQuestionAnswering.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)$FlaxBigBirdForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-FlaxBigBirdForSequenceClassification.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)!FlaxBigBirdForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxBigBirdForTokenClassification.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBigBirdModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBigBirdPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxBigBirdPreTrainedModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)&FlaxBlenderbotForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/FlaxBlenderbotForConditionalGeneration.__init__   r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBlenderbotModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBlenderbotModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBlenderbotPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxBlenderbotPreTrainedModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)+FlaxBlenderbotSmallForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z4FlaxBlenderbotSmallForConditionalGeneration.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxBlenderbotSmallModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxBlenderbotSmallModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)"FlaxBlenderbotSmallPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxBlenderbotSmallPreTrainedModel.__init__#  r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)FlaxBloomForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBloomForCausalLM.__init__*  r   r   Nr   r   r   r   r   r   '  r   r   r   c                       e Zd ZdgZd Zy)FlaxBloomModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxBloomModel.__init__1  r   r   Nr   r   r   r   r   r   .  r   r   r   c                       e Zd ZdgZd Zy)FlaxBloomPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxBloomPreTrainedModel.__init__8  r   r   Nr   r   r   r   r   r   5  r   r   r   c                       e Zd ZdgZd Zy)FlaxCLIPModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxCLIPModel.__init__?  r   r   Nr   r   r   r   r   r   <  r   r   r   c                       e Zd ZdgZd Zy)FlaxCLIPPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxCLIPPreTrainedModel.__init__F  r   r   Nr   r   r   r   r   r   C  r   r   r   c                       e Zd ZdgZd Zy)FlaxCLIPTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxCLIPTextModel.__init__M  r   r   Nr   r   r   r   r   r   J  r   r   r   c                       e Zd ZdgZd Zy)FlaxCLIPTextModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxCLIPTextModelWithProjection.__init__T  r   r   Nr   r   r   r   r  r  Q  r   r   r  c                       e Zd ZdgZd Zy)FlaxCLIPTextPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxCLIPTextPreTrainedModel.__init__[  r   r   Nr   r   r   r   r  r  X  r   r   r  c                       e Zd ZdgZd Zy)FlaxCLIPVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxCLIPVisionModel.__init__b  r   r   Nr   r   r   r   r	  r	  _  r   r   r	  c                       e Zd ZdgZd Zy)FlaxCLIPVisionPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxCLIPVisionPreTrainedModel.__init__i  r   r   Nr   r   r   r   r  r  f  r   r   r  c                       e Zd ZdgZd Zy) FlaxDinov2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxDinov2ForImageClassification.__init__p  r   r   Nr   r   r   r   r  r  m  r   r   r  c                       e Zd ZdgZd Zy)FlaxDinov2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxDinov2Model.__init__w  r   r   Nr   r   r   r   r  r  t  r   r   r  c                       e Zd ZdgZd Zy)FlaxDinov2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxDinov2PreTrainedModel.__init__~  r   r   Nr   r   r   r   r  r  {  r   r   r  c                       e Zd ZdgZd Zy)FlaxDistilBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxDistilBertForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxDistilBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxDistilBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"FlaxDistilBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxDistilBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)'FlaxDistilBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0FlaxDistilBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r!  r!    r   r   r!  c                       e Zd ZdgZd Zy)$FlaxDistilBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-FlaxDistilBertForTokenClassification.__init__  r   r   Nr   r   r   r   r$  r$    r   r   r$  c                       e Zd ZdgZd Zy)FlaxDistilBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxDistilBertModel.__init__  r   r   Nr   r   r   r   r'  r'    r   r   r'  c                       e Zd ZdgZd Zy)FlaxDistilBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxDistilBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r*  r*    r   r   r*  c                       e Zd ZdgZd Zy)FlaxElectraForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxElectraForCausalLM.__init__  r   r   Nr   r   r   r   r-  r-    r   r   r-  c                       e Zd ZdgZd Zy)FlaxElectraForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxElectraForMaskedLM.__init__  r   r   Nr   r   r   r   r0  r0    r   r   r0  c                       e Zd ZdgZd Zy)FlaxElectraForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxElectraForMultipleChoice.__init__  r   r   Nr   r   r   r   r3  r3    r   r   r3  c                       e Zd ZdgZd Zy)FlaxElectraForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxElectraForPreTraining.__init__  r   r   Nr   r   r   r   r6  r6    r   r   r6  c                       e Zd ZdgZd Zy)FlaxElectraForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxElectraForQuestionAnswering.__init__  r   r   Nr   r   r   r   r9  r9    r   r   r9  c                       e Zd ZdgZd Zy)$FlaxElectraForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-FlaxElectraForSequenceClassification.__init__  r   r   Nr   r   r   r   r<  r<    r   r   r<  c                       e Zd ZdgZd Zy)!FlaxElectraForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxElectraForTokenClassification.__init__  r   r   Nr   r   r   r   r?  r?    r   r   r?  c                       e Zd ZdgZd Zy)FlaxElectraModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxElectraModel.__init__  r   r   Nr   r   r   r   rB  rB    r   r   rB  c                       e Zd ZdgZd Zy)FlaxElectraPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxElectraPreTrainedModel.__init__  r   r   Nr   r   r   r   rE  rE    r   r   rE  c                       e Zd ZdgZd Zy)FlaxEncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxEncoderDecoderModel.__init__  r   r   Nr   r   r   r   rH  rH    r   r   rH  c                       e Zd ZdgZd Zy)FlaxGemmaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGemmaForCausalLM.__init__  r   r   Nr   r   r   r   rK  rK    r   r   rK  c                       e Zd ZdgZd Zy)FlaxGemmaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGemmaModel.__init__  r   r   Nr   r   r   r   rN  rN     r   r   rN  c                       e Zd ZdgZd Zy)FlaxGemmaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxGemmaPreTrainedModel.__init__
  r   r   Nr   r   r   r   rQ  rQ    r   r   rQ  c                       e Zd ZdgZd Zy)FlaxGPT2LMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPT2LMHeadModel.__init__  r   r   Nr   r   r   r   rT  rT    r   r   rT  c                       e Zd ZdgZd Zy)FlaxGPT2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPT2Model.__init__  r   r   Nr   r   r   r   rW  rW    r   r   rW  c                       e Zd ZdgZd Zy)FlaxGPT2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxGPT2PreTrainedModel.__init__  r   r   Nr   r   r   r   rZ  rZ    r   r   rZ  c                       e Zd ZdgZd Zy)FlaxGPTNeoForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPTNeoForCausalLM.__init__&  r   r   Nr   r   r   r   r]  r]  #  r   r   r]  c                       e Zd ZdgZd Zy)FlaxGPTNeoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPTNeoModel.__init__-  r   r   Nr   r   r   r   r`  r`  *  r   r   r`  c                       e Zd ZdgZd Zy)FlaxGPTNeoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxGPTNeoPreTrainedModel.__init__4  r   r   Nr   r   r   r   rc  rc  1  r   r   rc  c                       e Zd ZdgZd Zy)FlaxGPTJForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPTJForCausalLM.__init__;  r   r   Nr   r   r   r   rf  rf  8  r   r   rf  c                       e Zd ZdgZd Zy)FlaxGPTJModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxGPTJModel.__init__B  r   r   Nr   r   r   r   ri  ri  ?  r   r   ri  c                       e Zd ZdgZd Zy)FlaxGPTJPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxGPTJPreTrainedModel.__init__I  r   r   Nr   r   r   r   rl  rl  F  r   r   rl  c                       e Zd ZdgZd Zy)FlaxLlamaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxLlamaForCausalLM.__init__P  r   r   Nr   r   r   r   ro  ro  M  r   r   ro  c                       e Zd ZdgZd Zy)FlaxLlamaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxLlamaModel.__init__W  r   r   Nr   r   r   r   rr  rr  T  r   r   rr  c                       e Zd ZdgZd Zy)FlaxLlamaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxLlamaPreTrainedModel.__init__^  r   r   Nr   r   r   r   ru  ru  [  r   r   ru  c                       e Zd ZdgZd Zy)"FlaxLongT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxLongT5ForConditionalGeneration.__init__e  r   r   Nr   r   r   r   rx  rx  b  r   r   rx  c                       e Zd ZdgZd Zy)FlaxLongT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxLongT5Model.__init__l  r   r   Nr   r   r   r   r{  r{  i  r   r   r{  c                       e Zd ZdgZd Zy)FlaxLongT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxLongT5PreTrainedModel.__init__s  r   r   Nr   r   r   r   r~  r~  p  r   r   r~  c                       e Zd ZdgZd Zy)FlaxMarianModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMarianModel.__init__z  r   r   Nr   r   r   r   r  r  w  r   r   r  c                       e Zd ZdgZd Zy)FlaxMarianMTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMarianMTModel.__init__  r   r   Nr   r   r   r   r  r  ~  r   r   r  c                       e Zd ZdgZd Zy)FlaxMarianPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxMarianPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!FlaxMBartForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxMBartForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMBartForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxMBartForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"FlaxMBartForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxMBartForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMBartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMBartModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMBartPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FlaxMBartPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMistralForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMistralForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMistralModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMistralModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMistralPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxMistralPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMT5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMT5EncoderModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxMT5ForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxMT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxMT5Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxOPTForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxOPTForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxOPTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxOPTModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxOPTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxOPTPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)#FlaxPegasusForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxPegasusForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxPegasusModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxPegasusModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxPegasusPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxPegasusPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) FlaxRegNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxRegNetForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRegNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxRegNetModel.__init__  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)FlaxRegNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxRegNetPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) FlaxResNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxResNetForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxResNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxResNetModel.__init__"  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxResNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxResNetPreTrainedModel.__init__)  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxRobertaForCausalLM.__init__0  r   r   Nr   r   r   r   r  r  -  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxRobertaForMaskedLM.__init__7  r   r   Nr   r   r   r   r  r  4  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxRobertaForMultipleChoice.__init__>  r   r   Nr   r   r   r   r  r  ;  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxRobertaForQuestionAnswering.__init__E  r   r   Nr   r   r   r   r  r  B  r   r   r  c                       e Zd ZdgZd Zy)$FlaxRobertaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-FlaxRobertaForSequenceClassification.__init__L  r   r   Nr   r   r   r   r  r  I  r   r   r  c                       e Zd ZdgZd Zy)!FlaxRobertaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxRobertaForTokenClassification.__init__S  r   r   Nr   r   r   r   r  r  P  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxRobertaModel.__init__Z  r   r   Nr   r   r   r   r  r  W  r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxRobertaPreTrainedModel.__init__a  r   r   Nr   r   r   r   r  r  ^  r   r   r  c                       e Zd ZdgZd Zy)"FlaxRobertaPreLayerNormForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxRobertaPreLayerNormForCausalLM.__init__h  r   r   Nr   r   r   r   r  r  e  r   r   r  c                       e Zd ZdgZd Zy)"FlaxRobertaPreLayerNormForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxRobertaPreLayerNormForMaskedLM.__init__o  r   r   Nr   r   r   r   r  r  l  r   r   r  c                       e Zd ZdgZd Zy)(FlaxRobertaPreLayerNormForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z1FlaxRobertaPreLayerNormForMultipleChoice.__init__v  r   r   Nr   r   r   r   r  r  s  r   r   r  c                       e Zd ZdgZd Zy)+FlaxRobertaPreLayerNormForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z4FlaxRobertaPreLayerNormForQuestionAnswering.__init__}  r   r   Nr   r   r   r   r  r  z  r   r   r  c                       e Zd ZdgZd Zy)0FlaxRobertaPreLayerNormForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z9FlaxRobertaPreLayerNormForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)-FlaxRobertaPreLayerNormForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z6FlaxRobertaPreLayerNormForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRobertaPreLayerNormModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaxRobertaPreLayerNormModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)&FlaxRobertaPreLayerNormPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z/FlaxRobertaPreLayerNormPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRoFormerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxRoFormerForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRoFormerForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxRoFormerForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) FlaxRoFormerForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z)FlaxRoFormerForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)%FlaxRoFormerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.FlaxRoFormerForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"FlaxRoFormerForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxRoFormerForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRoFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxRoFormerModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxRoFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxRoFormerPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxSpeechEncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxSpeechEncoderDecoderModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxT5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxT5EncoderModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxT5ForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxT5Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaxT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxT5PreTrainedModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)FlaxVisionEncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxVisionEncoderDecoderModel.__init__  r   r   Nr   r   r   r   r#  r#    r   r   r#  c                       e Zd ZdgZd Zy)FlaxVisionTextDualEncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaxVisionTextDualEncoderModel.__init__  r   r   Nr   r   r   r   r&  r&    r   r   r&  c                       e Zd ZdgZd Zy)FlaxViTForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxViTForImageClassification.__init__  r   r   Nr   r   r   r   r)  r)    r   r   r)  c                       e Zd ZdgZd Zy)FlaxViTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxViTModel.__init__	  r   r   Nr   r   r   r   r,  r,    r   r   r,  c                       e Zd ZdgZd Zy)FlaxViTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxViTPreTrainedModel.__init__  r   r   Nr   r   r   r   r/  r/    r   r   r/  c                       e Zd ZdgZd Zy)FlaxWav2Vec2ForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxWav2Vec2ForCTC.__init__  r   r   Nr   r   r   r   r2  r2    r   r   r2  c                       e Zd ZdgZd Zy)FlaxWav2Vec2ForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxWav2Vec2ForPreTraining.__init__  r   r   Nr   r   r   r   r5  r5    r   r   r5  c                       e Zd ZdgZd Zy)FlaxWav2Vec2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxWav2Vec2Model.__init__%  r   r   Nr   r   r   r   r8  r8  "  r   r   r8  c                       e Zd ZdgZd Zy)FlaxWav2Vec2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$FlaxWav2Vec2PreTrainedModel.__init__,  r   r   Nr   r   r   r   r;  r;  )  r   r   r;  c                       e Zd ZdgZd Zy)!FlaxWhisperForAudioClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaxWhisperForAudioClassification.__init__3  r   r   Nr   r   r   r   r>  r>  0  r   r   r>  c                       e Zd ZdgZd Zy)#FlaxWhisperForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z,FlaxWhisperForConditionalGeneration.__init__:  r   r   Nr   r   r   r   rA  rA  7  r   r   rA  c                       e Zd ZdgZd Zy)FlaxWhisperModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxWhisperModel.__init__A  r   r   Nr   r   r   r   rD  rD  >  r   r   rD  c                       e Zd ZdgZd Zy)FlaxWhisperPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FlaxWhisperPreTrainedModel.__init__H  r   r   Nr   r   r   r   rG  rG  E  r   r   rG  c                       e Zd ZdgZd Zy)FlaxXGLMForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxXGLMForCausalLM.__init__O  r   r   Nr   r   r   r   rJ  rJ  L  r   r   rJ  c                       e Zd ZdgZd Zy)FlaxXGLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxXGLMModel.__init__V  r   r   Nr   r   r   r   rM  rM  S  r   r   rM  c                       e Zd ZdgZd Zy)FlaxXGLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaxXGLMPreTrainedModel.__init__]  r   r   Nr   r   r   r   rP  rP  Z  r   r   rP  c                       e Zd ZdgZd Zy)FlaxXLMRobertaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxXLMRobertaForCausalLM.__init__d  r   r   Nr   r   r   r   rS  rS  a  r   r   rS  c                       e Zd ZdgZd Zy)FlaxXLMRobertaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaxXLMRobertaForMaskedLM.__init__k  r   r   Nr   r   r   r   rV  rV  h  r   r   rV  c                       e Zd ZdgZd Zy)FlaxXLMRobertaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z(FlaxXLMRobertaForMultipleChoice.__init__r  r   r   Nr   r   r   r   rY  rY  o  r   r   rY  c                       e Zd ZdgZd Zy)"FlaxXLMRobertaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaxXLMRobertaForQuestionAnswering.__init__y  r   r   Nr   r   r   r   r\  r\  v  r   r   r\  c                       e Zd ZdgZd Zy)'FlaxXLMRobertaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0FlaxXLMRobertaForSequenceClassification.__init__  r   r   Nr   r   r   r   r_  r_  }  r   r   r_  c                       e Zd ZdgZd Zy)$FlaxXLMRobertaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-FlaxXLMRobertaForTokenClassification.__init__  r   r   Nr   r   r   r   rb  rb    r   r   rb  c                       e Zd ZdgZd Zy)FlaxXLMRobertaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaxXLMRobertaModel.__init__  r   r   Nr   r   r   r   re  re    r   r   re  c                       e Zd ZdgZd Zy)FlaxXLMRobertaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&FlaxXLMRobertaPreTrainedModel.__init__  r   r   Nr   r   r   r   rh  rh    r   r   rh  )utilsr   r   r   r   r   r"   r%   r(   r+   r.   r1   r4   r7   r:   r=   r@   rC   rF   rI   rL   rO   rR   rU   rX   r[   +FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING FLAX_MODEL_FOR_CAUSAL_LM_MAPPING+FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING FLAX_MODEL_FOR_MASKED_LM_MAPPING&FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPING/FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING"FLAX_MODEL_FOR_PRETRAINING_MAPPING)FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPING+FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING.FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING'FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING+FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING#FLAX_MODEL_FOR_VISION_2_SEQ_MAPPINGFLAX_MODEL_MAPPINGr^   ra   rd   rg   rj   rm   rp   rs   rv   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   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'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  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  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&  r)  r,  r/  r2  r5  r8  r;  r>  rA  rD  rG  rJ  rM  rP  rS  rV  rY  r\  r_  rb  re  rh  r   r   r   <module>ry     s	   2*+ **+ **{ **K **K ** ** **[ ** **+ **K **[ **[ **K **K **k **K ** **{ **K ** ** **+ * /3 + $(   /3 + $(   *. & 37 / &* " -1 ) /3 + 26 . +/ ' /3 + '+ #  *k ** **K ** **{ **{ **K **+ **+ **{ **k **K **; **{ **K ** **[ **+ **k ** **{ **{ **k ** **K **K **+ **+ **{ **[ **+ **{ **k ** **{ **{ **[ **+ ** **[ **+ ** **; **{ **K **k **K ** **; **[ **{ ** **k ** **+ ** **K **K **k ** ** **+ **+ ** **; ** **[ **K **k **{ **{ **[ **+ ** **[ **+ ** **; ** **[ **{ ** **K **k ** **k ** **+ **K **k ** **[ **{ ** **; ** **+ ** **+ **+ **+ **k **; **{ ** **{ ** **; **K ** **[ **; **[ **{ **K ** **; ** ** **+ ** ** **+ **{ **{ **[ ** **[ **+ ** **; **; **; ** **K ** **k **[ **{ ** **k ** **k **; **+ **K **k **; **{ **K **k **k **{ **k **[ **{ **; **; **+ **K **+ **K ** **; **K **k ** **+ **+ ** **; ** **[ **K **k *r   