
    gӄ                        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          Zd1S (     )DummyObjectrequires_backendsc                       e Zd ZdgZd ZdS )!FlaxForcedBOSTokenLogitsProcessorflaxc                 (    t          | dg           d S Nr   r   selfargskwargss      a/var/www/html/ai-engine/env/lib/python3.11/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dS )!FlaxForcedEOSTokenLogitsProcessorr   c                 (    t          | dg           d S 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dS )FlaxForceTokensLogitsProcessorr   c                 (    t          | dg           d S 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dS )FlaxGenerationMixinr   c                 (    t          | dg           d S 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dS )FlaxLogitsProcessorr   c                 (    t          | dg           d S 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dS )FlaxLogitsProcessorListr   c                 (    t          | dg           d S 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dS )FlaxLogitsWarperr   c                 (    t          | dg           d S 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dS )FlaxMinLengthLogitsProcessorr   c                 (    t          | dg           d S 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dS )(FlaxSuppressTokensAtBeginLogitsProcessorr   c                 (    t          | dg           d S 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dS )!FlaxSuppressTokensLogitsProcessorr   c                 (    t          | dg           d S 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dS )FlaxTemperatureLogitsWarperr   c                 (    t          | dg           d S 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dS )FlaxTopKLogitsWarperr   c                 (    t          | dg           d S 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dS )FlaxTopPLogitsWarperr   c                 (    t          | dg           d S 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dS )#FlaxWhisperTimeStampLogitsProcessorr   c                 (    t          | dg           d S 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dS )FlaxPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxAlbertForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxAlbertForMultipleChoicer   c                 (    t          | dg           d S 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dS )FlaxAlbertForPreTrainingr   c                 (    t          | dg           d S 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dS )FlaxAlbertForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )#FlaxAlbertForSequenceClassificationr   c                 (    t          | dg           d S 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dS ) FlaxAlbertForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxAlbertModelr   c                 (    t          | dg           d S 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dS )FlaxAlbertPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxAutoModelr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForCausalLMr   c                 (    t          | dg           d S 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dS )#FlaxAutoModelForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForMultipleChoicer   c                 (    t          | dg           d S 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dS )&FlaxAutoModelForNextSentencePredictionr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForPreTrainingr   c                 (    t          | dg           d S 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dS )!FlaxAutoModelForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForSeq2SeqLMr   c                 (    t          | dg           d S 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dS )&FlaxAutoModelForSequenceClassificationr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForSpeechSeq2Seqr   c                 (    t          | dg           d S 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dS )#FlaxAutoModelForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxAutoModelForVision2Seqr   c                 (    t          | dg           d S 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dS )FlaxBartDecoderPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxBartForCausalLMr   c                 (    t          | dg           d S 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dS ) FlaxBartForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxBartForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )!FlaxBartForSequenceClassificationr   c                 (    t          | dg           d S 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dS )FlaxBartModelr   c                 (    t          | dg           d S 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dS )FlaxBartPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxBeitForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxBeitForMaskedImageModelingr   c                 (    t          | dg           d S 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dS )FlaxBeitModelr   c                 (    t          | dg           d S 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dS )FlaxBeitPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxBertForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxBertForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxBertForMultipleChoicer   c                 (    t          | dg           d S 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dS )!FlaxBertForNextSentencePredictionr   c                 (    t          | dg           d S 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dS )FlaxBertForPreTrainingr   c                 (    t          | dg           d S 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dS )FlaxBertForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )!FlaxBertForSequenceClassificationr   c                 (    t          | dg           d S 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dS )FlaxBertForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxBertModelr   c                 (    t          | dg           d S 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dS )FlaxBertPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxBigBirdForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxBigBirdForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxBigBirdForMultipleChoicer   c                 (    t          | dg           d S 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dS )FlaxBigBirdForPreTrainingr   c                 (    t          | dg           d S 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dS )FlaxBigBirdForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )$FlaxBigBirdForSequenceClassificationr   c                 (    t          | dg           d S 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dS )!FlaxBigBirdForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxBigBirdModelr   c                 (    t          | dg           d S 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dS )FlaxBigBirdPreTrainedModelr   c                 (    t          | dg           d S 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dS )&FlaxBlenderbotForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxBlenderbotModelr   c                 (    t          | dg           d S 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dS )FlaxBlenderbotPreTrainedModelr   c                 (    t          | dg           d S 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dS )+FlaxBlenderbotSmallForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxBlenderbotSmallModelr   c                 (    t          | dg           d S 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dS )"FlaxBlenderbotSmallPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxBloomForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxBloomModelr   c                 (    t          | dg           d S 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dS )FlaxBloomPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPTextModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPTextModelWithProjectionr   c                 (    t          | dg           d S 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dS )FlaxCLIPTextPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPVisionModelr   c                 (    t          | dg           d S 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dS )FlaxCLIPVisionPreTrainedModelr   c                 (    t          | dg           d S 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dS ) FlaxDinov2ForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxDinov2Modelr   c                 (    t          | dg           d S 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dS )FlaxDinov2PreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxDistilBertForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxDistilBertForMultipleChoicer   c                 (    t          | dg           d S 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dS )"FlaxDistilBertForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )'FlaxDistilBertForSequenceClassificationr   c                 (    t          | dg           d S 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dS )$FlaxDistilBertForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxDistilBertModelr   c                 (    t          | dg           d S 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dS )FlaxDistilBertPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxElectraForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxElectraForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxElectraForMultipleChoicer   c                 (    t          | dg           d S 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dS )FlaxElectraForPreTrainingr   c                 (    t          | dg           d S 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dS )FlaxElectraForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )$FlaxElectraForSequenceClassificationr   c                 (    t          | dg           d S 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dS )!FlaxElectraForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxElectraModelr   c                 (    t          | dg           d S 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dS )FlaxElectraPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxEncoderDecoderModelr   c                 (    t          | dg           d S 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dS )FlaxGemmaForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxGemmaModelr   c                 (    t          | dg           d S 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dS )FlaxGemmaPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxGPT2LMHeadModelr   c                 (    t          | dg           d S 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dS )FlaxGPT2Modelr   c                 (    t          | dg           d S 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dS )FlaxGPT2PreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxGPTNeoForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxGPTNeoModelr   c                 (    t          | dg           d S 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dS )FlaxGPTNeoPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxGPTJForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxGPTJModelr   c                 (    t          | dg           d S 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dS )FlaxGPTJPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxLlamaForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxLlamaModelr   c                 (    t          | dg           d S 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dS )FlaxLlamaPreTrainedModelr   c                 (    t          | dg           d S 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dS )"FlaxLongT5ForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxLongT5Modelr   c                 (    t          | dg           d S 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dS )FlaxLongT5PreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxMarianModelr   c                 (    t          | dg           d S 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dS )FlaxMarianMTModelr   c                 (    t          | dg           d S 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dS )FlaxMarianPreTrainedModelr   c                 (    t          | dg           d S 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dS )!FlaxMBartForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxMBartForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )"FlaxMBartForSequenceClassificationr   c                 (    t          | dg           d S 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dS )FlaxMBartModelr   c                 (    t          | dg           d S 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dS )FlaxMBartPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxMistralForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxMistralModelr   c                 (    t          | dg           d S 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dS )FlaxMistralPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxMT5EncoderModelr   c                 (    t          | dg           d S 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dS )FlaxMT5ForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxMT5Modelr   c                 (    t          | dg           d S 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dS )FlaxOPTForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxOPTModelr   c                 (    t          | dg           d S 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dS )FlaxOPTPreTrainedModelr   c                 (    t          | dg           d S 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dS )#FlaxPegasusForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxPegasusModelr   c                 (    t          | dg           d S 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dS )FlaxPegasusPreTrainedModelr   c                 (    t          | dg           d S 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dS ) FlaxRegNetForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxRegNetModelr   c                 (    t          | dg           d S 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dS )FlaxRegNetPreTrainedModelr   c                 (    t          | dg           d S 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dS ) FlaxResNetForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxResNetModelr   c                 (    t          | dg           d S 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dS )FlaxResNetPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxRobertaForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxRobertaForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxRobertaForMultipleChoicer   c                 (    t          | dg           d S 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dS )FlaxRobertaForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )$FlaxRobertaForSequenceClassificationr   c                 (    t          | dg           d S 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dS )!FlaxRobertaForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxRobertaModelr   c                 (    t          | dg           d S 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dS )FlaxRobertaPreTrainedModelr   c                 (    t          | dg           d S 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dS )"FlaxRobertaPreLayerNormForCausalLMr   c                 (    t          | dg           d S 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dS )"FlaxRobertaPreLayerNormForMaskedLMr   c                 (    t          | dg           d S 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dS )(FlaxRobertaPreLayerNormForMultipleChoicer   c                 (    t          | dg           d S 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dS )+FlaxRobertaPreLayerNormForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )0FlaxRobertaPreLayerNormForSequenceClassificationr   c                 (    t          | dg           d S 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dS )-FlaxRobertaPreLayerNormForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxRobertaPreLayerNormModelr   c                 (    t          | dg           d S 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dS )&FlaxRobertaPreLayerNormPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxRoFormerForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxRoFormerForMultipleChoicer   c                 (    t          | dg           d S 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dS ) FlaxRoFormerForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )%FlaxRoFormerForSequenceClassificationr   c                 (    t          | dg           d S 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dS )"FlaxRoFormerForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxRoFormerModelr   c                 (    t          | dg           d S 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dS )FlaxRoFormerPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxSpeechEncoderDecoderModelr   c                 (    t          | dg           d S 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dS )FlaxT5EncoderModelr   c                 (    t          | dg           d S 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dS )FlaxT5ForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxT5Modelr   c                 (    t          | dg           d S 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dS )FlaxT5PreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxVisionEncoderDecoderModelr   c                 (    t          | dg           d S 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dS )FlaxVisionTextDualEncoderModelr   c                 (    t          | dg           d S 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dS )FlaxViTForImageClassificationr   c                 (    t          | dg           d S 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dS )FlaxViTModelr   c                 (    t          | dg           d S 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dS )FlaxViTPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxWav2Vec2ForCTCr   c                 (    t          | dg           d S 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dS )FlaxWav2Vec2ForPreTrainingr   c                 (    t          | dg           d S 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dS )FlaxWav2Vec2Modelr   c                 (    t          | dg           d S 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dS )FlaxWav2Vec2PreTrainedModelr   c                 (    t          | dg           d S 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dS )!FlaxWhisperForAudioClassificationr   c                 (    t          | dg           d S 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dS )#FlaxWhisperForConditionalGenerationr   c                 (    t          | dg           d S 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dS )FlaxWhisperModelr   c                 (    t          | dg           d S 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dS )FlaxWhisperPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxXGLMForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxXGLMModelr   c                 (    t          | dg           d S 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dS )FlaxXGLMPreTrainedModelr   c                 (    t          | dg           d S 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dS )FlaxXLMRobertaForCausalLMr   c                 (    t          | dg           d S 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dS )FlaxXLMRobertaForMaskedLMr   c                 (    t          | dg           d S 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dS )FlaxXLMRobertaForMultipleChoicer   c                 (    t          | dg           d S 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dS )"FlaxXLMRobertaForQuestionAnsweringr   c                 (    t          | dg           d S 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dS )'FlaxXLMRobertaForSequenceClassificationr   c                 (    t          | dg           d S 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dS )$FlaxXLMRobertaForTokenClassificationr   c                 (    t          | dg           d S 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dS )FlaxXLMRobertaModelr   c                 (    t          | dg           d S 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dS )FlaxXLMRobertaPreTrainedModelr   c                 (    t          | dg           d S 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 2 2 2 2 2 2 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   