U
    9%e                    @   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d'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G d1d2 d2edZd3d4 Zd5Z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Z#G dDdE dEedZ$G dFdG dGedZ%d5Z&d5Z'd5Z(d5Z)d5Z*d5Z+d5Z,d5Z-d5Z.d5Z/d5Z0d5Z1d5Z2d5Z3d5Z4d5Z5d5Z6d5Z7d5Z8d5Z9d5Z:d5Z;G dHdI dIedZ<G dJdK dKedZ=G dLdM dMedZ>G dNdO dOedZ?G dPdQ dQedZ@G dRdS dSedZAG dTdU dUedZBG dVdW dWedZCG dXdY dYedZDG dZd[ d[edZEG d\d] d]edZFG d^d_ d_edZGG d`da daedZHG dbdc dcedZIG ddde deedZJG dfdg dgedZKG dhdi diedZLG djdk dkedZMG dldm dmedZNG dndo doedZOG dpdq dqedZPG drds dsedZQG dtdu duedZRG dvdw dwedZSG dxdy dyedZTG dzd{ d{edZUd5ZVG d|d} d}edZWG d~d de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Zhd5ZiG 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Zpd5ZqG 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Zyd5ZzG dd dedZ{G dd dedZ|G ddÄ dedZ}G ddń dedZ~d5Z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Zd5ZG dd݄ dedZG dd߄ dedZG dd dedZG dd dedZd5ZG 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Zd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG 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Zd5ZG 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Zd5ZG 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Zd5Zd5Zd5ZG 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Zd5ZG d2d3 d3edZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZd5ZG 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Zd5ZG dLdM dMedZG dNdO dOedZG dPdQ dQedZG dRdS dSedZG dTdU dUedZd5ZG 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Zd5ZG 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Zd5ZG 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d dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZ G dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG ddÄ dedZG dĐdń dedZG dƐdǄ dedZd5ZG dȐdɄ dedZG dʐd˄ dedZG d̐d̈́ dedZG dΐdτ dedZG dАdф dedZG dҐdӄ dedZG dԐdՄ dedZG d֐dׄ dedZG dؐdل dedZG dڐdۄ dedZG dܐd݄ dedZ d5Z!G dސd߄ dedZ"G dd dedZ#G dd dedZ$G dd dedZ%G dd dedZ&G dd dedZ'G dd dedZ(G dd dedZ)G dd dedZ*G dd dedZ+d5Z,G dd dedZ-G dd dedZ.G dd dedZ/G dd dedZ0d5Z1G dd dedZ2G dd dedZ3G dd dedZ4G d d dedZ5G dd dedZ6G dd dedZ7G dd dedZ8G dd	 d	edZ9G d
d dedZ:G dd dedZ;G dd dedZ<d5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ZAG dd dedZBG dd dedZCG dd dedZDG dd dedZEG d d! d!edZFG d"d# d#edZGG d$d% d%edZHG d&d' d'edZIG d(d) d)edZJG d*d+ d+edZKG d,d- d-edZLG d.d/ d/edZMd5ZNG d0d1 d1edZOG d2d3 d3edZPG d4d5 d5edZQd5ZRG d6d7 d7edZSG d8d9 d9edZTG d:d; d;edZUG d<d= d=edZVG d>d? d?edZWG d@dA dAedZXG dBdC dCedZYG dDdE dEedZZG dFdG dGedZ[d5Z\G dHdI dIedZ]G dJdK dKedZ^G dLdM dMedZ_d5Z`G dNdO dOedZaG dPdQ dQedZbG dRdS dSedZcG dTdU dUedZdG dVdW dWedZeG dXdY dYedZfG dZd[ d[edZgG d\d] d]edZhG d^d_ d_edZid5ZjG d`da daedZkG dbdc dcedZlG ddde deedZmG dfdg dgedZnG dhdi diedZoG djdk dkedZpG dldm dmedZqG dndo doedZrG dpdq dqedZsd5ZtG drds dsedZuG dtdu duedZvG dvdw dwedZwG dxdy dyedZxG dzd{ d{edZyG d|d} d}edZzG d~d dedZ{G dd dedZ|G dd dedZ}d5Z~G dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG ddÄ dedZG dĐdń dedZG dƐdǄ dedZG dȐdɄ dedZG dʐd˄ dedZG d̐d̈́ dedZd5ZG dΐdτ dedZG dАdф dedZG dҐdӄ dedZG dԐdՄ dedZd5ZG d֐dׄ dedZG dؐdل dedZG dڐdۄ dedZd5ZG dܐd݄ dedZG dސd߄ dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZd5ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG d d dedZd5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ZԐdd ZG dd dedZd5S (     )DummyObjectrequires_backendsc                   @   s   e Zd ZdgZdd ZdS )TensorFlowBenchmarkArgumentstfc                 O   s   t | dg d S Nr   r   selfargskwargs r   b/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/utils/dummy_tf_objects.py__init__   s    z%TensorFlowBenchmarkArguments.__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 )TensorFlowBenchmarkr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTensorFlowBenchmark.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )TFForcedBOSTokenLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z(TFForcedBOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )TFForcedEOSTokenLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z(TFForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )TFForceTokensLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $   s    z%TFForceTokensLogitsProcessor.__init__Nr   r   r   r   r   r   !   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGenerationMixinr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +   s    zTFGenerationMixin.__init__Nr   r   r   r   r   r   (   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2   s    zTFLogitsProcessor.__init__Nr   r   r   r   r   r   /   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLogitsProcessorListr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9   s    zTFLogitsProcessorList.__init__Nr   r   r   r   r   r   6   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @   s    zTFLogitsWarper.__init__Nr   r   r   r   r   r   =   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMinLengthLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G   s    z#TFMinLengthLogitsProcessor.__init__Nr   r   r   r   r   r   D   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFNoBadWordsLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N   s    z$TFNoBadWordsLogitsProcessor.__init__Nr   r   r   r   r   r   K   s   r   c                   @   s   e Zd ZdgZdd ZdS )TFNoRepeatNGramLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U   s    z'TFNoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   r   R   s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFRepetitionPenaltyLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \   s    z+TFRepetitionPenaltyLogitsProcessor.__init__Nr   r   r   r   r   r   Y   s   r   c                   @   s   e Zd ZdgZdd ZdS )&TFSuppressTokensAtBeginLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c   s    z/TFSuppressTokensAtBeginLogitsProcessor.__init__Nr   r   r   r   r   r    `   s   r    c                   @   s   e Zd ZdgZdd ZdS )TFSuppressTokensLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j   s    z(TFSuppressTokensLogitsProcessor.__init__Nr   r   r   r   r   r!   g   s   r!   c                   @   s   e Zd ZdgZdd ZdS )TFTemperatureLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q   s    z"TFTemperatureLogitsWarper.__init__Nr   r   r   r   r   r"   n   s   r"   c                   @   s   e Zd ZdgZdd ZdS )TFTopKLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x   s    zTFTopKLogitsWarper.__init__Nr   r   r   r   r   r#   u   s   r#   c                   @   s   e Zd ZdgZdd ZdS )TFTopPLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFTopPLogitsWarper.__init__Nr   r   r   r   r   r$   |   s   r$   c                  O   s   t tdg d S r   )r   tf_top_k_top_p_filteringr
   r   r   r   r   r%      s    r%   c                   @   s   e Zd ZdgZdd ZdS )KerasMetricCallbackr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zKerasMetricCallback.__init__Nr   r   r   r   r   r'      s   r'   c                   @   s   e Zd ZdgZdd ZdS )PushToHubCallbackr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zPushToHubCallback.__init__Nr   r   r   r   r   r(      s   r(   c                   @   s   e Zd ZdgZdd ZdS )TFPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFPreTrainedModel.__init__Nr   r   r   r   r   r)      s   r)   c                   @   s   e Zd ZdgZdd ZdS )TFSequenceSummaryr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFSequenceSummary.__init__Nr   r   r   r   r   r*      s   r*   c                   @   s   e Zd ZdgZdd ZdS )TFSharedEmbeddingsr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFSharedEmbeddings.__init__Nr   r   r   r   r   r+      s   r+   c                  O   s   t tdg d S r   )r   
