o
    hK                     @   s  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZG d%d& d&edZG d'd( d(edZG d)d* d*edZG d+d, d,edZG d-d. d.edZG d/d0 d0edZd1Zd1Zd1Zd1Zd1Zd1Zd1Z d1Z!d1Z"d1Z#d1Z$d1Z%d1Z&d1Z'G d2d3 d3edZ(G d4d5 d5edZ)G d6d7 d7edZ*G d8d9 d9edZ+G d:d; d;edZ,G d<d= d=edZ-G d>d? d?edZ.G d@dA dAedZ/G dBdC dCedZ0G dDdE dEedZ1G dFdG dGedZ2G dHdI dIedZ3G dJdK dKedZ4G dLdM dMedZ5G dNdO dOedZ6G dPdQ dQedZ7G dRdS dSedZ8G dTdU dUedZ9G dVdW dWedZ:G dXdY dYedZ;G dZd[ d[edZ<G d\d] d]edZ=G d^d_ d_edZ>G d`da daedZ?G dbdc dcedZ@G ddde deedZAG dfdg dgedZBG dhdi diedZCG djdk dkedZDG dldm dmedZEG dndo doedZFG dpdq dqedZGG drds dsedZHG dtdu duedZIG dvdw dwedZJG dxdy dyedZKG dzd{ d{edZLG d|d} d}edZMG d~d dedZNG dd dedZOG dd dedZPG dd dedZQG dd dedZRG dd dedZSG dd dedZTG dd dedZUG dd dedZVG dd dedZWG dd dedZXG dd dedZYG dd dedZZG dd dedZ[G dd dedZ\G dd dedZ]G dd dedZ^G dd dedZ_G dd dedZ`G dd dedZaG dd dedZbG dd dedZcG dd dedZdG dd dedZeG dd dedZfG dd dedZgG dd dedZhG dd dedZiG dd dedZjG dd dedZkG dd dedZlG dd dedZmG dd dedZnG dd dedZoG ddÄ dedZpG ddń dedZqG ddǄ dedZrG ddɄ dedZsG dd˄ dedZtG dd̈́ dedZuG ddτ dedZvG ddф dedZwG ddӄ dedZxG ddՄ dedZyG ddׄ dedZzG ddل dedZ{G ddۄ dedZ|G dd݄ dedZ}G dd߄ dedZ~G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG d d dedZG dd dedZG dd dedZG dd dedZG dd	 d	edZG d
d dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG d d! d!edZG d"d# d#edZG d$d% d%edZG d&d' d'edZG d(d) d)edZG d*d+ d+edZG d,d- d-edZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZG d<d= d=edZG d>d? d?edZG d@dA dAedZG dBdC dCedZG dDdE dEedZG dFdG dGedZG dHdI dIedZG dJdK dKedZG dLdM dMedZG dNdO dOedZG dPdQ dQedZG dRdS dSedZG dTdU dUedZG dVdW dWedZG dXdY dYedZG dZd[ d[edZG d\d] d]edZG d^d_ d_edZG d`da daedZG dbdc dcedZG ddde deedZG dfdg dgedZG dhdi diedZG djdk dkedZG dldm dmedZd1ZG dndo doedZG dpdq dqedZG drds dsedZG dtdu duedZG dvdw dwedZG dxdy dyedZG dzd{ d{edZG d|d} d}edZd1S (~     )DummyObjectrequires_backendsc                   @      e Zd ZdgZdd ZdS )!FlaxForcedBOSTokenLogitsProcessorflaxc                 O      t | dg d S Nr   r   selfargskwargs r   [/var/www/html/ai/venv/lib/python3.10/site-packages/transformers/utils/dummy_flax_objects.py__init__      z*FlaxForcedBOSTokenLogitsProcessor.__init__N__name__
__module____qualname__	_backendsr   r   r   r   r   r          r   )	metaclassc                   @   r   )!FlaxForcedEOSTokenLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z*FlaxForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )FlaxForceTokensLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r      r   z'FlaxForceTokensLogitsProcessor.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )FlaxGenerationMixinr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxGenerationMixin.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )FlaxLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   $   r   zFlaxLogitsProcessor.__init__Nr   r   r   r   r   r   !   r   r   c                   @   r   )FlaxLogitsProcessorListr   c                 O   r   r   r	   r
   r   r   r   r   +   r   z FlaxLogitsProcessorList.__init__Nr   r   r   r   r   r   (   r   r   c                   @   r   )FlaxLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   2   r   zFlaxLogitsWarper.__init__Nr   r   r   r   r   r   /   r   r   c                   @   r   )FlaxMinLengthLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   9   r   z%FlaxMinLengthLogitsProcessor.__init__Nr   r   r   r   r   r   6   r   r   c                   @   r   )(FlaxSuppressTokensAtBeginLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   @   r   z1FlaxSuppressTokensAtBeginLogitsProcessor.__init__Nr   r   r   r   r   r    =   r   r    c                   @   r   )!FlaxSuppressTokensLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   G   r   z*FlaxSuppressTokensLogitsProcessor.__init__Nr   r   r   r   r   r!   D   r   r!   c                   @   r   )FlaxTemperatureLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   N   r   z$FlaxTemperatureLogitsWarper.__init__Nr   r   r   r   r   r"   K   r   r"   c                   @   r   )FlaxTopKLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   U   r   zFlaxTopKLogitsWarper.__init__Nr   r   r   r   r   r#   R   r   r#   c                   @   r   )FlaxTopPLogitsWarperr   c                 O   r   r   r	   r
   r   r   r   r   \   r   zFlaxTopPLogitsWarper.__init__Nr   r   r   r   r   r$   Y   r   r$   c                   @   r   )#FlaxWhisperTimeStampLogitsProcessorr   c                 O   r   r   r	   r
   r   r   r   r   c   r   z,FlaxWhisperTimeStampLogitsProcessor.__init__Nr   r   r   r   r   r%   `   r   r%   c                   @   r   )FlaxPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   j   r   zFlaxPreTrainedModel.__init__Nr   r   r   r   r   r&   g   r   r&   c                   @   r   )FlaxAlbertForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r   q   r   zFlaxAlbertForMaskedLM.__init__Nr   r   r   r   r   r'   n   r   r'   c                   @   r   )FlaxAlbertForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r   x   r   z$FlaxAlbertForMultipleChoice.__init__Nr   r   r   r   r   r(   u   r   r(   c                   @   r   )FlaxAlbertForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r      r   z!FlaxAlbertForPreTraining.__init__Nr   r   r   r   r   r)   |   r   r)   c                   @   r   )FlaxAlbertForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r      r   z'FlaxAlbertForQuestionAnswering.__init__Nr   r   r   r   r   r*      r   r*   c                   @   r   )#FlaxAlbertForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r      r   z,FlaxAlbertForSequenceClassification.__init__Nr   r   r   r   r   r+      r   r+   c                   @   r   ) FlaxAlbertForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r      r   z)FlaxAlbertForTokenClassification.__init__Nr   r   r   r   r   r,      r   r,   c                   @   r   )FlaxAlbertModelr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxAlbertModel.__init__Nr   r   r   r   r   r-      r   r-   c                   @   r   )FlaxAlbertPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r      r   z"FlaxAlbertPreTrainedModel.__init__Nr   r   r   r   r   r.      r   r.   Nc                   @   r   )FlaxAutoModelr   c                 O   r   r   r	   r
   r   r   r   r      r   zFlaxAutoModel.__init__Nr   r   r   r   r   r/      r   r/   c                   @   r   )FlaxAutoModelForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r      r   z!FlaxAutoModelForCausalLM.__init__Nr   r   r   r   r   r0      r   r0   c                   @   r   )#FlaxAutoModelForImageClassificationr   c                 O   r   r   r	   r
   r   r   r   r      r   z,FlaxAutoModelForImageClassification.__init__Nr   r   r   r   r   r1      r   r1   c                   @   r   )FlaxAutoModelForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r      r   z!FlaxAutoModelForMaskedLM.__init__Nr   r   r   r   r   r2      r   r2   c                   @   r   )FlaxAutoModelForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r      r   z'FlaxAutoModelForMultipleChoice.__init__Nr   r   r   r   r   r3      r   r3   c                   @   r   )&FlaxAutoModelForNextSentencePredictionr   c                 O   r   r   r	   r
   r   r   r   r      r   z/FlaxAutoModelForNextSentencePrediction.__init__Nr   r   r   r   r   r4      r   r4   c                   @   r   )FlaxAutoModelForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r      r   z$FlaxAutoModelForPreTraining.__init__Nr   r   r   r   r   r5      r   r5   c                   @   r   )!FlaxAutoModelForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxAutoModelForQuestionAnswering.__init__Nr   r   r   r   r   r6     r   r6   c                   @   r   )FlaxAutoModelForSeq2SeqLMr   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxAutoModelForSeq2SeqLM.__init__Nr   r   r   r   r   r7     r   r7   c                   @   r   )&FlaxAutoModelForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z/FlaxAutoModelForSequenceClassification.__init__Nr   r   r   r   r   r8     r   r8   c                   @   r   )FlaxAutoModelForSpeechSeq2Seqr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxAutoModelForSpeechSeq2Seq.__init__Nr   r   r   r   r   r9     r   r9   c                   @   r   )#FlaxAutoModelForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r      r   z,FlaxAutoModelForTokenClassification.__init__Nr   r   r   r   r   r:     r   r:   c                   @   r   )FlaxAutoModelForVision2Seqr   c                 O   r   r   r	   r
   r   r   r   r   '  r   z#FlaxAutoModelForVision2Seq.__init__Nr   r   r   r   r   r;   $  r   r;   c                   @   r   )FlaxBartDecoderPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   .  r   z'FlaxBartDecoderPreTrainedModel.__init__Nr   r   r   r   r   r<   +  r   r<   c                   @   r   )FlaxBartForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   5  r   zFlaxBartForCausalLM.__init__Nr   r   r   r   r   r=   2  r   r=   c                   @   r   ) FlaxBartForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r   <  r   z)FlaxBartForConditionalGeneration.__init__Nr   r   r   r   r   r>   9  r   r>   c                   @   r   )FlaxBartForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   C  r   z%FlaxBartForQuestionAnswering.__init__Nr   r   r   r   r   r?   @  r   r?   c                   @   r   )!FlaxBartForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r   J  r   z*FlaxBartForSequenceClassification.__init__Nr   r   r   r   r   r@   G  r   r@   c                   @   r   )FlaxBartModelr   c                 O   r   r   r	   r
   r   r   r   r   Q  r   zFlaxBartModel.__init__Nr   r   r   r   r   rA   N  r   rA   c                   @   r   )FlaxBartPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   X  r   z FlaxBartPreTrainedModel.__init__Nr   r   r   r   r   rB   U  r   rB   c                   @   r   )FlaxBeitForImageClassificationr   c                 O   r   r   r	   r
   r   r   r   r   _  r   z'FlaxBeitForImageClassification.__init__Nr   r   r   r   r   rC   \  r   rC   c                   @   r   )FlaxBeitForMaskedImageModelingr   c                 O   r   r   r	   r
   r   r   r   r   f  r   z'FlaxBeitForMaskedImageModeling.__init__Nr   r   r   r   r   rD   c  r   rD   c                   @   r   )FlaxBeitModelr   c                 O   r   r   r	   r
   r   r   r   r   m  r   zFlaxBeitModel.__init__Nr   r   r   r   r   rE   j  r   rE   c                   @   r   )FlaxBeitPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   t  r   z FlaxBeitPreTrainedModel.__init__Nr   r   r   r   r   rF   q  r   rF   c                   @   r   )FlaxBertForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   {  r   zFlaxBertForCausalLM.__init__Nr   r   r   r   r   rG   x  r   rG   c                   @   r   )FlaxBertForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBertForMaskedLM.__init__Nr   r   r   r   r   rH     r   rH   c                   @   r   )FlaxBertForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxBertForMultipleChoice.__init__Nr   r   r   r   r   rI     r   rI   c                   @   r   )!FlaxBertForNextSentencePredictionr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxBertForNextSentencePrediction.__init__Nr   r   r   r   r   rJ     r   rJ   c                   @   r   )FlaxBertForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBertForPreTraining.__init__Nr   r   r   r   r   rK     r   rK   c                   @   r   )FlaxBertForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r     r   z%FlaxBertForQuestionAnswering.__init__Nr   r   r   r   r   rL     r   rL   c                   @   r   )!FlaxBertForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxBertForSequenceClassification.__init__Nr   r   r   r   r   rM     r   rM   c                   @   r   )FlaxBertForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z'FlaxBertForTokenClassification.__init__Nr   r   r   r   r   rN     r   rN   c                   @   r   )FlaxBertModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBertModel.__init__Nr   r   r   r   r   rO     r   rO   c                   @   r   )FlaxBertPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z FlaxBertPreTrainedModel.__init__Nr   r   r   r   r   rP     r   rP   c                   @   r   )FlaxBigBirdForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBigBirdForCausalLM.__init__Nr   r   r   r   r   rQ     r   rQ   c                   @   r   )FlaxBigBirdForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBigBirdForMaskedLM.__init__Nr   r   r   r   r   rR     r   rR   c                   @   r   )FlaxBigBirdForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r     r   z%FlaxBigBirdForMultipleChoice.__init__Nr   r   r   r   r   rS     r   rS   c                   @   r   )FlaxBigBirdForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxBigBirdForPreTraining.__init__Nr   r   r   r   r   rT     r   rT   c                   @   r   )FlaxBigBirdForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r     r   z(FlaxBigBirdForQuestionAnswering.__init__Nr   r   r   r   r   rU     r   rU   c                   @   r   )$FlaxBigBirdForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z-FlaxBigBirdForSequenceClassification.__init__Nr   r   r   r   r   rV     r   rV   c                   @   r   )!FlaxBigBirdForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxBigBirdForTokenClassification.__init__Nr   r   r   r   r   rW     r   rW   c                   @   r   )FlaxBigBirdModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBigBirdModel.__init__Nr   r   r   r   r   rX     r   rX   c                   @   r   )FlaxBigBirdPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#FlaxBigBirdPreTrainedModel.__init__Nr   r   r   r   r   rY     r   rY   c                   @   r   )&FlaxBlenderbotForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r      r   z/FlaxBlenderbotForConditionalGeneration.__init__Nr   r   r   r   r   rZ     r   rZ   c                   @   r   )FlaxBlenderbotModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxBlenderbotModel.__init__Nr   r   r   r   r   r[     r   r[   c                   @   r   )FlaxBlenderbotPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxBlenderbotPreTrainedModel.__init__Nr   r   r   r   r   r\     r   r\   c                   @   r   )+FlaxBlenderbotSmallForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r     r   z4FlaxBlenderbotSmallForConditionalGeneration.__init__Nr   r   r   r   r   r]     r   r]   c                   @   r   )FlaxBlenderbotSmallModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!FlaxBlenderbotSmallModel.__init__Nr   r   r   r   r   r^     r   r^   c                   @   r   )"FlaxBlenderbotSmallPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   #  r   z+FlaxBlenderbotSmallPreTrainedModel.__init__Nr   r   r   r   r   r_      r   r_   c                   @   r   )FlaxBloomForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   *  r   zFlaxBloomForCausalLM.__init__Nr   r   r   r   r   r`   '  r   r`   c                   @   r   )FlaxBloomModelr   c                 O   r   r   r	   r
   r   r   r   r   1  r   zFlaxBloomModel.__init__Nr   r   r   r   r   ra   .  r   ra   c                   @   r   )FlaxBloomPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   8  r   z!FlaxBloomPreTrainedModel.__init__Nr   r   r   r   r   rb   5  r   rb   c                   @   r   )FlaxCLIPModelr   c                 O   r   r   r	   r
   r   r   r   r   ?  r   zFlaxCLIPModel.__init__Nr   r   r   r   r   rc   <  r   rc   c                   @   r   )FlaxCLIPPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   F  r   z FlaxCLIPPreTrainedModel.__init__Nr   r   r   r   r   rd   C  r   rd   c                   @   r   )FlaxCLIPTextModelr   c                 O   r   r   r	   r
   r   r   r   r   M  r   zFlaxCLIPTextModel.__init__Nr   r   r   r   r   re   J  r   re   c                   @   r   )FlaxCLIPTextModelWithProjectionr   c                 O   r   r   r	   r
   r   r   r   r   T  r   z(FlaxCLIPTextModelWithProjection.__init__Nr   r   r   r   r   rf   Q  r   rf   c                   @   r   )FlaxCLIPTextPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   [  r   z$FlaxCLIPTextPreTrainedModel.__init__Nr   r   r   r   r   rg   X  r   rg   c                   @   r   )FlaxCLIPVisionModelr   c                 O   r   r   r	   r
   r   r   r   r   b  r   zFlaxCLIPVisionModel.__init__Nr   r   r   r   r   rh   _  r   rh   c                   @   r   )FlaxCLIPVisionPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   i  r   z&FlaxCLIPVisionPreTrainedModel.__init__Nr   r   r   r   r   ri   f  r   ri   c                   @   r   )FlaxDistilBertForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r   p  r   z"FlaxDistilBertForMaskedLM.__init__Nr   r   r   r   r   rj   m  r   rj   c                   @   r   )FlaxDistilBertForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r   w  r   z(FlaxDistilBertForMultipleChoice.__init__Nr   r   r   r   r   rk   t  r   rk   c                   @   r   )"FlaxDistilBertForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   ~  r   z+FlaxDistilBertForQuestionAnswering.__init__Nr   r   r   r   r   rl   {  r   rl   c                   @   r   )'FlaxDistilBertForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z0FlaxDistilBertForSequenceClassification.__init__Nr   r   r   r   r   rm     r   rm   c                   @   r   )$FlaxDistilBertForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z-FlaxDistilBertForTokenClassification.__init__Nr   r   r   r   r   rn     r   rn   c                   @   r   )FlaxDistilBertModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxDistilBertModel.__init__Nr   r   r   r   r   ro     r   ro   c                   @   r   )FlaxDistilBertPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxDistilBertPreTrainedModel.__init__Nr   r   r   r   r   rp     r   rp   c                   @   r   )FlaxElectraForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxElectraForCausalLM.__init__Nr   r   r   r   r   rq     r   rq   c                   @   r   )FlaxElectraForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxElectraForMaskedLM.__init__Nr   r   r   r   r   rr     r   rr   c                   @   r   )FlaxElectraForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r     r   z%FlaxElectraForMultipleChoice.__init__Nr   r   r   r   r   rs     r   rs   c                   @   r   )FlaxElectraForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxElectraForPreTraining.__init__Nr   r   r   r   r   rt     r   rt   c                   @   r   )FlaxElectraForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r     r   z(FlaxElectraForQuestionAnswering.__init__Nr   r   r   r   r   ru     r   ru   c                   @   r   )$FlaxElectraForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z-FlaxElectraForSequenceClassification.__init__Nr   r   r   r   r   rv     r   rv   c                   @   r   )!FlaxElectraForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxElectraForTokenClassification.__init__Nr   r   r   r   r   rw     r   rw   c                   @   r   )FlaxElectraModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxElectraModel.__init__Nr   r   r   r   r   rx     r   rx   c                   @   r   )FlaxElectraPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#FlaxElectraPreTrainedModel.__init__Nr   r   r   r   r   ry     r   ry   c                   @   r   )FlaxEncoderDecoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z FlaxEncoderDecoderModel.__init__Nr   r   r   r   r   rz     r   rz   c                   @   r   )FlaxGPT2LMHeadModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPT2LMHeadModel.__init__Nr   r   r   r   r   r{     r   r{   c                   @   r   )FlaxGPT2Modelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPT2Model.__init__Nr   r   r   r   r   r|     r   r|   c                   @   r   )FlaxGPT2PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z FlaxGPT2PreTrainedModel.__init__Nr   r   r   r   r   r}     r   r}   c                   @   r   )FlaxGPTNeoForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPTNeoForCausalLM.__init__Nr   r   r   r   r   r~     r   r~   c                   @   r   )FlaxGPTNeoModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPTNeoModel.__init__Nr   r   r   r   r   r      r   r   c                   @   r   )FlaxGPTNeoPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   
  r   z"FlaxGPTNeoPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxGPTJForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPTJForCausalLM.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxGPTJModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxGPTJModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxGPTJPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z FlaxGPTJPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxLlamaForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   &  r   zFlaxLlamaForCausalLM.__init__Nr   r   r   r   r   r   #  r   r   c                   @   r   )FlaxLlamaModelr   c                 O   r   r   r	   r
   r   r   r   r   -  r   zFlaxLlamaModel.__init__Nr   r   r   r   r   r   *  r   r   c                   @   r   )FlaxLlamaPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   4  r   z!FlaxLlamaPreTrainedModel.__init__Nr   r   r   r   r   r   1  r   r   c                   @   r   )"FlaxLongT5ForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r   ;  r   z+FlaxLongT5ForConditionalGeneration.__init__Nr   r   r   r   r   r   8  r   r   c                   @   r   )FlaxLongT5Modelr   c                 O   r   r   r	   r
   r   r   r   r   B  r   zFlaxLongT5Model.__init__Nr   r   r   r   r   r   ?  r   r   c                   @   r   )FlaxLongT5PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   I  r   z"FlaxLongT5PreTrainedModel.__init__Nr   r   r   r   r   r   F  r   r   c                   @   r   )FlaxMarianModelr   c                 O   r   r   r	   r
   r   r   r   r   P  r   zFlaxMarianModel.__init__Nr   r   r   r   r   r   M  r   r   c                   @   r   )FlaxMarianMTModelr   c                 O   r   r   r	   r
   r   r   r   r   W  r   zFlaxMarianMTModel.__init__Nr   r   r   r   r   r   T  r   r   c                   @   r   )FlaxMarianPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   ^  r   z"FlaxMarianPreTrainedModel.__init__Nr   r   r   r   r   r   [  r   r   c                   @   r   )!FlaxMBartForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r   e  r   z*FlaxMBartForConditionalGeneration.__init__Nr   r   r   r   r   r   b  r   r   c                   @   r   )FlaxMBartForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   l  r   z&FlaxMBartForQuestionAnswering.__init__Nr   r   r   r   r   r   i  r   r   c                   @   r   )"FlaxMBartForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r   s  r   z+FlaxMBartForSequenceClassification.__init__Nr   r   r   r   r   r   p  r   r   c                   @   r   )FlaxMBartModelr   c                 O   r   r   r	   r
   r   r   r   r   z  r   zFlaxMBartModel.__init__Nr   r   r   r   r   r   w  r   r   c                   @   r   )FlaxMBartPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!FlaxMBartPreTrainedModel.__init__Nr   r   r   r   r   r   ~  r   r   c                   @   r   )FlaxMT5EncoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxMT5EncoderModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxMT5ForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r     r   z(FlaxMT5ForConditionalGeneration.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxMT5Modelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxMT5Model.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxOPTForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxOPTForCausalLM.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxOPTModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxOPTModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxOPTPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxOPTPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )#FlaxPegasusForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r     r   z,FlaxPegasusForConditionalGeneration.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxPegasusModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxPegasusModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxPegasusPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#FlaxPegasusPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   ) FlaxRegNetForImageClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z)FlaxRegNetForImageClassification.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRegNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxRegNetModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRegNetPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxRegNetPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   ) FlaxResNetForImageClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z)FlaxResNetForImageClassification.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxResNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxResNetModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxResNetPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z"FlaxResNetPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxRobertaForCausalLM.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxRobertaForMaskedLM.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r     r   z%FlaxRobertaForMultipleChoice.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r     r   z(FlaxRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )$FlaxRobertaForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z-FlaxRobertaForSequenceClassification.__init__Nr   r   r   r   r   r   
  r   r   c                   @   r   )!FlaxRobertaForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxRobertaForTokenClassification.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxRobertaModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRobertaPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   "  r   z#FlaxRobertaPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )"FlaxRobertaPreLayerNormForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   )  r   z+FlaxRobertaPreLayerNormForCausalLM.__init__Nr   r   r   r   r   r   &  r   r   c                   @   r   )"FlaxRobertaPreLayerNormForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r   0  r   z+FlaxRobertaPreLayerNormForMaskedLM.__init__Nr   r   r   r   r   r   -  r   r   c                   @   r   )(FlaxRobertaPreLayerNormForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r   7  r   z1FlaxRobertaPreLayerNormForMultipleChoice.__init__Nr   r   r   r   r   r   4  r   r   c                   @   r   )+FlaxRobertaPreLayerNormForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   >  r   z4FlaxRobertaPreLayerNormForQuestionAnswering.__init__Nr   r   r   r   r   r   ;  r   r   c                   @   r   )0FlaxRobertaPreLayerNormForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r   E  r   z9FlaxRobertaPreLayerNormForSequenceClassification.__init__Nr   r   r   r   r   r   B  r   r   c                   @   r   )-FlaxRobertaPreLayerNormForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r   L  r   z6FlaxRobertaPreLayerNormForTokenClassification.__init__Nr   r   r   r   r   r   I  r   r   c                   @   r   )FlaxRobertaPreLayerNormModelr   c                 O   r   r   r	   r
   r   r   r   r   S  r   z%FlaxRobertaPreLayerNormModel.__init__Nr   r   r   r   r   r   P  r   r   c                   @   r   )&FlaxRobertaPreLayerNormPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   Z  r   z/FlaxRobertaPreLayerNormPreTrainedModel.__init__Nr   r   r   r   r   r   W  r   r   c                   @   r   )FlaxRoFormerForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r   a  r   z FlaxRoFormerForMaskedLM.__init__Nr   r   r   r   r   r   ^  r   r   c                   @   r   )FlaxRoFormerForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r   h  r   z&FlaxRoFormerForMultipleChoice.__init__Nr   r   r   r   r   r   e  r   r   c                   @   r   ) FlaxRoFormerForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   o  r   z)FlaxRoFormerForQuestionAnswering.__init__Nr   r   r   r   r   r   l  r   r   c                   @   r   )%FlaxRoFormerForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r   v  r   z.FlaxRoFormerForSequenceClassification.__init__Nr   r   r   r   r   r   s  r   r   c                   @   r   )"FlaxRoFormerForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r   }  r   z+FlaxRoFormerForTokenClassification.__init__Nr   r   r   r   r   r   z  r   r   c                   @   r   )FlaxRoFormerModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxRoFormerModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxRoFormerPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$FlaxRoFormerPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxSpeechEncoderDecoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxSpeechEncoderDecoderModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxT5EncoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxT5EncoderModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxT5ForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r     r   z'FlaxT5ForConditionalGeneration.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxT5Modelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxT5Model.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxT5PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxT5PreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxVisionEncoderDecoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxVisionEncoderDecoderModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxVisionTextDualEncoderModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z'FlaxVisionTextDualEncoderModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxViTForImageClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z&FlaxViTForImageClassification.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxViTModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxViTModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxViTPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxViTPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWav2Vec2ForCTCr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxWav2Vec2ForCTC.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWav2Vec2ForPreTrainingr   c                 O   r   r   r	   r
   r   r   r   r     r   z#FlaxWav2Vec2ForPreTraining.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWav2Vec2Modelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxWav2Vec2Model.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWav2Vec2PreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$FlaxWav2Vec2PreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )!FlaxWhisperForAudioClassificationr   c                 O   r   r   r	   r
   r   r   r   r     r   z*FlaxWhisperForAudioClassification.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )#FlaxWhisperForConditionalGenerationr   c                 O   r   r   r	   r
   r   r   r   r     r   z,FlaxWhisperForConditionalGeneration.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWhisperModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxWhisperModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxWhisperPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   	  r   z#FlaxWhisperPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxXGLMForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxXGLMForCausalLM.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxXGLMModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFlaxXGLMModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxXGLMPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z FlaxXGLMPreTrainedModel.__init__Nr   r   r   r   r   r     r   r   c                   @   r   )FlaxXLMRobertaForCausalLMr   c                 O   r   r   r	   r
   r   r   r   r   (  r   z"FlaxXLMRobertaForCausalLM.__init__Nr   r   r   r   r   r   %  r   r   c                   @   r   )FlaxXLMRobertaForMaskedLMr   c                 O   r   r   r	   r
   r   r   r   r   /  r   z"FlaxXLMRobertaForMaskedLM.__init__Nr   r   r   r   r   r   ,  r   r   c                   @   r   )FlaxXLMRobertaForMultipleChoicer   c                 O   r   r   r	   r
   r   r   r   r   6  r   z(FlaxXLMRobertaForMultipleChoice.__init__Nr   r   r   r   r   r   3  r   r   c                   @   r   )"FlaxXLMRobertaForQuestionAnsweringr   c                 O   r   r   r	   r
   r   r   r   r   =  r   z+FlaxXLMRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r   :  r   r   c                   @   r   )'FlaxXLMRobertaForSequenceClassificationr   c                 O   r   r   r	   r
   r   r   r   r   D  r   z0FlaxXLMRobertaForSequenceClassification.__init__Nr   r   r   r   r   r   A  r   r   c                   @   r   )$FlaxXLMRobertaForTokenClassificationr   c                 O   r   r   r	   r
   r   r   r   r   K  r   z-FlaxXLMRobertaForTokenClassification.__init__Nr   r   r   r   r   r   H  r   r   c                   @   r   )FlaxXLMRobertaModelr   c                 O   r   r   r	   r
   r   r   r   r   R  r   zFlaxXLMRobertaModel.__init__Nr   r   r   r   r   r   O  r   r   c                   @   r   )FlaxXLMRobertaPreTrainedModelr   c                 O   r   r   r	   r
   r   r   r   r   Y  r   z&FlaxXLMRobertaPreTrainedModel.__init__Nr   r   r   r   r   r   V  r   r   )utilsr   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   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/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   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   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   .FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   r   <module>   s  