U
    9%eg}                     @   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Zd1ZG 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Zd1S (x     )DummyObjectrequires_backendsc                   @   s   e Zd ZdgZdd ZdS )!FlaxForcedBOSTokenLogitsProcessorflaxc                 O   s   t | dg d S Nr   r   selfargskwargs r   d/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/utils/dummy_flax_objects.py__init__   s    z*FlaxForcedBOSTokenLogitsProcessor.__init__N__name__
__module____qualname__Z	_backendsr   r   r   r   r   r      s   r   )	metaclassc                   @   s   e Zd ZdgZdd ZdS )!FlaxForcedEOSTokenLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z*FlaxForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxForceTokensLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z'FlaxForceTokensLogitsProcessor.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxGenerationMixinr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zFlaxGenerationMixin.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $   s    zFlaxLogitsProcessor.__init__Nr   r   r   r   r   r   !   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxLogitsProcessorListr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +   s    z FlaxLogitsProcessorList.__init__Nr   r   r   r   r   r   (   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2   s    zFlaxLogitsWarper.__init__Nr   r   r   r   r   r   /   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMinLengthLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9   s    z%FlaxMinLengthLogitsProcessor.__init__Nr   r   r   r   r   r   6   s   r   c                   @   s   e Zd ZdgZdd ZdS )(FlaxSuppressTokensAtBeginLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @   s    z1FlaxSuppressTokensAtBeginLogitsProcessor.__init__Nr   r   r   r   r   r   =   s   r   c                   @   s   e Zd ZdgZdd ZdS )!FlaxSuppressTokensLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G   s    z*FlaxSuppressTokensLogitsProcessor.__init__Nr   r   r   r   r   r   D   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxTemperatureLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N   s    z$FlaxTemperatureLogitsWarper.__init__Nr   r   r   r   r   r   K   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxTopKLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U   s    zFlaxTopKLogitsWarper.__init__Nr   r   r   r   r   r   R   s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxTopPLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \   s    zFlaxTopPLogitsWarper.__init__Nr   r   r   r   r   r   Y   s   r   c                   @   s   e Zd ZdgZdd ZdS )#FlaxWhisperTimeStampLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c   s    z,FlaxWhisperTimeStampLogitsProcessor.__init__Nr   r   r   r   r   r    `   s   r    c                   @   s   e Zd ZdgZdd ZdS )FlaxPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j   s    zFlaxPreTrainedModel.__init__Nr   r   r   r   r   r!   g   s   r!   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q   s    zFlaxAlbertForMaskedLM.__init__Nr   r   r   r   r   r"   n   s   r"   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   x   s    z$FlaxAlbertForMultipleChoice.__init__Nr   r   r   r   r   r#   u   s   r#   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z!FlaxAlbertForPreTraining.__init__Nr   r   r   r   r   r$   |   s   r$   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z'FlaxAlbertForQuestionAnswering.__init__Nr   r   r   r   r   r%      s   r%   c                   @   s   e Zd ZdgZdd ZdS )#FlaxAlbertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z,FlaxAlbertForSequenceClassification.__init__Nr   r   r   r   r   r&      s   r&   c                   @   s   e Zd ZdgZdd ZdS ) FlaxAlbertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z)FlaxAlbertForTokenClassification.__init__Nr   r   r   r   r   r'      s   r'   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zFlaxAlbertModel.__init__Nr   r   r   r   r   r(      s   r(   c                   @   s   e Zd ZdgZdd ZdS )FlaxAlbertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"FlaxAlbertPreTrainedModel.__init__Nr   r   r   r   r   r)      s   r)   Nc                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zFlaxAutoModel.__init__Nr   r   r   r   r   r*      s   r*   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z!FlaxAutoModelForCausalLM.__init__Nr   r   r   r   r   r+      s   r+   c                   @   s   e Zd ZdgZdd ZdS )#FlaxAutoModelForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z,FlaxAutoModelForImageClassification.__init__Nr   r   r   r   r   r,      s   r,   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z!FlaxAutoModelForMaskedLM.__init__Nr   r   r   r   r   r-      s   r-   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z'FlaxAutoModelForMultipleChoice.__init__Nr   r   r   r   r   r.      s   r.   c                   @   s   e Zd ZdgZdd ZdS )&FlaxAutoModelForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z/FlaxAutoModelForNextSentencePrediction.__init__Nr   r   r   r   r   r/      s   r/   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z$FlaxAutoModelForPreTraining.__init__Nr   r   r   r   r   r0      s   r0   c                   @   s   e Zd ZdgZdd ZdS )!FlaxAutoModelForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxAutoModelForQuestionAnswering.__init__Nr   r   r   r   r   r1     s   r1   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForSeq2SeqLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxAutoModelForSeq2SeqLM.__init__Nr   r   r   r   r   r2     s   r2   c                   @   s   e Zd ZdgZdd ZdS )&FlaxAutoModelForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/FlaxAutoModelForSequenceClassification.__init__Nr   r   r   r   r   r3     s   r3   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForSpeechSeq2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FlaxAutoModelForSpeechSeq2Seq.__init__Nr   r   r   r   r   r4     s   r4   c                   @   s   e Zd ZdgZdd ZdS )#FlaxAutoModelForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z,FlaxAutoModelForTokenClassification.__init__Nr   r   r   r   r   r5     s   r5   c                   @   s   e Zd ZdgZdd ZdS )FlaxAutoModelForVision2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    z#FlaxAutoModelForVision2Seq.__init__Nr   r   r   r   r   r6   $  s   r6   c                   @   s   e Zd ZdgZdd ZdS )FlaxBartDecoderPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    z'FlaxBartDecoderPreTrainedModel.__init__Nr   r   r   r   r   r7   +  s   r7   c                   @   s   e Zd ZdgZdd ZdS )FlaxBartForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    zFlaxBartForCausalLM.__init__Nr   r   r   r   r   r8   2  s   r8   c                   @   s   e Zd ZdgZdd ZdS ) FlaxBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    z)FlaxBartForConditionalGeneration.__init__Nr   r   r   r   r   r9   9  s   r9   c                   @   s   e Zd ZdgZdd ZdS )FlaxBartForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    z%FlaxBartForQuestionAnswering.__init__Nr   r   r   r   r   r:   @  s   r:   c                   @   s   e Zd ZdgZdd ZdS )!FlaxBartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    z*FlaxBartForSequenceClassification.__init__Nr   r   r   r   r   r;   G  s   r;   c                   @   s   e Zd ZdgZdd ZdS )FlaxBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    zFlaxBartModel.__init__Nr   r   r   r   r   r<   N  s   r<   c                   @   s   e Zd ZdgZdd ZdS )FlaxBartPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   X  s    z FlaxBartPreTrainedModel.__init__Nr   r   r   r   r   r=   U  s   r=   c                   @   s   e Zd ZdgZdd ZdS )FlaxBeitForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   _  s    z'FlaxBeitForImageClassification.__init__Nr   r   r   r   r   r>   \  s   r>   c                   @   s   e Zd ZdgZdd ZdS )FlaxBeitForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   f  s    z'FlaxBeitForMaskedImageModeling.__init__Nr   r   r   r   r   r?   c  s   r?   c                   @   s   e Zd ZdgZdd ZdS )FlaxBeitModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   m  s    zFlaxBeitModel.__init__Nr   r   r   r   r   r@   j  s   r@   c                   @   s   e Zd ZdgZdd ZdS )FlaxBeitPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   t  s    z FlaxBeitPreTrainedModel.__init__Nr   r   r   r   r   rA   q  s   rA   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    zFlaxBertForCausalLM.__init__Nr   r   r   r   r   rB   x  s   rB   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBertForMaskedLM.__init__Nr   r   r   r   r   rC     s   rC   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxBertForMultipleChoice.__init__Nr   r   r   r   r   rD     s   rD   c                   @   s   e Zd ZdgZdd ZdS )!FlaxBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxBertForNextSentencePrediction.__init__Nr   r   r   r   r   rE     s   rE   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBertForPreTraining.__init__Nr   r   r   r   r   rF     s   rF   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%FlaxBertForQuestionAnswering.__init__Nr   r   r   r   r   rG     s   rG   c                   @   s   e Zd ZdgZdd ZdS )!FlaxBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxBertForSequenceClassification.__init__Nr   r   r   r   r   rH     s   rH   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'FlaxBertForTokenClassification.__init__Nr   r   r   r   r   rI     s   rI   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBertModel.__init__Nr   r   r   r   r   rJ     s   rJ   c                   @   s   e Zd ZdgZdd ZdS )FlaxBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z FlaxBertPreTrainedModel.__init__Nr   r   r   r   r   rK     s   rK   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBigBirdForCausalLM.__init__Nr   r   r   r   r   rL     s   rL   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBigBirdForMaskedLM.__init__Nr   r   r   r   r   rM     s   rM   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%FlaxBigBirdForMultipleChoice.__init__Nr   r   r   r   r   rN     s   rN   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxBigBirdForPreTraining.__init__Nr   r   r   r   r   rO     s   rO   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(FlaxBigBirdForQuestionAnswering.__init__Nr   r   r   r   r   rP     s   rP   c                   @   s   e Zd ZdgZdd ZdS )$FlaxBigBirdForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-FlaxBigBirdForSequenceClassification.__init__Nr   r   r   r   r   rQ     s   rQ   c                   @   s   e Zd ZdgZdd ZdS )!FlaxBigBirdForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxBigBirdForTokenClassification.__init__Nr   r   r   r   r   rR     s   rR   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBigBirdModel.__init__Nr   r   r   r   r   rS     s   rS   c                   @   s   e Zd ZdgZdd ZdS )FlaxBigBirdPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxBigBirdPreTrainedModel.__init__Nr   r   r   r   r   rT     s   rT   c                   @   s   e Zd ZdgZdd ZdS )&FlaxBlenderbotForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z/FlaxBlenderbotForConditionalGeneration.__init__Nr   r   r   r   r   rU     s   rU   c                   @   s   e Zd ZdgZdd ZdS )FlaxBlenderbotModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxBlenderbotModel.__init__Nr   r   r   r   r   rV     s   rV   c                   @   s   e Zd ZdgZdd ZdS )FlaxBlenderbotPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FlaxBlenderbotPreTrainedModel.__init__Nr   r   r   r   r   rW     s   rW   c                   @   s   e Zd ZdgZdd ZdS )+FlaxBlenderbotSmallForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z4FlaxBlenderbotSmallForConditionalGeneration.__init__Nr   r   r   r   r   rX     s   rX   c                   @   s   e Zd ZdgZdd ZdS )FlaxBlenderbotSmallModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!FlaxBlenderbotSmallModel.__init__Nr   r   r   r   r   rY     s   rY   c                   @   s   e Zd ZdgZdd ZdS )"FlaxBlenderbotSmallPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    z+FlaxBlenderbotSmallPreTrainedModel.__init__Nr   r   r   r   r   rZ      s   rZ   c                   @   s   e Zd ZdgZdd ZdS )FlaxBloomForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    zFlaxBloomForCausalLM.__init__Nr   r   r   r   r   r[   '  s   r[   c                   @   s   e Zd ZdgZdd ZdS )FlaxBloomModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   1  s    zFlaxBloomModel.__init__Nr   r   r   r   r   r\   .  s   r\   c                   @   s   e Zd ZdgZdd ZdS )FlaxBloomPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   8  s    z!FlaxBloomPreTrainedModel.__init__Nr   r   r   r   r   r]   5  s   r]   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ?  s    zFlaxCLIPModel.__init__Nr   r   r   r   r   r^   <  s   r^   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   F  s    z FlaxCLIPPreTrainedModel.__init__Nr   r   r   r   r   r_   C  s   r_   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   M  s    zFlaxCLIPTextModel.__init__Nr   r   r   r   r   r`   J  s   r`   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPTextModelWithProjectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   T  s    z(FlaxCLIPTextModelWithProjection.__init__Nr   r   r   r   r   ra   Q  s   ra   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPTextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [  s    z$FlaxCLIPTextPreTrainedModel.__init__Nr   r   r   r   r   rb   X  s   rb   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b  s    zFlaxCLIPVisionModel.__init__Nr   r   r   r   r   rc   _  s   rc   c                   @   s   e Zd ZdgZdd ZdS )FlaxCLIPVisionPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   i  s    z&FlaxCLIPVisionPreTrainedModel.__init__Nr   r   r   r   r   rd   f  s   rd   c                   @   s   e Zd ZdgZdd ZdS )FlaxDistilBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p  s    z"FlaxDistilBertForMaskedLM.__init__Nr   r   r   r   r   re   m  s   re   c                   @   s   e Zd ZdgZdd ZdS )FlaxDistilBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   w  s    z(FlaxDistilBertForMultipleChoice.__init__Nr   r   r   r   r   rf   t  s   rf   c                   @   s   e Zd ZdgZdd ZdS )"FlaxDistilBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ~  s    z+FlaxDistilBertForQuestionAnswering.__init__Nr   r   r   r   r   rg   {  s   rg   c                   @   s   e Zd ZdgZdd ZdS )'FlaxDistilBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0FlaxDistilBertForSequenceClassification.__init__Nr   r   r   r   r   rh     s   rh   c                   @   s   e Zd ZdgZdd ZdS )$FlaxDistilBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-FlaxDistilBertForTokenClassification.__init__Nr   r   r   r   r   ri     s   ri   c                   @   s   e Zd ZdgZdd ZdS )FlaxDistilBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxDistilBertModel.__init__Nr   r   r   r   r   rj     s   rj   c                   @   s   e Zd ZdgZdd ZdS )FlaxDistilBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FlaxDistilBertPreTrainedModel.__init__Nr   r   r   r   r   rk     s   rk   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxElectraForCausalLM.__init__Nr   r   r   r   r   rl     s   rl   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxElectraForMaskedLM.__init__Nr   r   r   r   r   rm     s   rm   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%FlaxElectraForMultipleChoice.__init__Nr   r   r   r   r   rn     s   rn   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxElectraForPreTraining.__init__Nr   r   r   r   r   ro     s   ro   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(FlaxElectraForQuestionAnswering.__init__Nr   r   r   r   r   rp     s   rp   c                   @   s   e Zd ZdgZdd ZdS )$FlaxElectraForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-FlaxElectraForSequenceClassification.__init__Nr   r   r   r   r   rq     s   rq   c                   @   s   e Zd ZdgZdd ZdS )!FlaxElectraForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxElectraForTokenClassification.__init__Nr   r   r   r   r   rr     s   rr   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxElectraModel.__init__Nr   r   r   r   r   rs     s   rs   c                   @   s   e Zd ZdgZdd ZdS )FlaxElectraPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxElectraPreTrainedModel.__init__Nr   r   r   r   r   rt     s   rt   c                   @   s   e Zd ZdgZdd ZdS )FlaxEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z FlaxEncoderDecoderModel.__init__Nr   r   r   r   r   ru     s   ru   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPT2LMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPT2LMHeadModel.__init__Nr   r   r   r   r   rv     s   rv   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPT2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPT2Model.__init__Nr   r   r   r   r   rw     s   rw   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPT2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z FlaxGPT2PreTrainedModel.__init__Nr   r   r   r   r   rx     s   rx   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTNeoForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPTNeoForCausalLM.__init__Nr   r   r   r   r   ry     s   ry   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTNeoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPTNeoModel.__init__Nr   r   r   r   r   rz      s   rz   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTNeoPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z"FlaxGPTNeoPreTrainedModel.__init__Nr   r   r   r   r   r{     s   r{   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTJForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPTJForCausalLM.__init__Nr   r   r   r   r   r|     s   r|   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTJModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxGPTJModel.__init__Nr   r   r   r   r   r}     s   r}   c                   @   s   e Zd ZdgZdd ZdS )FlaxGPTJPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z FlaxGPTJPreTrainedModel.__init__Nr   r   r   r   r   r~     s   r~   c                   @   s   e Zd ZdgZdd ZdS )"FlaxLongT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    z+FlaxLongT5ForConditionalGeneration.__init__Nr   r   r   r   r   r   #  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxLongT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    zFlaxLongT5Model.__init__Nr   r   r   r   r   r   *  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxLongT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z"FlaxLongT5PreTrainedModel.__init__Nr   r   r   r   r   r   1  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMarianModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zFlaxMarianModel.__init__Nr   r   r   r   r   r   8  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMarianMTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zFlaxMarianMTModel.__init__Nr   r   r   r   r   r   ?  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMarianPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z"FlaxMarianPreTrainedModel.__init__Nr   r   r   r   r   r   F  s   r   c                   @   s   e Zd ZdgZdd ZdS )!FlaxMBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    z*FlaxMBartForConditionalGeneration.__init__Nr   r   r   r   r   r   M  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMBartForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    z&FlaxMBartForQuestionAnswering.__init__Nr   r   r   r   r   r   T  s   r   c                   @   s   e Zd ZdgZdd ZdS )"FlaxMBartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    z+FlaxMBartForSequenceClassification.__init__Nr   r   r   r   r   r   [  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e  s    zFlaxMBartModel.__init__Nr   r   r   r   r   r   b  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMBartPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   l  s    z!FlaxMBartPreTrainedModel.__init__Nr   r   r   r   r   r   i  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   s  s    zFlaxMT5EncoderModel.__init__Nr   r   r   r   r   r   p  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   z  s    z(FlaxMT5ForConditionalGeneration.__init__Nr   r   r   r   r   r   w  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxMT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxMT5Model.__init__Nr   r   r   r   r   r   ~  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxOPTForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxOPTForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxOPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxOPTModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxOPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxOPTPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )#FlaxPegasusForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,FlaxPegasusForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxPegasusModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxPegasusModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxPegasusPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxPegasusPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS ) FlaxRegNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)FlaxRegNetForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRegNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxRegNetModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRegNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxRegNetPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS ) FlaxResNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)FlaxResNetForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxResNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxResNetModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxResNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxResNetPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxRobertaForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxRobertaForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%FlaxRobertaForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(FlaxRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )$FlaxRobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-FlaxRobertaForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )!FlaxRobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxRobertaForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxRobertaModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxRobertaPreTrainedModel.__init__Nr   r   r   r   r   r   
  s   r   c                   @   s   e Zd ZdgZdd ZdS )"FlaxRobertaPreLayerNormForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+FlaxRobertaPreLayerNormForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"FlaxRobertaPreLayerNormForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+FlaxRobertaPreLayerNormForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )(FlaxRobertaPreLayerNormForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   "  s    z1FlaxRobertaPreLayerNormForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )+FlaxRobertaPreLayerNormForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )  s    z4FlaxRobertaPreLayerNormForQuestionAnswering.__init__Nr   r   r   r   r   r   &  s   r   c                   @   s   e Zd ZdgZdd ZdS )0FlaxRobertaPreLayerNormForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   0  s    z9FlaxRobertaPreLayerNormForSequenceClassification.__init__Nr   r   r   r   r   r   -  s   r   c                   @   s   e Zd ZdgZdd ZdS )-FlaxRobertaPreLayerNormForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7  s    z6FlaxRobertaPreLayerNormForTokenClassification.__init__Nr   r   r   r   r   r   4  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRobertaPreLayerNormModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    z%FlaxRobertaPreLayerNormModel.__init__Nr   r   r   r   r   r   ;  s   r   c                   @   s   e Zd ZdgZdd ZdS )&FlaxRobertaPreLayerNormPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    z/FlaxRobertaPreLayerNormPreTrainedModel.__init__Nr   r   r   r   r   r   B  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRoFormerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    z FlaxRoFormerForMaskedLM.__init__Nr   r   r   r   r   r   I  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRoFormerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z&FlaxRoFormerForMultipleChoice.__init__Nr   r   r   r   r   r   P  s   r   c                   @   s   e Zd ZdgZdd ZdS ) FlaxRoFormerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    z)FlaxRoFormerForQuestionAnswering.__init__Nr   r   r   r   r   r   W  s   r   c                   @   s   e Zd ZdgZdd ZdS )%FlaxRoFormerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z.FlaxRoFormerForSequenceClassification.__init__Nr   r   r   r   r   r   ^  s   r   c                   @   s   e Zd ZdgZdd ZdS )"FlaxRoFormerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z+FlaxRoFormerForTokenClassification.__init__Nr   r   r   r   r   r   e  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRoFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zFlaxRoFormerModel.__init__Nr   r   r   r   r   r   l  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxRoFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z$FlaxRoFormerPreTrainedModel.__init__Nr   r   r   r   r   r   s  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxSpeechEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    z&FlaxSpeechEncoderDecoderModel.__init__Nr   r   r   r   r   r   z  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxT5EncoderModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'FlaxT5ForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxT5Model.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxT5PreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxVisionEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FlaxVisionEncoderDecoderModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxVisionTextDualEncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'FlaxVisionTextDualEncoderModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxViTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FlaxViTForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxViTModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxViTPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWav2Vec2ForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxWav2Vec2ForCTC.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWav2Vec2ForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxWav2Vec2ForPreTraining.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWav2Vec2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxWav2Vec2Model.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWav2Vec2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$FlaxWav2Vec2PreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )!FlaxWhisperForAudioClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*FlaxWhisperForAudioClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )#FlaxWhisperForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,FlaxWhisperForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWhisperModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxWhisperModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxWhisperPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FlaxWhisperPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXGLMForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxXGLMForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXGLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlaxXGLMModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXGLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z FlaxXGLMPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXLMRobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxXLMRobertaForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXLMRobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"FlaxXLMRobertaForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXLMRobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   !  s    z(FlaxXLMRobertaForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"FlaxXLMRobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   (  s    z+FlaxXLMRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r   %  s   r   c                   @   s   e Zd ZdgZdd ZdS )'FlaxXLMRobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   /  s    z0FlaxXLMRobertaForSequenceClassification.__init__Nr   r   r   r   r   r   ,  s   r   c                   @   s   e Zd ZdgZdd ZdS )$FlaxXLMRobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   6  s    z-FlaxXLMRobertaForTokenClassification.__init__Nr   r   r   r   r   r   3  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXLMRobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   =  s    zFlaxXLMRobertaModel.__init__Nr   r   r   r   r   r   :  s   r   c                   @   s   e Zd ZdgZdd ZdS )FlaxXLMRobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   D  s    z&FlaxXLMRobertaPreTrainedModel.__init__Nr   r   r   r   r   r   A  s   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)   Z+FLAX_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGZ FLAX_MODEL_FOR_CAUSAL_LM_MAPPINGZ+FLAX_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGZ FLAX_MODEL_FOR_MASKED_LM_MAPPINGZ&FLAX_MODEL_FOR_MULTIPLE_CHOICE_MAPPINGZ/FLAX_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGZ"FLAX_MODEL_FOR_PRETRAINING_MAPPINGZ)FLAX_MODEL_FOR_QUESTION_ANSWERING_MAPPINGZ+FLAX_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGZ.FLAX_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGZ'FLAX_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGZ+FLAX_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGZ#FLAX_MODEL_FOR_VISION_2_SEQ_MAPPINGZFLAX_MODEL_MAPPINGr*   r+   r,   r-   r.   r/   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   Z.FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   r   <module>   s  