shape_listr&   r   r   r   r,      s    r,   Nc                   @   s   e Zd ZdgZdd ZdS )TFAlbertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFAlbertForMaskedLM.__init__Nr   r   r   r   r   r-      s   r-   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"TFAlbertForMultipleChoice.__init__Nr   r   r   r   r   r.      s   r.   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFAlbertForPreTraining.__init__Nr   r   r   r   r   r/      s   r/   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z%TFAlbertForQuestionAnswering.__init__Nr   r   r   r   r   r0      s   r0   c                   @   s   e Zd ZdgZdd ZdS )!TFAlbertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z*TFAlbertForSequenceClassification.__init__Nr   r   r   r   r   r1      s   r1   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z'TFAlbertForTokenClassification.__init__Nr   r   r   r   r   r2      s   r2   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFAlbertMainLayer.__init__Nr   r   r   r   r   r3      s   r3   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFAlbertModel.__init__Nr   r   r   r   r   r4      s   r4   c                   @   s   e Zd ZdgZdd ZdS )TFAlbertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z TFAlbertPreTrainedModel.__init__Nr   r   r   r   r   r5      s   r5   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    zTFAutoModel.__init__Nr   r   r   r   r   r6   2  s   r6   c                   @   s   e Zd ZdgZdd ZdS )!TFAutoModelForAudioClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    z*TFAutoModelForAudioClassification.__init__Nr   r   r   r   r   r7   9  s   r7   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    zTFAutoModelForCausalLM.__init__Nr   r   r   r   r   r8   @  s   r8   c                   @   s   e Zd ZdgZdd ZdS )'TFAutoModelForDocumentQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    z0TFAutoModelForDocumentQuestionAnswering.__init__Nr   r   r   r   r   r9   G  s   r9   c                   @   s   e Zd ZdgZdd ZdS )!TFAutoModelForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    z*TFAutoModelForImageClassification.__init__Nr   r   r   r   r   r:   N  s   r:   c                   @   s   e Zd ZdgZdd ZdS )!TFAutoModelForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   X  s    z*TFAutoModelForMaskedImageModeling.__init__Nr   r   r   r   r   r;   U  s   r;   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   _  s    zTFAutoModelForMaskedLM.__init__Nr   r   r   r   r   r<   \  s   r<   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForMaskGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   f  s    z%TFAutoModelForMaskGeneration.__init__Nr   r   r   r   r   r=   c  s   r=   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   m  s    z%TFAutoModelForMultipleChoice.__init__Nr   r   r   r   r   r>   j  s   r>   c                   @   s   e Zd ZdgZdd ZdS )$TFAutoModelForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   t  s    z-TFAutoModelForNextSentencePrediction.__init__Nr   r   r   r   r   r?   q  s   r?   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    z"TFAutoModelForPreTraining.__init__Nr   r   r   r   r   r@   x  s   r@   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFAutoModelForQuestionAnswering.__init__Nr   r   r   r   r   rA     s   rA   c                   @   s   e Zd ZdgZdd ZdS )"TFAutoModelForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TFAutoModelForSemanticSegmentation.__init__Nr   r   r   r   r   rB     s   rB   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForSeq2SeqLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFAutoModelForSeq2SeqLM.__init__Nr   r   r   r   r   rC     s   rC   c                   @   s   e Zd ZdgZdd ZdS )$TFAutoModelForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TFAutoModelForSequenceClassification.__init__Nr   r   r   r   r   rD     s   rD   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForSpeechSeq2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFAutoModelForSpeechSeq2Seq.__init__Nr   r   r   r   r   rE     s   rE   c                   @   s   e Zd ZdgZdd ZdS )$TFAutoModelForTableQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TFAutoModelForTableQuestionAnswering.__init__Nr   r   r   r   r   rF     s   rF   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForTextEncodingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFAutoModelForTextEncoding.__init__Nr   r   r   r   r   rG     s   rG   c                   @   s   e Zd ZdgZdd ZdS )!TFAutoModelForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*TFAutoModelForTokenClassification.__init__Nr   r   r   r   r   rH     s   rH   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelForVision2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFAutoModelForVision2Seq.__init__Nr   r   r   r   r   rI     s   rI   c                   @   s   e Zd ZdgZdd ZdS ))TFAutoModelForZeroShotImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z2TFAutoModelForZeroShotImageClassification.__init__Nr   r   r   r   r   rJ     s   rJ   c                   @   s   e Zd ZdgZdd ZdS )TFAutoModelWithLMHeadr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFAutoModelWithLMHead.__init__Nr   r   r   r   r   rK     s   rK   c                   @   s   e Zd ZdgZdd ZdS )TFBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'TFBartForConditionalGeneration.__init__Nr   r   r   r   r   rL     s   rL   c                   @   s   e Zd ZdgZdd ZdS )TFBartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFBartForSequenceClassification.__init__Nr   r   r   r   r   rM     s   rM   c                   @   s   e Zd ZdgZdd ZdS )TFBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBartModel.__init__Nr   r   r   r   r   rN     s   rN   c                   @   s   e Zd ZdgZdd ZdS )TFBartPretrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBartPretrainedModel.__init__Nr   r   r   r   r   rO     s   rO   c                   @   s   e Zd ZdgZdd ZdS )TFBertEmbeddingsr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBertEmbeddings.__init__Nr   r   r   r   r   rP     s   rP   c                   @   s   e Zd ZdgZdd ZdS )TFBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBertForMaskedLM.__init__Nr   r   r   r   r   rQ     s   rQ   c                   @   s   e Zd ZdgZdd ZdS )TFBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFBertForMultipleChoice.__init__Nr   r   r   r   r   rR     s   rR   c                   @   s   e Zd ZdgZdd ZdS )TFBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFBertForNextSentencePrediction.__init__Nr   r   r   r   r   rS      s   rS   c                   @   s   e Zd ZdgZdd ZdS )TFBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFBertForPreTraining.__init__Nr   r   r   r   r   rT     s   rT   c                   @   s   e Zd ZdgZdd ZdS )TFBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFBertForQuestionAnswering.__init__Nr   r   r   r   r   rU     s   rU   c                   @   s   e Zd ZdgZdd ZdS )TFBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFBertForSequenceClassification.__init__Nr   r   r   r   r   rV     s   rV   c                   @   s   e Zd ZdgZdd ZdS )TFBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFBertForTokenClassification.__init__Nr   r   r   r   r   rW     s   rW   c                   @   s   e Zd ZdgZdd ZdS )TFBertLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zTFBertLMHeadModel.__init__Nr   r   r   r   r   rX   #  s   rX   c                   @   s   e Zd ZdgZdd ZdS )TFBertMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    zTFBertMainLayer.__init__Nr   r   r   r   r   rY   *  s   rY   c                   @   s   e Zd ZdgZdd ZdS )TFBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    zTFBertModel.__init__Nr   r   r   r   r   rZ   1  s   rZ   c                   @   s   e Zd ZdgZdd ZdS )TFBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zTFBertPreTrainedModel.__init__Nr   r   r   r   r   r[   8  s   r[   c                   @   s   e Zd ZdgZdd ZdS )$TFBlenderbotForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    z-TFBlenderbotForConditionalGeneration.__init__Nr   r   r   r   r   r\   ?  s   r\   c                   @   s   e Zd ZdgZdd ZdS )TFBlenderbotModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    zTFBlenderbotModel.__init__Nr   r   r   r   r   r]   F  s   r]   c                   @   s   e Zd ZdgZdd ZdS )TFBlenderbotPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    z$TFBlenderbotPreTrainedModel.__init__Nr   r   r   r   r   r^   M  s   r^   c                   @   s   e Zd ZdgZdd ZdS ))TFBlenderbotSmallForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    z2TFBlenderbotSmallForConditionalGeneration.__init__Nr   r   r   r   r   r_   T  s   r_   c                   @   s   e Zd ZdgZdd ZdS )TFBlenderbotSmallModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    zTFBlenderbotSmallModel.__init__Nr   r   r   r   r   r`   [  s   r`   c                   @   s   e Zd ZdgZdd ZdS ) TFBlenderbotSmallPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e  s    z)TFBlenderbotSmallPreTrainedModel.__init__Nr   r   r   r   r   ra   b  s   ra   c                   @   s   e Zd ZdgZdd ZdS )TFBlipForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    z'TFBlipForConditionalGeneration.__init__Nr   r   r   r   r   rb   l  s   rb   c                   @   s   e Zd ZdgZdd ZdS )TFBlipForImageTextRetrievalr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z$TFBlipForImageTextRetrieval.__init__Nr   r   r   r   r   rc   s  s   rc   c                   @   s   e Zd ZdgZdd ZdS )TFBlipForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    z#TFBlipForQuestionAnswering.__init__Nr   r   r   r   r   rd   z  s   rd   c                   @   s   e Zd ZdgZdd ZdS )TFBlipModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBlipModel.__init__Nr   r   r   r   r   re     s   re   c                   @   s   e Zd ZdgZdd ZdS )TFBlipPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBlipPreTrainedModel.__init__Nr   r   r   r   r   rf     s   rf   c                   @   s   e Zd ZdgZdd ZdS )TFBlipTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBlipTextModel.__init__Nr   r   r   r   r   rg     s   rg   c                   @   s   e Zd ZdgZdd ZdS )TFBlipVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFBlipVisionModel.__init__Nr   r   r   r   r   rh     s   rh   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCamembertForCausalLM.__init__Nr   r   r   r   r   ri     s   ri   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCamembertForMaskedLM.__init__Nr   r   r   r   r   rj     s   rj   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFCamembertForMultipleChoice.__init__Nr   r   r   r   r   rk     s   rk   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFCamembertForQuestionAnswering.__init__Nr   r   r   r   r   rl     s   rl   c                   @   s   e Zd ZdgZdd ZdS )$TFCamembertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TFCamembertForSequenceClassification.__init__Nr   r   r   r   r   rm     s   rm   c                   @   s   e Zd ZdgZdd ZdS )!TFCamembertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*TFCamembertForTokenClassification.__init__Nr   r   r   r   r   rn     s   rn   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCamembertModel.__init__Nr   r   r   r   r   ro     s   ro   c                   @   s   e Zd ZdgZdd ZdS )TFCamembertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFCamembertPreTrainedModel.__init__Nr   r   r   r   r   rp     s   rp   c                   @   s   e Zd ZdgZdd ZdS )TFCLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCLIPModel.__init__Nr   r   r   r   r   rq     s   rq   c                   @   s   e Zd ZdgZdd ZdS )TFCLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCLIPPreTrainedModel.__init__Nr   r   r   r   r   rr     s   rr   c                   @   s   e Zd ZdgZdd ZdS )TFCLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCLIPTextModel.__init__Nr   r   r   r   r   rs     s   rs   c                   @   s   e Zd ZdgZdd ZdS )TFCLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFCLIPVisionModel.__init__Nr   r   r   r   r   rt     s   rt   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFConvBertForMaskedLM.__init__Nr   r   r   r   r   ru     s   ru   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFConvBertForMultipleChoice.__init__Nr   r   r   r   r   rv     s   rv   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'TFConvBertForQuestionAnswering.__init__Nr   r   r   r   r   rw     s   rw   c                   @   s   e Zd ZdgZdd ZdS )#TFConvBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,TFConvBertForSequenceClassification.__init__Nr   r   r   r   r   rx     s   rx   c                   @   s   e Zd ZdgZdd ZdS ) TFConvBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFConvBertForTokenClassification.__init__Nr   r   r   r   r   ry     s   ry   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFConvBertLayer.__init__Nr   r   r   r   r   rz     s   rz   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    zTFConvBertModel.__init__Nr   r   r   r   r   r{   $  s   r{   c                   @   s   e Zd ZdgZdd ZdS )TFConvBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    z"TFConvBertPreTrainedModel.__init__Nr   r   r   r   r   r|   +  s   r|   c                   @   s   e Zd ZdgZdd ZdS ) TFConvNextForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    z)TFConvNextForImageClassification.__init__Nr   r   r   r   r   r}   2  s   r}   c                   @   s   e Zd ZdgZdd ZdS )TFConvNextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    zTFConvNextModel.__init__Nr   r   r   r   r   r~   9  s   r~   c                   @   s   e Zd ZdgZdd ZdS )TFConvNextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    z"TFConvNextPreTrainedModel.__init__Nr   r   r   r   r   r   @  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCTRLForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   M  s    z(TFCTRLForSequenceClassification.__init__Nr   r   r   r   r   r   J  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCTRLLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   T  s    zTFCTRLLMHeadModel.__init__Nr   r   r   r   r   r   Q  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCTRLModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [  s    zTFCTRLModel.__init__Nr   r   r   r   r   r   X  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCTRLPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b  s    zTFCTRLPreTrainedModel.__init__Nr   r   r   r   r   r   _  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCvtForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   l  s    z$TFCvtForImageClassification.__init__Nr   r   r   r   r   r   i  s   r   c                   @   s   e Zd ZdgZdd ZdS )
TFCvtModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   s  s    zTFCvtModel.__init__Nr   r   r   r   r   r   p  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFCvtPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   z  s    zTFCvtPreTrainedModel.__init__Nr   r   r   r   r   r   w  s   r   c                   @   s   e Zd ZdgZdd ZdS )&TFData2VecVisionForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/TFData2VecVisionForImageClassification.__init__Nr   r   r   r   r   r   ~  s   r   c                   @   s   e Zd ZdgZdd ZdS )'TFData2VecVisionForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0TFData2VecVisionForSemanticSegmentation.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFData2VecVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFData2VecVisionModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFData2VecVisionPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFData2VecVisionPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDebertaForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFDebertaForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFDebertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TFDebertaForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFDebertaForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDebertaModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFDebertaPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaV2ForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDebertaV2ForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaV2ForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFDebertaV2ForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaV2ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFDebertaV2ForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )$TFDebertaV2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TFDebertaV2ForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )!TFDebertaV2ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*TFDebertaV2ForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaV2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDebertaV2Model.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDebertaV2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFDebertaV2PreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDeiTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFDeiTForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )'TFDeiTForImageClassificationWithTeacherr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0TFDeiTForImageClassificationWithTeacher.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDeiTForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFDeiTForMaskedImageModeling.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDeiTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDeiTModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDeiTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDeiTPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDistilBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    z TFDistilBertForMaskedLM.__init__Nr   r   r   r   r   r   $  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDistilBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    z&TFDistilBertForMultipleChoice.__init__Nr   r   r   r   r   r   +  s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFDistilBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    z)TFDistilBertForQuestionAnswering.__init__Nr   r   r   r   r   r   2  s   r   c                   @   s   e Zd ZdgZdd ZdS )%TFDistilBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    z.TFDistilBertForSequenceClassification.__init__Nr   r   r   r   r   r   9  s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFDistilBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    z+TFDistilBertForTokenClassification.__init__Nr   r   r   r   r   r   @  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDistilBertMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    zTFDistilBertMainLayer.__init__Nr   r   r   r   r   r   G  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDistilBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    zTFDistilBertModel.__init__Nr   r   r   r   r   r   N  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDistilBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   X  s    z$TFDistilBertPreTrainedModel.__init__Nr   r   r   r   r   r   U  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRContextEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    zTFDPRContextEncoder.__init__Nr   r   r   r   r   r   e  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRPretrainedContextEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    z&TFDPRPretrainedContextEncoder.__init__Nr   r   r   r   r   r   l  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRPretrainedQuestionEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z'TFDPRPretrainedQuestionEncoder.__init__Nr   r   r   r   r   r   s  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRPretrainedReaderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    zTFDPRPretrainedReader.__init__Nr   r   r   r   r   r   z  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRQuestionEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDPRQuestionEncoder.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFDPRReaderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFDPRReader.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )'TFEfficientFormerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0TFEfficientFormerForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )2TFEfficientFormerForImageClassificationWithTeacherr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z;TFEfficientFormerForImageClassificationWithTeacher.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEfficientFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFEfficientFormerModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFEfficientFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFEfficientFormerPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFElectraForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFElectraForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFElectraForPreTraining.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFElectraForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFElectraForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TFElectraForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFElectraForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFElectraModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFElectraPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFElectraPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFEncoderDecoderModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEsmForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFEsmForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEsmForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'TFEsmForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEsmForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFEsmForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )
TFEsmModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFEsmModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFEsmPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFEsmPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFlaubertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFFlaubertForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )$TFFlaubertForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    z-TFFlaubertForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )#TFFlaubertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    z,TFFlaubertForSequenceClassification.__init__Nr   r   r   r   r   r   '  s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFFlaubertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   1  s    z)TFFlaubertForTokenClassification.__init__Nr   r   r   r   r   r   .  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFlaubertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   8  s    zTFFlaubertModel.__init__Nr   r   r   r   r   r   5  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFlaubertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ?  s    z"TFFlaubertPreTrainedModel.__init__Nr   r   r   r   r   r   <  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFlaubertWithLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   F  s    z"TFFlaubertWithLMHeadModel.__init__Nr   r   r   r   r   r   C  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelBaseModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    zTFFunnelBaseModel.__init__Nr   r   r   r   r   r   M  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    zTFFunnelForMaskedLM.__init__Nr   r   r   r   r   r   T  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    z"TFFunnelForMultipleChoice.__init__Nr   r   r   r   r   r   [  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e  s    zTFFunnelForPreTraining.__init__Nr   r   r   r   r   r   b  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   l  s    z%TFFunnelForQuestionAnswering.__init__Nr   r   r   r   r   r   i  s   r   c                   @   s   e Zd ZdgZdd ZdS )!TFFunnelForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   s  s    z*TFFunnelForSequenceClassification.__init__Nr   r   r   r   r   r   p  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   z  s    z'TFFunnelForTokenClassification.__init__Nr   r   r   r   r   r   w  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFFunnelModel.__init__Nr   r   r   r   r   r   ~  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFFunnelPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFFunnelPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2DoubleHeadsModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPT2DoubleHeadsModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFGPT2ForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2LMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPT2LMHeadModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2MainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPT2MainLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPT2Model.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPT2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPT2PreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPTJForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPTJForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPTJForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFGPTJForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPTJForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFGPTJForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPTJModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPTJModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGPTJPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGPTJPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGroupViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGroupViTModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGroupViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"TFGroupViTPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGroupViTTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGroupViTTextModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFGroupViTVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFGroupViTVisionModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFHubertForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFHubertForCTC.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFHubertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFHubertModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFHubertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFHubertPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFLayoutLMForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z'TFLayoutLMForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )#TFLayoutLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    z,TFLayoutLMForSequenceClassification.__init__Nr   r   r   r   r   r   $  s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFLayoutLMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    z)TFLayoutLMForTokenClassification.__init__Nr   r   r   r   r   r   +  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    zTFLayoutLMMainLayer.__init__Nr   r   r   r   r   r   2  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    zTFLayoutLMModel.__init__Nr   r   r   r   r   r   9  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    z"TFLayoutLMPreTrainedModel.__init__Nr   r   r   r   r   r   @  s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFLayoutLMv3ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   M  s    z)TFLayoutLMv3ForQuestionAnswering.__init__Nr   r   r   r   r   r   J  s   r   c                   @   s   e Zd ZdgZdd ZdS )%TFLayoutLMv3ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   T  s    z.TFLayoutLMv3ForSequenceClassification.__init__Nr   r   r   r   r   r   Q  s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFLayoutLMv3ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [  s    z+TFLayoutLMv3ForTokenClassification.__init__Nr   r   r   r   r   r   X  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMv3Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b  s    zTFLayoutLMv3Model.__init__Nr   r   r   r   r   r   _  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLayoutLMv3PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   i  s    z$TFLayoutLMv3PreTrainedModel.__init__Nr   r   r   r   r   r   f  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLEDForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p  s    z&TFLEDForConditionalGeneration.__init__Nr   r   r   r   r   r   m  s   r   c                   @   s   e Zd ZdgZdd ZdS )
TFLEDModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   w  s    zTFLEDModel.__init__Nr   r   r   r   r   r   t  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLEDPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ~  s    zTFLEDPreTrainedModel.__init__Nr   r   r   r   r   r   {  s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLongformerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFLongformerForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLongformerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFLongformerForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS ) TFLongformerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFLongformerForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )%TFLongformerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.TFLongformerForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"TFLongformerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TFLongformerForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLongformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFLongformerModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLongformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFLongformerPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLongformerSelfAttentionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"TFLongformerSelfAttention.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLxmertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFLxmertForPreTraining.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLxmertMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFLxmertMainLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLxmertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFLxmertModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLxmertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFLxmertPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFLxmertVisualFeatureEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%TFLxmertVisualFeatureEncoder.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMarianModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMarianModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMarianMTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMarianMTModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMarianPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFMarianPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFMBartForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMBartModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMBartPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFMBartPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFMobileBertForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFMobileBertForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )%TFMobileBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   !  s    z.TFMobileBertForNextSentencePrediction.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   (  s    z#TFMobileBertForPreTraining.__init__Nr   r   r   r   r   r  %  s   r  c                   @   s   e Zd ZdgZdd ZdS ) TFMobileBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   /  s    z)TFMobileBertForQuestionAnswering.__init__Nr   r   r   r   r   r  ,  s   r  c                   @   s   e Zd ZdgZdd ZdS )%TFMobileBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   6  s    z.TFMobileBertForSequenceClassification.__init__Nr   r   r   r   r   r  3  s   r  c                   @   s   e Zd ZdgZdd ZdS )"TFMobileBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   =  s    z+TFMobileBertForTokenClassification.__init__Nr   r   r   r   r   r  :  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   D  s    zTFMobileBertMainLayer.__init__Nr   r   r   r   r   r  A  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   K  s    zTFMobileBertModel.__init__Nr   r   r   r   r   r	  H  s   r	  c                   @   s   e Zd ZdgZdd ZdS )TFMobileBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    z$TFMobileBertPreTrainedModel.__init__Nr   r   r   r   r   r
  O  s   r
  c                   @   s   e Zd ZdgZdd ZdS )!TFMobileViTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    z*TFMobileViTForImageClassification.__init__Nr   r   r   r   r   r  Y  s   r  c                   @   s   e Zd ZdgZdd ZdS )"TFMobileViTForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c  s    z+TFMobileViTForSemanticSegmentation.__init__Nr   r   r   r   r   r  `  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j  s    zTFMobileViTModel.__init__Nr   r   r   r   r   r  g  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMobileViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q  s    z#TFMobileViTPreTrainedModel.__init__Nr   r   r   r   r   r  n  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    zTFMPNetForMaskedLM.__init__Nr   r   r   r   r   r  x  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFMPNetForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFMPNetForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) TFMPNetForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFMPNetForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFMPNetForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMPNetMainLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMPNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMPNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMPNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMT5EncoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFMT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFMT5ForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )
TFMT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFMT5Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOpenAIGPTDoubleHeadsModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$TFOpenAIGPTDoubleHeadsModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )$TFOpenAIGPTForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TFOpenAIGPTForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOpenAIGPTLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOpenAIGPTLMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOpenAIGPTMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOpenAIGPTMainLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOpenAIGPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOpenAIGPTModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOpenAIGPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFOpenAIGPTPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFOPTForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOPTForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )
TFOPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOPTModel.__init__Nr   r   r   r   r   r!    s   r!  c                   @   s   e Zd ZdgZdd ZdS )TFOPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFOPTPreTrainedModel.__init__Nr   r   r   r   r   r"     s   r"  c                   @   s   e Zd ZdgZdd ZdS )!TFPegasusForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z*TFPegasusForConditionalGeneration.__init__Nr   r   r   r   r   r#    s   r#  c                   @   s   e Zd ZdgZdd ZdS )TFPegasusModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFPegasusModel.__init__Nr   r   r   r   r   r$    s   r$  c                   @   s   e Zd ZdgZdd ZdS )TFPegasusPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFPegasusPreTrainedModel.__init__Nr   r   r   r   r   r%    s   r%  c                   @   s   e Zd ZdgZdd ZdS )
TFRagModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRagModel.__init__Nr   r   r   r   r   r&    s   r&  c                   @   s   e Zd ZdgZdd ZdS )TFRagPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zTFRagPreTrainedModel.__init__Nr   r   r   r   r   r'  #  s   r'  c                   @   s   e Zd ZdgZdd ZdS )TFRagSequenceForGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z#TFRagSequenceForGeneration.__init__Nr   r   r   r   r   r(  *  s   r(  c                   @   s   e Zd ZdgZdd ZdS )TFRagTokenForGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z TFRagTokenForGeneration.__init__Nr   r   r   r   r   r)  1  s   r)  c                   @   s   e Zd ZdgZdd ZdS )TFRegNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    z'TFRegNetForImageClassification.__init__Nr   r   r   r   r   r*  ;  s   r*  c                   @   s   e Zd ZdgZdd ZdS )TFRegNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    zTFRegNetModel.__init__Nr   r   r   r   r   r+  B  s   r+  c                   @   s   e Zd ZdgZdd ZdS )TFRegNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    z TFRegNetPreTrainedModel.__init__Nr   r   r   r   r   r,  I  s   r,  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   V  s    zTFRemBertForCausalLM.__init__Nr   r   r   r   r   r-  S  s   r-  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ]  s    zTFRemBertForMaskedLM.__init__Nr   r   r   r   r   r.  Z  s   r.  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   d  s    z#TFRemBertForMultipleChoice.__init__Nr   r   r   r   r   r/  a  s   r/  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   k  s    z&TFRemBertForQuestionAnswering.__init__Nr   r   r   r   r   r0  h  s   r0  c                   @   s   e Zd ZdgZdd ZdS )"TFRemBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   r  s    z+TFRemBertForSequenceClassification.__init__Nr   r   r   r   r   r1  o  s   r1  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    z(TFRemBertForTokenClassification.__init__Nr   r   r   r   r   r2  v  s   r2  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRemBertLayer.__init__Nr   r   r   r   r   r3  }  s   r3  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRemBertModel.__init__Nr   r   r   r   r   r4    s   r4  c                   @   s   e Zd ZdgZdd ZdS )TFRemBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFRemBertPreTrainedModel.__init__Nr   r   r   r   r   r5    s   r5  c                   @   s   e Zd ZdgZdd ZdS )TFResNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'TFResNetForImageClassification.__init__Nr   r   r   r   r   r6    s   r6  c                   @   s   e Zd ZdgZdd ZdS )TFResNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFResNetModel.__init__Nr   r   r   r   r   r7    s   r7  c                   @   s   e Zd ZdgZdd ZdS )TFResNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFResNetPreTrainedModel.__init__Nr   r   r   r   r   r8    s   r8  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRobertaForCausalLM.__init__Nr   r   r   r   r   r9    s   r9  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRobertaForMaskedLM.__init__Nr   r   r   r   r   r:    s   r:  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TFRobertaForMultipleChoice.__init__Nr   r   r   r   r   r;    s   r;  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&TFRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r<    s   r<  c                   @   s   e Zd ZdgZdd ZdS )"TFRobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TFRobertaForSequenceClassification.__init__Nr   r   r   r   r   r=    s   r=  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(TFRobertaForTokenClassification.__init__Nr   r   r   r   r   r>    s   r>  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRobertaMainLayer.__init__Nr   r   r   r   r   r?    s   r?  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFRobertaModel.__init__Nr   r   r   r   r   r@    s   r@  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TFRobertaPreTrainedModel.__init__Nr   r   r   r   r   rA    s   rA  c                   @   s   e Zd ZdgZdd ZdS ) TFRobertaPreLayerNormForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFRobertaPreLayerNormForCausalLM.__init__Nr   r   r   r   r   rB    s   rB  c                   @   s   e Zd ZdgZdd ZdS ) TFRobertaPreLayerNormForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TFRobertaPreLayerNormForMaskedLM.__init__Nr   r   r   r   r   rC    s   rC  c                   @   s   e Zd ZdgZdd ZdS )&TFRobertaPreLayerNormForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r    	  s    z/TFRobertaPreLayerNormForMultipleChoice.__init__Nr   r   r   r   r   rD    s   rD  c                   @   s   e Zd ZdgZdd ZdS ))TFRobertaPreLayerNormForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z2TFRobertaPreLayerNormForQuestionAnswering.__init__Nr   r   r   r   r   rE  	  s   rE  c                   @   s   e Zd ZdgZdd ZdS ).TFRobertaPreLayerNormForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z7TFRobertaPreLayerNormForSequenceClassification.__init__Nr   r   r   r   r   rF  	  s   rF  c                   @   s   e Zd ZdgZdd ZdS )+TFRobertaPreLayerNormForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z4TFRobertaPreLayerNormForTokenClassification.__init__Nr   r   r   r   r   rG  	  s   rG  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaPreLayerNormMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z'TFRobertaPreLayerNormMainLayer.__init__Nr   r   r   r   r   rH  	  s   rH  c                   @   s   e Zd ZdgZdd ZdS )TFRobertaPreLayerNormModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #	  s    z#TFRobertaPreLayerNormModel.__init__Nr   r   r   r   r   rI   	  s   rI  c                   @   s   e Zd ZdgZdd ZdS )$TFRobertaPreLayerNormPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *	  s    z-TFRobertaPreLayerNormPreTrainedModel.__init__Nr   r   r   r   r   rJ  '	  s   rJ  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4	  s    zTFRoFormerForCausalLM.__init__Nr   r   r   r   r   rK  1	  s   rK  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;	  s    zTFRoFormerForMaskedLM.__init__Nr   r   r   r   r   rL  8	  s   rL  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   B	  s    z$TFRoFormerForMultipleChoice.__init__Nr   r   r   r   r   rM  ?	  s   rM  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I	  s    z'TFRoFormerForQuestionAnswering.__init__Nr   r   r   r   r   rN  F	  s   rN  c                   @   s   e Zd ZdgZdd ZdS )#TFRoFormerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P	  s    z,TFRoFormerForSequenceClassification.__init__Nr   r   r   r   r   rO  M	  s   rO  c                   @   s   e Zd ZdgZdd ZdS ) TFRoFormerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W	  s    z)TFRoFormerForTokenClassification.__init__Nr   r   r   r   r   rP  T	  s   rP  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^	  s    zTFRoFormerLayer.__init__Nr   r   r   r   r   rQ  [	  s   rQ  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e	  s    zTFRoFormerModel.__init__Nr   r   r   r   r   rR  b	  s   rR  c                   @   s   e Zd ZdgZdd ZdS )TFRoFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   l	  s    z"TFRoFormerPreTrainedModel.__init__Nr   r   r   r   r   rS  i	  s   rS  c                   @   s   e Zd ZdgZdd ZdS )
TFSamModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v	  s    zTFSamModel.__init__Nr   r   r   r   r   rT  s	  s   rT  c                   @   s   e Zd ZdgZdd ZdS )TFSamPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }	  s    zTFSamPreTrainedModel.__init__Nr   r   r   r   r   rU  z	  s   rU  c                   @   s   e Zd ZdgZdd ZdS )TFSegformerDecodeHeadr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFSegformerDecodeHead.__init__Nr   r   r   r   r   rV  	  s   rV  c                   @   s   e Zd ZdgZdd ZdS )!TFSegformerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z*TFSegformerForImageClassification.__init__Nr   r   r   r   r   rW  	  s   rW  c                   @   s   e Zd ZdgZdd ZdS )"TFSegformerForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z+TFSegformerForSemanticSegmentation.__init__Nr   r   r   r   r   rX  	  s   rX  c                   @   s   e Zd ZdgZdd ZdS )TFSegformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFSegformerModel.__init__Nr   r   r   r   r   rY  	  s   rY  c                   @   s   e Zd ZdgZdd ZdS )TFSegformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z#TFSegformerPreTrainedModel.__init__Nr   r   r   r   r   rZ  	  s   rZ  c                   @   s   e Zd ZdgZdd ZdS )%TFSpeech2TextForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z.TFSpeech2TextForConditionalGeneration.__init__Nr   r   r   r   r   r[  	  s   r[  c                   @   s   e Zd ZdgZdd ZdS )TFSpeech2TextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFSpeech2TextModel.__init__Nr   r   r   r   r   r\  	  s   r\  c                   @   s   e Zd ZdgZdd ZdS )TFSpeech2TextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z%TFSpeech2TextPreTrainedModel.__init__Nr   r   r   r   r   r]  	  s   r]  c                   @   s   e Zd ZdgZdd ZdS )TFSwinForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z%TFSwinForImageClassification.__init__Nr   r   r   r   r   r^  	  s   r^  c                   @   s   e Zd ZdgZdd ZdS )TFSwinForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z%TFSwinForMaskedImageModeling.__init__Nr   r   r   r   r   r_  	  s   r_  c                   @   s   e Zd ZdgZdd ZdS )TFSwinModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFSwinModel.__init__Nr   r   r   r   r   r`  	  s   r`  c                   @   s   e Zd ZdgZdd ZdS )TFSwinPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFSwinPreTrainedModel.__init__Nr   r   r   r   r   ra  	  s   ra  c                   @   s   e Zd ZdgZdd ZdS )TFT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFT5EncoderModel.__init__Nr   r   r   r   r   rb  	  s   rb  c                   @   s   e Zd ZdgZdd ZdS )TFT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z%TFT5ForConditionalGeneration.__init__Nr   r   r   r   r   rc  	  s   rc  c                   @   s   e Zd ZdgZdd ZdS )	TFT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFT5Model.__init__Nr   r   r   r   r   rd  	  s   rd  c                   @   s   e Zd ZdgZdd ZdS )TFT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zTFT5PreTrainedModel.__init__Nr   r   r   r   r   re  	  s   re  c                   @   s   e Zd ZdgZdd ZdS )TFTapasForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFTapasForMaskedLM.__init__Nr   r   r   r   r   rf   
  s   rf  c                   @   s   e Zd ZdgZdd ZdS )TFTapasForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   

  s    z$TFTapasForQuestionAnswering.__init__Nr   r   r   r   r   rg  
  s   rg  c                   @   s   e Zd ZdgZdd ZdS ) TFTapasForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z)TFTapasForSequenceClassification.__init__Nr   r   r   r   r   rh  
  s   rh  c                   @   s   e Zd ZdgZdd ZdS )TFTapasModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFTapasModel.__init__Nr   r   r   r   r   ri  
  s   ri  c                   @   s   e Zd ZdgZdd ZdS )TFTapasPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFTapasPreTrainedModel.__init__Nr   r   r   r   r   rj  
  s   rj  c                   @   s   e Zd ZdgZdd ZdS )TFAdaptiveEmbeddingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )
  s    zTFAdaptiveEmbedding.__init__Nr   r   r   r   r   rk  &
  s   rk  c                   @   s   e Zd ZdgZdd ZdS )$TFTransfoXLForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   0
  s    z-TFTransfoXLForSequenceClassification.__init__Nr   r   r   r   r   rl  -
  s   rl  c                   @   s   e Zd ZdgZdd ZdS )TFTransfoXLLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7
  s    zTFTransfoXLLMHeadModel.__init__Nr   r   r   r   r   rm  4
  s   rm  c                   @   s   e Zd ZdgZdd ZdS )TFTransfoXLMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >
  s    zTFTransfoXLMainLayer.__init__Nr   r   r   r   r   rn  ;
  s   rn  c                   @   s   e Zd ZdgZdd ZdS )TFTransfoXLModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E
  s    zTFTransfoXLModel.__init__Nr   r   r   r   r   ro  B
  s   ro  c                   @   s   e Zd ZdgZdd ZdS )TFTransfoXLPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L
  s    z#TFTransfoXLPreTrainedModel.__init__Nr   r   r   r   r   rp  I
  s   rp  c                   @   s   e Zd ZdgZdd ZdS )TFVisionEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S
  s    z$TFVisionEncoderDecoderModel.__init__Nr   r   r   r   r   rq  P
  s   rq  c                   @   s   e Zd ZdgZdd ZdS )TFVisionTextDualEncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z
  s    z%TFVisionTextDualEncoderModel.__init__Nr   r   r   r   r   rr  W
  s   rr  c                   @   s   e Zd ZdgZdd ZdS )TFViTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a
  s    z$TFViTForImageClassification.__init__Nr   r   r   r   r   rs  ^
  s   rs  c                   @   s   e Zd ZdgZdd ZdS )
TFViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h
  s    zTFViTModel.__init__Nr   r   r   r   r   rt  e
  s   rt  c                   @   s   e Zd ZdgZdd ZdS )TFViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o
  s    zTFViTPreTrainedModel.__init__Nr   r   r   r   r   ru  l
  s   ru  c                   @   s   e Zd ZdgZdd ZdS )TFViTMAEForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v
  s    zTFViTMAEForPreTraining.__init__Nr   r   r   r   r   rv  s
  s   rv  c                   @   s   e Zd ZdgZdd ZdS )TFViTMAEModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }
  s    zTFViTMAEModel.__init__Nr   r   r   r   r   rw  z
  s   rw  c                   @   s   e Zd ZdgZdd ZdS )TFViTMAEPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z TFViTMAEPreTrainedModel.__init__Nr   r   r   r   r   rx  
  s   rx  c                   @   s   e Zd ZdgZdd ZdS )TFWav2Vec2ForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFWav2Vec2ForCTC.__init__Nr   r   r   r   r   ry  
  s   ry  c                   @   s   e Zd ZdgZdd ZdS )#TFWav2Vec2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z,TFWav2Vec2ForSequenceClassification.__init__Nr   r   r   r   r   rz  
  s   rz  c                   @   s   e Zd ZdgZdd ZdS )TFWav2Vec2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFWav2Vec2Model.__init__Nr   r   r   r   r   r{  
  s   r{  c                   @   s   e Zd ZdgZdd ZdS )TFWav2Vec2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z"TFWav2Vec2PreTrainedModel.__init__Nr   r   r   r   r   r|  
  s   r|  c                   @   s   e Zd ZdgZdd ZdS )!TFWhisperForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z*TFWhisperForConditionalGeneration.__init__Nr   r   r   r   r   r}  
  s   r}  c                   @   s   e Zd ZdgZdd ZdS )TFWhisperModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFWhisperModel.__init__Nr   r   r   r   r   r~  
  s   r~  c                   @   s   e Zd ZdgZdd ZdS )TFWhisperPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z!TFWhisperPreTrainedModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXGLMForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFXGLMForCausalLM.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXGLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFXGLMModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXGLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFXGLMPreTrainedModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFXLMForMultipleChoice.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z(TFXLMForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z'TFXLMForSequenceClassification.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z$TFXLMForTokenClassification.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zTFXLMMainLayer.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )
TFXLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTFXLMModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFXLMPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMWithLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFXLMWithLMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMRobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFXLMRobertaForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMRobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TFXLMRobertaForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMRobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    z&TFXLMRobertaForMultipleChoice.__init__Nr   r   r   r   r   r  #  s   r  c                   @   s   e Zd ZdgZdd ZdS ) TFXLMRobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z)TFXLMRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )%TFXLMRobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z.TFXLMRobertaForSequenceClassification.__init__Nr   r   r   r   r   r  1  s   r  c                   @   s   e Zd ZdgZdd ZdS )"TFXLMRobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    z+TFXLMRobertaForTokenClassification.__init__Nr   r   r   r   r   r  8  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMRobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zTFXLMRobertaModel.__init__Nr   r   r   r   r   r  ?  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLMRobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z$TFXLMRobertaPreTrainedModel.__init__Nr   r   r   r   r   r  F  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z!TFXLNetForMultipleChoice.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )!TFXLNetForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    z*TFXLNetForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS ) TFXLNetForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z)TFXLNetForSequenceClassification.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z&TFXLNetForTokenClassification.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zTFXLNetLMHeadModel.__init__Nr   r   r   r   r   r  l  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetMainLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    zTFXLNetMainLayer.__init__Nr   r   r   r   r   r  s  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    zTFXLNetModel.__init__Nr   r   r   r   r   r  z  s   r  c                   @   s   e Zd ZdgZdd ZdS )TFXLNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFXLNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )AdamWeightDecayr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAdamWeightDecay.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GradientAccumulatorr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGradientAccumulator.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )WarmUpr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWarmUp.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   create_optimizerr&   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )	TFTrainerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTFTrainer.__init__Nr   r   r   r   r   r    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*   r+   r,   Z'TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LISTr-   r.   r/   r0   r1   r2   r3   r4   r5   Z)TF_MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGZTF_MODEL_FOR_CAUSAL_LM_MAPPINGZ0TF_MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGZ)TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGZ$TF_MODEL_FOR_MASK_GENERATION_MAPPINGZ*TF_MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGZTF_MODEL_FOR_MASKED_LM_MAPPINGZ$TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPINGZ-TF_MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGZ TF_MODEL_FOR_PRETRAINING_MAPPINGZ'TF_MODEL_FOR_QUESTION_ANSWERING_MAPPINGZ*TF_MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGZ)TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGZ,TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGZ%TF_MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGZ-TF_MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGZ"TF_MODEL_FOR_TEXT_ENCODING_MAPPINGZ)TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGZ!TF_MODEL_FOR_VISION_2_SEQ_MAPPINGZ3TF_MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGZTF_MODEL_MAPPINGZTF_MODEL_WITH_LM_HEAD_MAPPINGr6   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   Z%TF_BERT_PRETRAINED_MODEL_ARCHIVE_LISTrP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   Z%TF_BLIP_PRETRAINED_MODEL_ARCHIVE_LISTrb   rc   rd   re   rf   rg   rh   Z*TF_CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LISTri   rj   rk   rl   rm   rn   ro   rp   Z%TF_CLIP_PRETRAINED_MODEL_ARCHIVE_LISTrq   rr   rs   rt   Z)TF_CONVBERT_PRETRAINED_MODEL_ARCHIVE_LISTru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   Z%TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z$TF_CVT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z(TF_DEBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   Z+TF_DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z%TF_DEIT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z+TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   Z4TF_DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LISTZ5TF_DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LISTZ+TF_DPR_READER_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   Z0TF_EFFICIENTFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z(TF_ELECTRA_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   Z!ESM_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z)TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z'TF_FUNNEL_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   Z%TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   Z)TF_GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z'TF_HUBERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   Z)TF_LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z+TF_LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   Z+TF_LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   Z'TF_LXMERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   Z+TF_MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r	  r
  Z*TF_MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z&TF_MPNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  r  Z+TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r   r!  r"  r#  r$  r%  r&  r'  r(  r)  Z'TF_REGNET_PRETRAINED_MODEL_ARCHIVE_LISTr*  r+  r,  Z(TF_REMBERT_PRETRAINED_MODEL_ARCHIVE_LISTr-  r.  r/  r0  r1  r2  r3  r4  r5  Z'TF_RESNET_PRETRAINED_MODEL_ARCHIVE_LISTr6  r7  r8  Z(TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr9  r:  r;  r<  r=  r>  r?  r@  rA  Z5TF_ROBERTA_PRELAYERNORM_PRETRAINED_MODEL_ARCHIVE_LISTrB  rC  rD  rE  rF  rG  rH  rI  rJ  Z)TF_ROFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrK  rL  rM  rN  rO  rP  rQ  rR  rS  Z$TF_SAM_PRETRAINED_MODEL_ARCHIVE_LISTrT  rU  Z*TF_SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrV  rW  rX  rY  rZ  Z/TF_SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LISTr[  r\  r]  Z%TF_SWIN_PRETRAINED_MODEL_ARCHIVE_LISTr^  r_  r`  ra  Z#TF_T5_PRETRAINED_MODEL_ARCHIVE_LISTrb  rc  rd  re  Z&TF_TAPAS_PRETRAINED_MODEL_ARCHIVE_LISTrf  rg  rh  ri  rj  Z+TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LISTrk  rl  rm  rn  ro  rp  rq  rr  rs  rt  ru  rv  rw  rx  Z,TF_WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LISTry  rz  r{  r|  Z(TF_WHISPER_PRETRAINED_MODEL_ARCHIVE_LISTr}  r~  r  Z%TF_XGLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z$TF_XLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z,TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z&TF_XLNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  r  r  r  r   r   r   r   <module>   s  