a
    d                    @   sk  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZG d%d& d&edZG d'd( d(edZG d)d* d*edZG d+d, d,edZG d-d. d.edZG d/d0 d0edZG d1d2 d2edZG d3d4 d4edZG d5d6 d6edZG d7d8 d8edZG d9d: 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 dAdB dBedZ"G dCdD dDedZ#G dEdF dFedZ$G dGdH dHedZ%G dIdJ dJedZ&G dKdL dLedZ'G dMdN dNedZ(G dOdP dPedZ)G dQdR dRedZ*dSdT Z+G dUdV dVedZ,dWZ-G dXdY dYedZ.G dZd[ d[edZ/G d\d] d]edZ0G d^d_ d_edZ1G d`da daedZ2G dbdc dcedZ3G ddde deedZ4G dfdg dgedZ5dhdi Z6dWZ7G djdk dkedZ8G dldm dmedZ9G dndo doedZ:G dpdq dqedZ;dWZ<G drds dsedZ=G dtdu duedZ>G dvdw dwedZ?G dxdy dyedZ@dWZAG dzd{ d{edZBG d|d} d}edZCG d~d dedZDdWZEdWZFdWZGdWZHdWZIdWZJdWZKdWZLdWZMdWZNdWZOdWZPdWZQdWZRdWZSdWZTdWZUdWZVdWZWdWZXdWZYdWZZdWZ[dWZ\dWZ]dWZ^dWZ_dWZ`dWZadWZbdWZcdWZddWZeG dd dedZfG dd dedZgG dd dedZhG dd dedZiG dd dedZjG dd dedZkG dd dedZlG dd dedZmG dd dedZnG dd dedZoG dd dedZpG dd dedZqG dd dedZrG dd dedZsG dd dedZtG dd dedZuG dd dedZvG dd dedZwG dd dedZxG dd dedZyG dd dedZzG dd dedZ{G dd dedZ|G dd dedZ}G dd dedZ~G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZdWZG ddÄ dedZG ddń dedZG ddǄ dedZdWZG 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ZdWZG ddׄ dedZG ddل dedZG ddۄ dedZG dd݄ dedZG dd߄ dedZdWZG 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Zdd ZG dd dedZG dd dedZG dd dedZdd ZdWZG 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 ZdWZG 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ZdWZG 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ZdWZG d,d- d-edZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZdWZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZdWZG d<d= d=edZG d>d? d?edZG d@dA dAedZG dBdC dCedZdWZG 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ZdWZG dRdS dSedZG dTdU dUedZG dVdW dWedZG dXdY dYedZG dZd[ d[edZdWZG 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ZdWZG dhdi diedZG djdk dkedZG dldm dmedZG dndo doedZG dpdq dqedZdWZG drds dsedZG dtdu duedZG dvdw dwedZG dxdy dyedZG dzd{ d{edZG d|d} d}edZG d~d dedZG dd dedZdWZG 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Zdd ZdWZG dd dedZ G dd dedZG dd dedZG dd dedZdW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dW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dW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dWZG dd dedZG dd dedZG ddÄ dedZdW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!dW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*dܐd݄ Z+dWZ,G dސd߄ dedZ-G dd dedZ.G dd dedZ/G dd dedZ0dWZ1G dd dedZ2G dd dedZ3G dd dedZ4G dd dedZ5dWZ6G dd dedZ7G dd dedZ8G dd dedZ9dW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>dWZ?G dd dedZ@G dd dedZAG d d dedZBdWZCdWZDdW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ZMG dd dedZNG dd dedZOG dd dedZPG dd dedZQG dd dedZRG dd dedZSG dd dedZTG d d! d!edZUG d"d# d#edZVG d$d% d%edZWdWZXG d&d' d'edZYG d(d) d)edZZG d*d+ d+edZ[G d,d- d-edZ\G d.d/ d/edZ]G d0d1 d1edZ^dWZ_G d2d3 d3edZ`G d4d5 d5edZaG d6d7 d7edZbG d8d9 d9edZcG d:d; d;edZdG d<d= d=edZeG d>d? d?edZfdWZgG d@dA dAedZhG dBdC dCedZiG dDdE dEedZjG dFdG dGedZkdWZlG dHdI dIedZmG dJdK dKedZnG dLdM dMedZodWZpG dNdO dOedZqG dPdQ dQedZrG dRdS dSedZsG dTdU dUedZtG dVdW dWedZudWZvG dXdY dYedZwG dZd[ d[edZxG d\d] d]edZydWZzG d^d_ d_edZ{G d`da daedZ|G dbdc dcedZ}G ddde deedZ~dWZG dfdg dgedZG dhdi diedZG djdk dkedZG dldm dmedZdWZG dndo doedZG dpdq dqedZG drds dsedZG dtdu duedZG dvdw dwedZG dxdy dyedZG dzd{ d{edZdWZG d|d} d}edZG d~d dedZdWZdWZdW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ZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZdW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Zdd ZG dd dedZdW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ZdW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ZdW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ZdW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ZdWZG dd dedZG dd dedZG dd 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ZdW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ZdW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ZdWZG d,d- d-edZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZG d<d= d=edZd>d? ZdWZG d@dA dAedZG dBdC dCedZ G dDdE dEedZG dFdG dGedZdWZG dHdI dIedZG dJdK dKedZG dLdM dMedZdWZG dNdO dOedZG dPdQ dQedZ	G dRdS dSedZ
G dTdU dUedZG dVdW dWedZG dXdY dYedZG dZd[ d[edZd\d] ZdWZG d^d_ d_edZG d`da daedZG dbdc dcedZG ddde deedZG dfdg dgedZdWZG dhdi diedZG djdk dkedZG dldm dmedZG dndo doedZG dpdq dqedZG drds dsedZdtdu ZdWZG dvdw dwedZG dxdy dyedZ G dzd{ d{edZ!G d|d} d}edZ"G d~d dedZ#G dd dedZ$G dd dedZ%dW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*dW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Z0dWZ1G dd dedZ2G dd dedZ3G dd dedZ4dWZ5G dd dedZ6G dd dedZ7G dd dedZ8dWZ9G dd dedZ:G dd dedZ;G dd dedZ<G dd dedZ=dWZ>G dd dedZ?G dd dedZ@G dd dedZAG dd dedZBdWZCG dd dedZDG dd dedZEG dd dedZFG dd dedZGG dd dedZHG dd dedZIG dd dedZJdWZKG dd dedZLG ddÄ dedZMG dĐdń dedZNG dƐdǄ dedZOdȐdɄ ZPdWZQG dʐd˄ dedZRG d̐d̈́ dedZSG dΐdτ dedZTdWZUG dАdф dedZVG dҐdӄ dedZWG dԐdՄ dedZXG d֐dׄ dedZYdWZZG 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`dWZaG dd dedZbG dd dedZcG dd dedZdG dd dedZeG dd dedZfdWZgG dd dedZhG dd dedZiG dd dedZjG dd dedZkG dd dedZldWZmG dd dedZnG dd dedZoG dd dedZpG dd dedZqG d d dedZrdWZsG dd dedZtG dd dedZuG dd dedZvG dd	 d	edZwdWZxG d
d dedZyG dd dedZzG 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ZG dd dedZdW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ZdWZG d,d- d-edZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZdWZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZG d<d= d=edZG d>d? d?edZG d@dA dAedZG dBdC dCedZG dDdE dEedZG dFdG dGedZG dHdI dIedZG dJdK dKedZG dLdM dMedZG dNdO dOedZG dPdQ dQedZG dRdS dSedZG dTdU dUedZdWZG dVdW dWedZG dXdY dYedZG dZd[ d[edZG d\d] d]edZG d^d_ d_edZG d`da daedZdW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ZdWZG dldm dmedZG dndo doedZG dpdq dqedZdWZG drds dsedZG dtdu duedZG dvdw dwedZG dxdy dyedZG dzd{ d{edZG d|d} d}edZG d~d dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZdW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ZdW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ZdW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ZdW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ZdАdф ZdWZG dҐdӄ dedZG dԐdՄ dedZG d֐dׄ dedZdؐdل ZdWZG dڐdۄ dedZG dܐd݄ dedZG dސd߄ dedZG dd dedZdd ZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG 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 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
dW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dW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dW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 d0d1 d1edZ dWZ!G d2d3 d3edZ"G d4d5 d5edZ#G d6d7 d7edZ$G d8d9 d9edZ%G d:d; d;edZ&dWZ'G d<d= d=edZ(G d>d? d?edZ)G d@dA dAedZ*G dBdC dCedZ+G dDdE dEedZ,G dFdG dGedZ-G dHdI dIedZ.G dJdK dKedZ/dWZ0G dLdM dMedZ1G dNdO dOedZ2G dPdQ dQedZ3G dRdS dSedZ4G dTdU dUedZ5G dVdW dWedZ6G dXdY dYedZ7dWZ8G dZd[ d[edZ9G d\d] d]edZ:G d^d_ d_edZ;G d`da daedZ<G dbdc dcedZ=ddde Z>dWZ?G dfdg dgedZ@G dhdi diedZAG djdk dkedZBG dldm dmedZCG dndo doedZDdWZEG dpdq dqedZFG drds dsedZGG dtdu duedZHG dvdw dwedZIG dxdy dyedZJG dzd{ d{edZKG d|d} d}edZLG d~d dedZMG dd dedZNdWZOG dd dedZPG dd dedZQG dd dedZRdWZSG dd dedZTG dd dedZUG dd dedZVG dd dedZWG dd dedZXG dd dedZYG dd dedZZG dd dedZ[G dd dedZ\G dd dedZ]dWZ^G dd dedZ_G dd dedZ`G dd dedZaG dd dedZbdWZcG dd dedZdG dd dedZeG dd dedZfG dd dedZgG dd dedZhdWZiG dd dedZjG dd dedZkG dd dedZldWZmG dd dedZnG dd dedZoG dd dedZpG dd dedZqG dd dedZrG dd dedZsdWZtG dd dedZuG ddÄ dedZvG dĐdń dedZwG dƐdǄ dedZxG dȐdɄ dedZyG dʐd˄ dedZzG d̐d̈́ dedZ{G dΐdτ dedZ|G dАdф dedZ}G dҐdӄ dedZ~dԐdՄ ZG d֐dׄ dedZG dؐdل dedZG dڐdۄ dedZG dܐd݄ dedZdW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Zdd ZdW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ZdWZG dd dedZG d d dedZG dd dedZdW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 ZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG d d! d!edZG d"d# d#edZdWZG d$d% d%edZG d&d' d'edZG d(d) d)edZG d*d+ d+edZG d,d- d-edZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZdWZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZG d<d= d=edZG d>d? d?edZG d@dA dAedZG dBdC dCedZdWZG dDdE dEedZG dFdG dGedZG dHdI dIedZG dJdK dKedZG dLdM dMedZG dNdO dOedZG dPdQ dQedZG dRdS dSedZG dTdU dUedZG dVdW dWedZɐdXdY ZdWZG dZd[ d[edZG d\d] d]edZG d^d_ d_edZG d`da daedZG dbdc dcedZG ddde deedZG dfdg dgedZG dhdi diedZG djdk dkedZԐdldm ZdWZG dndo doedZG dpdq dqedZG drds dsedZdWZG dtdu duedZG dvdw dwedZdW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ZdWZG dd dedZG dd dedZG dd dedZG dd dedZdW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ZdW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ZdW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ZdW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dW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
dWZG dƐdǄ dedZG dȐdɄ dedZG dʐd˄ dedZdW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dWZG d֐dׄ dedZG dؐdل dedZG dڐdۄ dedZdW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dW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$dW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)dd Z*dWZ+G dd dedZ,G dd dedZ-G dd dedZ.dWZ/G d d dedZ0G dd dedZ1G dd dedZ2G dd dedZ3G dd	 d	edZ4d
d Z5dWZ6G dd dedZ7G dd dedZ8G dd dedZ9dW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>dWZ?G dd dedZ@G dd dedZAdW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ZGd(d) ZHdWZIG d*d+ d+edZJG d,d- d-edZKdWZLG d.d/ d/edZMG d0d1 d1edZNG d2d3 d3edZOG d4d5 d5edZPdWZQG d6d7 d7edZRG d8d9 d9edZSG d:d; d;edZTG d<d= d=edZUG d>d? d?edZVdWZWG d@dA dAedZXG dBdC dCedZYG dDdE dEedZZG dFdG dGedZ[G dHdI dIedZ\G dJdK dKedZ]G dLdM dMedZ^G dNdO dOedZ_G dPdQ dQedZ`dWZaG dRdS dSedZbG dTdU dUedZcG dVdW dWedZddWZeG dXdY dYedZfG dZd[ d[edZgG d\d] d]edZhG d^d_ d_edZidW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ZsG drds dsedZtdW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}dWZ~G dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZG dd dedZdWZG dd dedZG dd dedZG dd dedZdW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ZdW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ZdW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ZdWZG dʐd˄ dedZG d̐d̈́ dedZG dΐdτ dedZG dАdф dedZdWZG dҐdӄ dedZG dԐdՄ dedZG d֐dׄ dedZG dؐdل dedZdWZG dڐdۄ dedZG dܐd݄ dedZG dސd߄ dedZdW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ZdW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ZdWZG dd dedZG dd 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ZdW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ZdW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- ZdWZG d.d/ d/edZG d0d1 d1edZG d2d3 d3edZG d4d5 d5edZG d6d7 d7edZG d8d9 d9edZG d:d; d;edZG d<d= d=edZdWZG d>d? d?edZG d@dA dAedZG dBdC dCedZdWZG dDdE dEedZG dFdG dGedZG dHdI dIedZG dJdK dKedZG dLdM dMedZG dNdO dOedZG dPdQ dQedZG dRdS dSedZG dTdU dUedZG dVdW dWedZdXdY ZdZd[ Zd\d] Zd^d_ Zd`da Zdbdc Zddde Z dfdg ZG dhdi diedZdjdk Zdldm ZG dndo doedZdpdq ZG drds dsedZdWS (t     )DummyObjectrequires_backendsc                   @   s   e Zd ZdgZdd ZdS )PyTorchBenchmarktorchc                 O   s   t | dg d S Nr   r   selfargskwargs r   l/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/transformers/utils/dummy_pt_objects.py__init__   s    zPyTorchBenchmark.__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 )PyTorchBenchmarkArgumentsr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"PyTorchBenchmarkArguments.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )GlueDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zGlueDataset.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )GlueDataTrainingArgumentsr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"GlueDataTrainingArguments.__init__Nr   r   r   r   r   r      s   r   c                   @   s   e Zd ZdgZdd ZdS )LineByLineTextDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $   s    zLineByLineTextDataset.__init__Nr   r   r   r   r   r   !   s   r   c                   @   s   e Zd ZdgZdd ZdS )LineByLineWithRefDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +   s    z!LineByLineWithRefDataset.__init__Nr   r   r   r   r   r   (   s   r   c                   @   s   e Zd ZdgZdd ZdS )LineByLineWithSOPTextDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2   s    z%LineByLineWithSOPTextDataset.__init__Nr   r   r   r   r   r   /   s   r   c                   @   s   e Zd ZdgZdd ZdS )SquadDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9   s    zSquadDataset.__init__Nr   r   r   r   r   r   6   s   r   c                   @   s   e Zd ZdgZdd ZdS )SquadDataTrainingArgumentsr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @   s    z#SquadDataTrainingArguments.__init__Nr   r   r   r   r   r   =   s   r   c                   @   s   e Zd ZdgZdd ZdS )TextDatasetr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G   s    zTextDataset.__init__Nr   r   r   r   r   r   D   s   r   c                   @   s   e Zd ZdgZdd ZdS )$TextDatasetForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N   s    z-TextDatasetForNextSentencePrediction.__init__Nr   r   r   r   r   r   K   s   r   c                   @   s   e Zd ZdgZdd ZdS )
BeamScorerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U   s    zBeamScorer.__init__Nr   r   r   r   r   r   R   s   r   c                   @   s   e Zd ZdgZdd ZdS )BeamSearchScorerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \   s    zBeamSearchScorer.__init__Nr   r   r   r   r   r   Y   s   r   c                   @   s   e Zd ZdgZdd ZdS )ConstrainedBeamSearchScorerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c   s    z$ConstrainedBeamSearchScorer.__init__Nr   r   r   r   r   r    `   s   r    c                   @   s   e Zd ZdgZdd ZdS )
Constraintr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j   s    zConstraint.__init__Nr   r   r   r   r   r!   g   s   r!   c                   @   s   e Zd ZdgZdd ZdS )ConstraintListStater   c                 O   s   t | dg d S r   r   r   r   r   r   r   q   s    zConstraintListState.__init__Nr   r   r   r   r   r"   n   s   r"   c                   @   s   e Zd ZdgZdd ZdS )DisjunctiveConstraintr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x   s    zDisjunctiveConstraint.__init__Nr   r   r   r   r   r#   u   s   r#   c                   @   s   e Zd ZdgZdd ZdS )ForcedBOSTokenLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z&ForcedBOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r$   |   s   r$   c                   @   s   e Zd ZdgZdd ZdS )ForcedEOSTokenLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z&ForcedEOSTokenLogitsProcessor.__init__Nr   r   r   r   r   r%      s   r%   c                   @   s   e Zd ZdgZdd ZdS )GenerationMixinr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zGenerationMixin.__init__Nr   r   r   r   r   r&      s   r&   c                   @   s   e Zd ZdgZdd ZdS )HammingDiversityLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z(HammingDiversityLogitsProcessor.__init__Nr   r   r   r   r   r'      s   r'   c                   @   s   e Zd ZdgZdd ZdS )InfNanRemoveLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z$InfNanRemoveLogitsProcessor.__init__Nr   r   r   r   r   r(      s   r(   c                   @   s   e Zd ZdgZdd ZdS )LogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zLogitsProcessor.__init__Nr   r   r   r   r   r)      s   r)   c                   @   s   e Zd ZdgZdd ZdS )LogitsProcessorListr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zLogitsProcessorList.__init__Nr   r   r   r   r   r*      s   r*   c                   @   s   e Zd ZdgZdd ZdS )LogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zLogitsWarper.__init__Nr   r   r   r   r   r+      s   r+   c                   @   s   e Zd ZdgZdd ZdS )MaxLengthCriteriar   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zMaxLengthCriteria.__init__Nr   r   r   r   r   r,      s   r,   c                   @   s   e Zd ZdgZdd ZdS )MaxTimeCriteriar   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zMaxTimeCriteria.__init__Nr   r   r   r   r   r-      s   r-   c                   @   s   e Zd ZdgZdd ZdS )MinLengthLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z!MinLengthLogitsProcessor.__init__Nr   r   r   r   r   r.      s   r.   c                   @   s   e Zd ZdgZdd ZdS )!MinNewTokensLengthLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z*MinNewTokensLengthLogitsProcessor.__init__Nr   r   r   r   r   r/      s   r/   c                   @   s   e Zd ZdgZdd ZdS )NoBadWordsLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"NoBadWordsLogitsProcessor.__init__Nr   r   r   r   r   r0      s   r0   c                   @   s   e Zd ZdgZdd ZdS )NoRepeatNGramLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z%NoRepeatNGramLogitsProcessor.__init__Nr   r   r   r   r   r1      s   r1   c                   @   s   e Zd ZdgZdd ZdS )PhrasalConstraintr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zPhrasalConstraint.__init__Nr   r   r   r   r   r2      s   r2   c                   @   s   e Zd ZdgZdd ZdS ) PrefixConstrainedLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z)PrefixConstrainedLogitsProcessor.__init__Nr   r   r   r   r   r3      s   r3   c                   @   s   e Zd ZdgZdd ZdS ) RepetitionPenaltyLogitsProcessorr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z)RepetitionPenaltyLogitsProcessor.__init__Nr   r   r   r   r   r4      s   r4   c                   @   s   e Zd ZdgZdd ZdS )StoppingCriteriar   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zStoppingCriteria.__init__Nr   r   r   r   r   r5      s   r5   c                   @   s   e Zd ZdgZdd ZdS )StoppingCriteriaListr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zStoppingCriteriaList.__init__Nr   r   r   r   r   r6      s   r6   c                   @   s   e Zd ZdgZdd ZdS )TemperatureLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z TemperatureLogitsWarper.__init__Nr   r   r   r   r   r7     s   r7   c                   @   s   e Zd ZdgZdd ZdS )TopKLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTopKLogitsWarper.__init__Nr   r   r   r   r   r8     s   r8   c                   @   s   e Zd ZdgZdd ZdS )TopPLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTopPLogitsWarper.__init__Nr   r   r   r   r   r9     s   r9   c                   @   s   e Zd ZdgZdd ZdS )TypicalLogitsWarperr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTypicalLogitsWarper.__init__Nr   r   r   r   r   r:     s   r:   c                  O   s   t tdg d S r   )r   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 )PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $  s    zPreTrainedModel.__init__Nr   r   r   r   r   r=   !  s   r=   Nc                   @   s   e Zd ZdgZdd ZdS )AlbertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    zAlbertForMaskedLM.__init__Nr   r   r   r   r   r>   +  s   r>   c                   @   s   e Zd ZdgZdd ZdS )AlbertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    z AlbertForMultipleChoice.__init__Nr   r   r   r   r   r?   2  s   r?   c                   @   s   e Zd ZdgZdd ZdS )AlbertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    zAlbertForPreTraining.__init__Nr   r   r   r   r   r@   9  s   r@   c                   @   s   e Zd ZdgZdd ZdS )AlbertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    z#AlbertForQuestionAnswering.__init__Nr   r   r   r   r   rA   @  s   rA   c                   @   s   e Zd ZdgZdd ZdS )AlbertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    z(AlbertForSequenceClassification.__init__Nr   r   r   r   r   rB   G  s   rB   c                   @   s   e Zd ZdgZdd ZdS )AlbertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    z%AlbertForTokenClassification.__init__Nr   r   r   r   r   rC   N  s   rC   c                   @   s   e Zd ZdgZdd ZdS )AlbertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   X  s    zAlbertModel.__init__Nr   r   r   r   r   rD   U  s   rD   c                   @   s   e Zd ZdgZdd ZdS )AlbertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   _  s    zAlbertPreTrainedModel.__init__Nr   r   r   r   r   rE   \  s   rE   c                  O   s   t tdg d S r   )r   load_tf_weights_in_albertr<   r   r   r   rF   c  s    rF   c                   @   s   e Zd ZdgZdd ZdS )
AlignModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   m  s    zAlignModel.__init__Nr   r   r   r   r   rG   j  s   rG   c                   @   s   e Zd ZdgZdd ZdS )AlignPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   t  s    zAlignPreTrainedModel.__init__Nr   r   r   r   r   rH   q  s   rH   c                   @   s   e Zd ZdgZdd ZdS )AlignTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    zAlignTextModel.__init__Nr   r   r   r   r   rI   x  s   rI   c                   @   s   e Zd ZdgZdd ZdS )AlignVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAlignVisionModel.__init__Nr   r   r   r   r   rJ     s   rJ   c                   @   s   e Zd ZdgZdd ZdS )AltCLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAltCLIPModel.__init__Nr   r   r   r   r   rK     s   rK   c                   @   s   e Zd ZdgZdd ZdS )AltCLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAltCLIPPreTrainedModel.__init__Nr   r   r   r   r   rL     s   rL   c                   @   s   e Zd ZdgZdd ZdS )AltCLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAltCLIPTextModel.__init__Nr   r   r   r   r   rM     s   rM   c                   @   s   e Zd ZdgZdd ZdS )AltCLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAltCLIPVisionModel.__init__Nr   r   r   r   r   rN     s   rN   c                   @   s   e Zd ZdgZdd ZdS )ASTForAudioClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"ASTForAudioClassification.__init__Nr   r   r   r   r   rO     s   rO   c                   @   s   e Zd ZdgZdd ZdS )ASTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zASTModel.__init__Nr   r   r   r   r   rP     s   rP   c                   @   s   e Zd ZdgZdd ZdS )ASTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zASTPreTrainedModel.__init__Nr   r   r   r   r   rQ     s   rQ   c                   @   s   e Zd ZdgZdd ZdS )AutoBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    zAutoBackbone.__init__Nr   r   r   r   r   rR      s   rR   c                   @   s   e Zd ZdgZdd ZdS )	AutoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    zAutoModel.__init__Nr   r   r   r   r   rS   '  s   rS   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForAudioClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   1  s    z(AutoModelForAudioClassification.__init__Nr   r   r   r   r   rT   .  s   rT   c                   @   s   e Zd ZdgZdd ZdS )$AutoModelForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   8  s    z-AutoModelForAudioFrameClassification.__init__Nr   r   r   r   r   rU   5  s   rU   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForAudioXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ?  s    z!AutoModelForAudioXVector.__init__Nr   r   r   r   r   rV   <  s   rV   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   F  s    zAutoModelForCausalLM.__init__Nr   r   r   r   r   rW   C  s   rW   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r   M  s    zAutoModelForCTC.__init__Nr   r   r   r   r   rX   J  s   rX   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForDepthEstimationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   T  s    z$AutoModelForDepthEstimation.__init__Nr   r   r   r   r   rY   Q  s   rY   c                   @   s   e Zd ZdgZdd ZdS )%AutoModelForDocumentQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [  s    z.AutoModelForDocumentQuestionAnswering.__init__Nr   r   r   r   r   rZ   X  s   rZ   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b  s    z(AutoModelForImageClassification.__init__Nr   r   r   r   r   r[   _  s   r[   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForImageSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   i  s    z&AutoModelForImageSegmentation.__init__Nr   r   r   r   r   r\   f  s   r\   c                   @   s   e Zd ZdgZdd ZdS ) AutoModelForInstanceSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p  s    z)AutoModelForInstanceSegmentation.__init__Nr   r   r   r   r   r]   m  s   r]   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   w  s    z(AutoModelForMaskedImageModeling.__init__Nr   r   r   r   r   r^   t  s   r^   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ~  s    zAutoModelForMaskedLM.__init__Nr   r   r   r   r   r_   {  s   r_   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForMaskGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#AutoModelForMaskGeneration.__init__Nr   r   r   r   r   r`     s   r`   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#AutoModelForMultipleChoice.__init__Nr   r   r   r   r   ra     s   ra   c                   @   s   e Zd ZdgZdd ZdS )"AutoModelForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+AutoModelForNextSentencePrediction.__init__Nr   r   r   r   r   rb     s   rb   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$AutoModelForObjectDetection.__init__Nr   r   r   r   r   rc     s   rc   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z AutoModelForPreTraining.__init__Nr   r   r   r   r   rd     s   rd   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&AutoModelForQuestionAnswering.__init__Nr   r   r   r   r   re     s   re   c                   @   s   e Zd ZdgZdd ZdS ) AutoModelForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)AutoModelForSemanticSegmentation.__init__Nr   r   r   r   r   rf     s   rf   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForSeq2SeqLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAutoModelForSeq2SeqLM.__init__Nr   r   r   r   r   rg     s   rg   c                   @   s   e Zd ZdgZdd ZdS )"AutoModelForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+AutoModelForSequenceClassification.__init__Nr   r   r   r   r   rh     s   rh   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForSpeechSeq2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"AutoModelForSpeechSeq2Seq.__init__Nr   r   r   r   r   ri     s   ri   c                   @   s   e Zd ZdgZdd ZdS )"AutoModelForTableQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+AutoModelForTableQuestionAnswering.__init__Nr   r   r   r   r   rj     s   rj   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(AutoModelForTokenClassification.__init__Nr   r   r   r   r   rk     s   rk   c                   @   s   e Zd ZdgZdd ZdS )!AutoModelForUniversalSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*AutoModelForUniversalSegmentation.__init__Nr   r   r   r   r   rl     s   rl   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForVideoClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(AutoModelForVideoClassification.__init__Nr   r   r   r   r   rm     s   rm   c                   @   s   e Zd ZdgZdd ZdS )AutoModelForVision2Seqr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAutoModelForVision2Seq.__init__Nr   r   r   r   r   rn     s   rn   c                   @   s   e Zd ZdgZdd ZdS )#AutoModelForVisualQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,AutoModelForVisualQuestionAnswering.__init__Nr   r   r   r   r   ro     s   ro   c                   @   s   e Zd ZdgZdd ZdS )'AutoModelForZeroShotImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0AutoModelForZeroShotImageClassification.__init__Nr   r   r   r   r   rp     s   rp   c                   @   s   e Zd ZdgZdd ZdS )#AutoModelForZeroShotObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,AutoModelForZeroShotObjectDetection.__init__Nr   r   r   r   r   rq     s   rq   c                   @   s   e Zd ZdgZdd ZdS )AutoModelWithLMHeadr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAutoModelWithLMHead.__init__Nr   r   r   r   r   rr      s   rr   c                   @   s   e Zd ZdgZdd ZdS )AutoformerForPredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z AutoformerForPrediction.__init__Nr   r   r   r   r   rs   
  s   rs   c                   @   s   e Zd ZdgZdd ZdS )AutoformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAutoformerModel.__init__Nr   r   r   r   r   rt     s   rt   c                   @   s   e Zd ZdgZdd ZdS )AutoformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"AutoformerPreTrainedModel.__init__Nr   r   r   r   r   ru     s   ru   c                   @   s   e Zd ZdgZdd ZdS )BartForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   %  s    zBartForCausalLM.__init__Nr   r   r   r   r   rv   "  s   rv   c                   @   s   e Zd ZdgZdd ZdS )BartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ,  s    z%BartForConditionalGeneration.__init__Nr   r   r   r   r   rw   )  s   rw   c                   @   s   e Zd ZdgZdd ZdS )BartForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   3  s    z!BartForQuestionAnswering.__init__Nr   r   r   r   r   rx   0  s   rx   c                   @   s   e Zd ZdgZdd ZdS )BartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    z&BartForSequenceClassification.__init__Nr   r   r   r   r   ry   7  s   ry   c                   @   s   e Zd ZdgZdd ZdS )	BartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   A  s    zBartModel.__init__Nr   r   r   r   r   rz   >  s   rz   c                   @   s   e Zd ZdgZdd ZdS )BartPretrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   H  s    zBartPretrainedModel.__init__Nr   r   r   r   r   r{   E  s   r{   c                   @   s   e Zd ZdgZdd ZdS )PretrainedBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   O  s    zPretrainedBartModel.__init__Nr   r   r   r   r   r|   L  s   r|   c                   @   s   e Zd ZdgZdd ZdS )BeitForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    z#BeitForImageClassification.__init__Nr   r   r   r   r   r}   V  s   r}   c                   @   s   e Zd ZdgZdd ZdS )BeitForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   `  s    z#BeitForMaskedImageModeling.__init__Nr   r   r   r   r   r~   ]  s   r~   c                   @   s   e Zd ZdgZdd ZdS )BeitForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   g  s    z$BeitForSemanticSegmentation.__init__Nr   r   r   r   r   r   d  s   r   c                   @   s   e Zd ZdgZdd ZdS )	BeitModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   n  s    zBeitModel.__init__Nr   r   r   r   r   r   k  s   r   c                   @   s   e Zd ZdgZdd ZdS )BeitPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   u  s    zBeitPreTrainedModel.__init__Nr   r   r   r   r   r   r  s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertForMaskedLM.__init__Nr   r   r   r   r   r   |  s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&BertForNextSentencePrediction.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertForPreTraining.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!BertForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&BertForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#BertForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )	BertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertLMHeadModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )	BertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                  O   s   t tdg d S r   )r   load_tf_weights_in_bertr<   r   r   r   r     s    r   c                   @   s   e Zd ZdgZdd ZdS )BertGenerationDecoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertGenerationDecoder.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertGenerationEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBertGenerationEncoder.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BertGenerationPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&BertGenerationPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                  O   s   t tdg d S r   )r   "load_tf_weights_in_bert_generationr<   r   r   r   r     s    r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBigBirdForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBigBirdForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!BigBirdForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBigBirdForPreTraining.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$BigBirdForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS ) BigBirdForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)BigBirdForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&BigBirdForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBigBirdLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $  s    zBigBirdModel.__init__Nr   r   r   r   r   r   !  s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +  s    zBigBirdPreTrainedModel.__init__Nr   r   r   r   r   r   (  s   r   c                  O   s   t tdg d S r   )r   load_tf_weights_in_big_birdr<   r   r   r   r   /  s    r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdPegasusForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9  s    z"BigBirdPegasusForCausalLM.__init__Nr   r   r   r   r   r   6  s   r   c                   @   s   e Zd ZdgZdd ZdS )&BigBirdPegasusForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @  s    z/BigBirdPegasusForConditionalGeneration.__init__Nr   r   r   r   r   r   =  s   r   c                   @   s   e Zd ZdgZdd ZdS )"BigBirdPegasusForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G  s    z+BigBirdPegasusForQuestionAnswering.__init__Nr   r   r   r   r   r   D  s   r   c                   @   s   e Zd ZdgZdd ZdS )'BigBirdPegasusForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N  s    z0BigBirdPegasusForSequenceClassification.__init__Nr   r   r   r   r   r   K  s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdPegasusModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U  s    zBigBirdPegasusModel.__init__Nr   r   r   r   r   r   R  s   r   c                   @   s   e Zd ZdgZdd ZdS )BigBirdPegasusPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    z&BigBirdPegasusPreTrainedModel.__init__Nr   r   r   r   r   r   Y  s   r   c                   @   s   e Zd ZdgZdd ZdS )BioGptForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   f  s    zBioGptForCausalLM.__init__Nr   r   r   r   r   r   c  s   r   c                   @   s   e Zd ZdgZdd ZdS )BioGptForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   m  s    z(BioGptForSequenceClassification.__init__Nr   r   r   r   r   r   j  s   r   c                   @   s   e Zd ZdgZdd ZdS )BioGptForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   t  s    z%BioGptForTokenClassification.__init__Nr   r   r   r   r   r   q  s   r   c                   @   s   e Zd ZdgZdd ZdS )BioGptModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    zBioGptModel.__init__Nr   r   r   r   r   r   x  s   r   c                   @   s   e Zd ZdgZdd ZdS )BioGptPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBioGptPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BitBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBitBackbone.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BitForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"BitForImageClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BitModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBitModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BitPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBitPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlenderbotForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"BlenderbotForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+BlenderbotForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlenderbotModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"BlenderbotPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotSmallForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#BlenderbotSmallForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )'BlenderbotSmallForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z0BlenderbotSmallForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotSmallModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlenderbotSmallModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlenderbotSmallPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'BlenderbotSmallPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%BlipForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipForImageTextRetrievalr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"BlipForImageTextRetrieval.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!BlipForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )	BlipModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlipModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlipPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlipTextModel.__init__Nr   r   r   r   r   r   	  s   r   c                   @   s   e Zd ZdgZdd ZdS )BlipVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBlipVisionModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )Blip2ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&Blip2ForConditionalGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )
Blip2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $  s    zBlip2Model.__init__Nr   r   r   r   r   r   !  s   r   c                   @   s   e Zd ZdgZdd ZdS )Blip2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +  s    zBlip2PreTrainedModel.__init__Nr   r   r   r   r   r   (  s   r   c                   @   s   e Zd ZdgZdd ZdS )Blip2QFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2  s    zBlip2QFormerModel.__init__Nr   r   r   r   r   r   /  s   r   c                   @   s   e Zd ZdgZdd ZdS )Blip2VisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9  s    zBlip2VisionModel.__init__Nr   r   r   r   r   r   6  s   r   c                   @   s   e Zd ZdgZdd ZdS )BloomForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    zBloomForCausalLM.__init__Nr   r   r   r   r   r   @  s   r   c                   @   s   e Zd ZdgZdd ZdS )BloomForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    z"BloomForQuestionAnswering.__init__Nr   r   r   r   r   r   G  s   r   c                   @   s   e Zd ZdgZdd ZdS )BloomForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    z'BloomForSequenceClassification.__init__Nr   r   r   r   r   r   N  s   r   c                   @   s   e Zd ZdgZdd ZdS )BloomForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   X  s    z$BloomForTokenClassification.__init__Nr   r   r   r   r   r   U  s   r   c                   @   s   e Zd ZdgZdd ZdS )
BloomModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   _  s    zBloomModel.__init__Nr   r   r   r   r   r   \  s   r   c                   @   s   e Zd ZdgZdd ZdS )BloomPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   f  s    zBloomPreTrainedModel.__init__Nr   r   r   r   r   r   c  s   r   c                   @   s   e Zd ZdgZdd ZdS )!BridgeTowerForContrastiveLearningr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p  s    z*BridgeTowerForContrastiveLearning.__init__Nr   r   r   r   r   r   m  s   r   c                   @   s   e Zd ZdgZdd ZdS )#BridgeTowerForImageAndTextRetrievalr   c                 O   s   t | dg d S r   r   r   r   r   r   r   w  s    z,BridgeTowerForImageAndTextRetrieval.__init__Nr   r   r   r   r   r   t  s   r   c                   @   s   e Zd ZdgZdd ZdS )BridgeTowerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ~  s    zBridgeTowerForMaskedLM.__init__Nr   r   r   r   r   r   {  s   r   c                   @   s   e Zd ZdgZdd ZdS )BridgeTowerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zBridgeTowerModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )BridgeTowerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#BridgeTowerPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCamembertForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCamembertForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#CamembertForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&CamembertForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )"CamembertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+CamembertForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(CamembertForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCamembertModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CamembertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!CamembertPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z CanineForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#CanineForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(CanineForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%CanineForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCanineLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CanineModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCanineModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CaninePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCaninePreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                  O   s   t tdg d S r   )r   load_tf_weights_in_caniner<   r   r   r   r     s    r   c                   @   s   e Zd ZdgZdd ZdS )ChineseCLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zChineseCLIPModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ChineseCLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#ChineseCLIPPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ChineseCLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zChineseCLIPTextModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ChineseCLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zChineseCLIPVisionModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapAudioModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   (  s    zClapAudioModel.__init__Nr   r   r   r   r   r   %  s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapAudioModelWithProjectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   /  s    z%ClapAudioModelWithProjection.__init__Nr   r   r   r   r   r   ,  s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapFeatureExtractorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   6  s    zClapFeatureExtractor.__init__Nr   r   r   r   r   r   3  s   r   c                   @   s   e Zd ZdgZdd ZdS )	ClapModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   =  s    zClapModel.__init__Nr   r   r   r   r   r   :  s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   D  s    zClapPreTrainedModel.__init__Nr   r   r   r   r   r   A  s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   K  s    zClapTextModel.__init__Nr   r   r   r   r   r   H  s   r   c                   @   s   e Zd ZdgZdd ZdS )ClapTextModelWithProjectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    z$ClapTextModelWithProjection.__init__Nr   r   r   r   r   r   O  s   r   c                   @   s   e Zd ZdgZdd ZdS )	CLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    zCLIPModel.__init__Nr   r   r   r   r   r   Y  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c  s    zCLIPPreTrainedModel.__init__Nr   r   r   r   r   r   `  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j  s    zCLIPTextModel.__init__Nr   r   r   r   r   r   g  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPTextModelWithProjectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q  s    z$CLIPTextModelWithProjection.__init__Nr   r   r   r   r   r   n  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x  s    zCLIPVisionModel.__init__Nr   r   r   r   r   r   u  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPVisionModelWithProjectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&CLIPVisionModelWithProjection.__init__Nr   r   r   r   r   r   |  s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPSegForImageSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$CLIPSegForImageSegmentation.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPSegModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCLIPSegModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPSegPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCLIPSegPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPSegTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCLIPSegTextModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CLIPSegVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCLIPSegVisionModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CodeGenForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCodeGenForCausalLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CodeGenModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCodeGenModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )CodeGenPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCodeGenPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )!ConditionalDetrForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*ConditionalDetrForObjectDetection.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConditionalDetrForSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'ConditionalDetrForSegmentation.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConditionalDetrModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zConditionalDetrModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConditionalDetrPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'ConditionalDetrPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zConvBertForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"ConvBertForMultipleChoice.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%ConvBertForQuestionAnswering.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )!ConvBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*ConvBertForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'ConvBertForTokenClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zConvBertLayer.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zConvBertModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ConvBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z ConvBertPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                  O   s   t tdg d S r   )r   load_tf_weights_in_convbertr<   r   r   r   r     s    r   c                   @   s   e Zd ZdgZdd ZdS )ConvNextBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r   %  s    zConvNextBackbone.__init__Nr   r   r   r   r   r  "  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ,  s    z'ConvNextForImageClassification.__init__Nr   r   r   r   r   r  )  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   3  s    zConvNextModel.__init__Nr   r   r   r   r   r  0  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    z ConvNextPreTrainedModel.__init__Nr   r   r   r   r   r  7  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextV2Backboner   c                 O   s   t | dg d S r   r   r   r   r   r   r   D  s    zConvNextV2Backbone.__init__Nr   r   r   r   r   r  A  s   r  c                   @   s   e Zd ZdgZdd ZdS ) ConvNextV2ForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   K  s    z)ConvNextV2ForImageClassification.__init__Nr   r   r   r   r   r  H  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextV2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    zConvNextV2Model.__init__Nr   r   r   r   r   r  O  s   r  c                   @   s   e Zd ZdgZdd ZdS )ConvNextV2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    z"ConvNextV2PreTrainedModel.__init__Nr   r   r   r   r   r  V  s   r  c                   @   s   e Zd ZdgZdd ZdS )CpmAntForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c  s    zCpmAntForCausalLM.__init__Nr   r   r   r   r   r	  `  s   r	  c                   @   s   e Zd ZdgZdd ZdS )CpmAntModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j  s    zCpmAntModel.__init__Nr   r   r   r   r   r
  g  s   r
  c                   @   s   e Zd ZdgZdd ZdS )CpmAntPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q  s    zCpmAntPreTrainedModel.__init__Nr   r   r   r   r   r  n  s   r  c                   @   s   e Zd ZdgZdd ZdS )CTRLForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    z&CTRLForSequenceClassification.__init__Nr   r   r   r   r   r  x  s   r  c                   @   s   e Zd ZdgZdd ZdS )CTRLLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCTRLLMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	CTRLModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCTRLModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )CTRLPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCTRLPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )CvtForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"CvtForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )CvtModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCvtModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )CvtPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zCvtPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )(Data2VecAudioForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z1Data2VecAudioForAudioFrameClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecAudioForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zData2VecAudioForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )&Data2VecAudioForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/Data2VecAudioForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecAudioForXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Data2VecAudioForXVector.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecAudioModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zData2VecAudioModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecAudioPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%Data2VecAudioPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecTextForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Data2VecTextForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecTextForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Data2VecTextForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecTextForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&Data2VecTextForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) Data2VecTextForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)Data2VecTextForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )%Data2VecTextForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.Data2VecTextForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"Data2VecTextForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+Data2VecTextForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zData2VecTextModel.__init__Nr   r   r   r   r   r  	  s   r  c                   @   s   e Zd ZdgZdd ZdS )Data2VecTextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$Data2VecTextPreTrainedModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )$Data2VecVisionForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-Data2VecVisionForImageClassification.__init__Nr   r   r   r   r   r!    s   r!  c                   @   s   e Zd ZdgZdd ZdS )%Data2VecVisionForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   !  s    z.Data2VecVisionForSemanticSegmentation.__init__Nr   r   r   r   r   r"    s   r"  c                   @   s   e Zd ZdgZdd ZdS )Data2VecVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   (  s    zData2VecVisionModel.__init__Nr   r   r   r   r   r#  %  s   r#  c                   @   s   e Zd ZdgZdd ZdS )Data2VecVisionPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   /  s    z&Data2VecVisionPreTrainedModel.__init__Nr   r   r   r   r   r$  ,  s   r$  c                   @   s   e Zd ZdgZdd ZdS )DebertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9  s    zDebertaForMaskedLM.__init__Nr   r   r   r   r   r%  6  s   r%  c                   @   s   e Zd ZdgZdd ZdS )DebertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @  s    z$DebertaForQuestionAnswering.__init__Nr   r   r   r   r   r&  =  s   r&  c                   @   s   e Zd ZdgZdd ZdS ) DebertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G  s    z)DebertaForSequenceClassification.__init__Nr   r   r   r   r   r'  D  s   r'  c                   @   s   e Zd ZdgZdd ZdS )DebertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N  s    z&DebertaForTokenClassification.__init__Nr   r   r   r   r   r(  K  s   r(  c                   @   s   e Zd ZdgZdd ZdS )DebertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U  s    zDebertaModel.__init__Nr   r   r   r   r   r)  R  s   r)  c                   @   s   e Zd ZdgZdd ZdS )DebertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    zDebertaPreTrainedModel.__init__Nr   r   r   r   r   r*  Y  s   r*  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2ForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   f  s    zDebertaV2ForMaskedLM.__init__Nr   r   r   r   r   r+  c  s   r+  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2ForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   m  s    z#DebertaV2ForMultipleChoice.__init__Nr   r   r   r   r   r,  j  s   r,  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   t  s    z&DebertaV2ForQuestionAnswering.__init__Nr   r   r   r   r   r-  q  s   r-  c                   @   s   e Zd ZdgZdd ZdS )"DebertaV2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   {  s    z+DebertaV2ForSequenceClassification.__init__Nr   r   r   r   r   r.  x  s   r.  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(DebertaV2ForTokenClassification.__init__Nr   r   r   r   r   r/    s   r/  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDebertaV2Model.__init__Nr   r   r   r   r   r0    s   r0  c                   @   s   e Zd ZdgZdd ZdS )DebertaV2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!DebertaV2PreTrainedModel.__init__Nr   r   r   r   r   r1    s   r1  c                   @   s   e Zd ZdgZdd ZdS )DecisionTransformerGPT2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%DecisionTransformerGPT2Model.__init__Nr   r   r   r   r   r2    s   r2  c                   @   s   e Zd ZdgZdd ZdS )&DecisionTransformerGPT2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/DecisionTransformerGPT2PreTrainedModel.__init__Nr   r   r   r   r   r3    s   r3  c                   @   s   e Zd ZdgZdd ZdS )DecisionTransformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!DecisionTransformerModel.__init__Nr   r   r   r   r   r4    s   r4  c                   @   s   e Zd ZdgZdd ZdS )"DecisionTransformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+DecisionTransformerPreTrainedModel.__init__Nr   r   r   r   r   r5    s   r5  c                   @   s   e Zd ZdgZdd ZdS ) DeformableDetrForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)DeformableDetrForObjectDetection.__init__Nr   r   r   r   r   r6    s   r6  c                   @   s   e Zd ZdgZdd ZdS )DeformableDetrModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDeformableDetrModel.__init__Nr   r   r   r   r   r7    s   r7  c                   @   s   e Zd ZdgZdd ZdS )DeformableDetrPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&DeformableDetrPreTrainedModel.__init__Nr   r   r   r   r   r8    s   r8  c                   @   s   e Zd ZdgZdd ZdS )DeiTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#DeiTForImageClassification.__init__Nr   r   r   r   r   r9    s   r9  c                   @   s   e Zd ZdgZdd ZdS )%DeiTForImageClassificationWithTeacherr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.DeiTForImageClassificationWithTeacher.__init__Nr   r   r   r   r   r:    s   r:  c                   @   s   e Zd ZdgZdd ZdS )DeiTForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#DeiTForMaskedImageModeling.__init__Nr   r   r   r   r   r;    s   r;  c                   @   s   e Zd ZdgZdd ZdS )	DeiTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDeiTModel.__init__Nr   r   r   r   r   r<    s   r<  c                   @   s   e Zd ZdgZdd ZdS )DeiTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDeiTPreTrainedModel.__init__Nr   r   r   r   r   r=    s   r=  c                   @   s   e Zd ZdgZdd ZdS )DetaForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDetaForObjectDetection.__init__Nr   r   r   r   r   r>    s   r>  c                   @   s   e Zd ZdgZdd ZdS )	DetaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zDetaModel.__init__Nr   r   r   r   r   r?    s   r?  c                   @   s   e Zd ZdgZdd ZdS )DetaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDetaPreTrainedModel.__init__Nr   r   r   r   r   r@  	  s   r@  c                   @   s   e Zd ZdgZdd ZdS )DetrForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDetrForObjectDetection.__init__Nr   r   r   r   r   rA  	  s   rA  c                   @   s   e Zd ZdgZdd ZdS )DetrForSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDetrForSegmentation.__init__Nr   r   r   r   r   rB  	  s   rB  c                   @   s   e Zd ZdgZdd ZdS )	DetrModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDetrModel.__init__Nr   r   r   r   r   rC  	  s   rC  c                   @   s   e Zd ZdgZdd ZdS )DetrPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $	  s    zDetrPreTrainedModel.__init__Nr   r   r   r   r   rD  !	  s   rD  c                   @   s   e Zd ZdgZdd ZdS )DinatBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r   .	  s    zDinatBackbone.__init__Nr   r   r   r   r   rE  +	  s   rE  c                   @   s   e Zd ZdgZdd ZdS )DinatForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5	  s    z$DinatForImageClassification.__init__Nr   r   r   r   r   rF  2	  s   rF  c                   @   s   e Zd ZdgZdd ZdS )
DinatModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <	  s    zDinatModel.__init__Nr   r   r   r   r   rG  9	  s   rG  c                   @   s   e Zd ZdgZdd ZdS )DinatPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C	  s    zDinatPreTrainedModel.__init__Nr   r   r   r   r   rH  @	  s   rH  c                   @   s   e Zd ZdgZdd ZdS )DistilBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   M	  s    zDistilBertForMaskedLM.__init__Nr   r   r   r   r   rI  J	  s   rI  c                   @   s   e Zd ZdgZdd ZdS )DistilBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   T	  s    z$DistilBertForMultipleChoice.__init__Nr   r   r   r   r   rJ  Q	  s   rJ  c                   @   s   e Zd ZdgZdd ZdS )DistilBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [	  s    z'DistilBertForQuestionAnswering.__init__Nr   r   r   r   r   rK  X	  s   rK  c                   @   s   e Zd ZdgZdd ZdS )#DistilBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b	  s    z,DistilBertForSequenceClassification.__init__Nr   r   r   r   r   rL  _	  s   rL  c                   @   s   e Zd ZdgZdd ZdS ) DistilBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   i	  s    z)DistilBertForTokenClassification.__init__Nr   r   r   r   r   rM  f	  s   rM  c                   @   s   e Zd ZdgZdd ZdS )DistilBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p	  s    zDistilBertModel.__init__Nr   r   r   r   r   rN  m	  s   rN  c                   @   s   e Zd ZdgZdd ZdS )DistilBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   w	  s    z"DistilBertPreTrainedModel.__init__Nr   r   r   r   r   rO  t	  s   rO  c                   @   s   e Zd ZdgZdd ZdS )DonutSwinModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDonutSwinModel.__init__Nr   r   r   r   r   rP  ~	  s   rP  c                   @   s   e Zd ZdgZdd ZdS )DonutSwinPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z!DonutSwinPreTrainedModel.__init__Nr   r   r   r   r   rQ  	  s   rQ  c                   @   s   e Zd ZdgZdd ZdS )DPRContextEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPRContextEncoder.__init__Nr   r   r   r   r   rR  	  s   rR  c                   @   s   e Zd ZdgZdd ZdS )DPRPretrainedContextEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z$DPRPretrainedContextEncoder.__init__Nr   r   r   r   r   rS  	  s   rS  c                   @   s   e Zd ZdgZdd ZdS )DPRPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPRPreTrainedModel.__init__Nr   r   r   r   r   rT  	  s   rT  c                   @   s   e Zd ZdgZdd ZdS )DPRPretrainedQuestionEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z%DPRPretrainedQuestionEncoder.__init__Nr   r   r   r   r   rU  	  s   rU  c                   @   s   e Zd ZdgZdd ZdS )DPRPretrainedReaderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPRPretrainedReader.__init__Nr   r   r   r   r   rV  	  s   rV  c                   @   s   e Zd ZdgZdd ZdS )DPRQuestionEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPRQuestionEncoder.__init__Nr   r   r   r   r   rW  	  s   rW  c                   @   s   e Zd ZdgZdd ZdS )	DPRReaderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPRReader.__init__Nr   r   r   r   r   rX  	  s   rX  c                   @   s   e Zd ZdgZdd ZdS )DPTForDepthEstimationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPTForDepthEstimation.__init__Nr   r   r   r   r   rY  	  s   rY  c                   @   s   e Zd ZdgZdd ZdS )DPTForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z#DPTForSemanticSegmentation.__init__Nr   r   r   r   r   rZ  	  s   rZ  c                   @   s   e Zd ZdgZdd ZdS )DPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPTModel.__init__Nr   r   r   r   r   r[  	  s   r[  c                   @   s   e Zd ZdgZdd ZdS )DPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zDPTPreTrainedModel.__init__Nr   r   r   r   r   r\  	  s   r\  c                   @   s   e Zd ZdgZdd ZdS )%EfficientFormerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z.EfficientFormerForImageClassification.__init__Nr   r   r   r   r   r]  	  s   r]  c                   @   s   e Zd ZdgZdd ZdS )0EfficientFormerForImageClassificationWithTeacherr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    z9EfficientFormerForImageClassificationWithTeacher.__init__Nr   r   r   r   r   r^  	  s   r^  c                   @   s   e Zd ZdgZdd ZdS )EfficientFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zEfficientFormerModel.__init__Nr   r   r   r   r   r_  	  s   r_  c                   @   s   e Zd ZdgZdd ZdS )EfficientFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r    
  s    z'EfficientFormerPreTrainedModel.__init__Nr   r   r   r   r   r`  	  s   r`  c                   @   s   e Zd ZdgZdd ZdS )"EfficientNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   

  s    z+EfficientNetForImageClassification.__init__Nr   r   r   r   r   ra  
  s   ra  c                   @   s   e Zd ZdgZdd ZdS )EfficientNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zEfficientNetModel.__init__Nr   r   r   r   r   rb  
  s   rb  c                   @   s   e Zd ZdgZdd ZdS )EfficientNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z$EfficientNetPreTrainedModel.__init__Nr   r   r   r   r   rc  
  s   rc  c                   @   s   e Zd ZdgZdd ZdS )ElectraForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   "
  s    zElectraForCausalLM.__init__Nr   r   r   r   r   rd  
  s   rd  c                   @   s   e Zd ZdgZdd ZdS )ElectraForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )
  s    zElectraForMaskedLM.__init__Nr   r   r   r   r   re  &
  s   re  c                   @   s   e Zd ZdgZdd ZdS )ElectraForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   0
  s    z!ElectraForMultipleChoice.__init__Nr   r   r   r   r   rf  -
  s   rf  c                   @   s   e Zd ZdgZdd ZdS )ElectraForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7
  s    zElectraForPreTraining.__init__Nr   r   r   r   r   rg  4
  s   rg  c                   @   s   e Zd ZdgZdd ZdS )ElectraForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >
  s    z$ElectraForQuestionAnswering.__init__Nr   r   r   r   r   rh  ;
  s   rh  c                   @   s   e Zd ZdgZdd ZdS ) ElectraForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E
  s    z)ElectraForSequenceClassification.__init__Nr   r   r   r   r   ri  B
  s   ri  c                   @   s   e Zd ZdgZdd ZdS )ElectraForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L
  s    z&ElectraForTokenClassification.__init__Nr   r   r   r   r   rj  I
  s   rj  c                   @   s   e Zd ZdgZdd ZdS )ElectraModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S
  s    zElectraModel.__init__Nr   r   r   r   r   rk  P
  s   rk  c                   @   s   e Zd ZdgZdd ZdS )ElectraPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z
  s    zElectraPreTrainedModel.__init__Nr   r   r   r   r   rl  W
  s   rl  c                  O   s   t tdg d S r   )r   load_tf_weights_in_electrar<   r   r   r   rm  ^
  s    rm  c                   @   s   e Zd ZdgZdd ZdS )EncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e
  s    zEncoderDecoderModel.__init__Nr   r   r   r   r   rn  b
  s   rn  c                   @   s   e Zd ZdgZdd ZdS )ErnieForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o
  s    zErnieForCausalLM.__init__Nr   r   r   r   r   ro  l
  s   ro  c                   @   s   e Zd ZdgZdd ZdS )ErnieForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v
  s    zErnieForMaskedLM.__init__Nr   r   r   r   r   rp  s
  s   rp  c                   @   s   e Zd ZdgZdd ZdS )ErnieForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   }
  s    zErnieForMultipleChoice.__init__Nr   r   r   r   r   rq  z
  s   rq  c                   @   s   e Zd ZdgZdd ZdS )ErnieForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z'ErnieForNextSentencePrediction.__init__Nr   r   r   r   r   rr  
  s   rr  c                   @   s   e Zd ZdgZdd ZdS )ErnieForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zErnieForPreTraining.__init__Nr   r   r   r   r   rs  
  s   rs  c                   @   s   e Zd ZdgZdd ZdS )ErnieForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z"ErnieForQuestionAnswering.__init__Nr   r   r   r   r   rt  
  s   rt  c                   @   s   e Zd ZdgZdd ZdS )ErnieForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z'ErnieForSequenceClassification.__init__Nr   r   r   r   r   ru  
  s   ru  c                   @   s   e Zd ZdgZdd ZdS )ErnieForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z$ErnieForTokenClassification.__init__Nr   r   r   r   r   rv  
  s   rv  c                   @   s   e Zd ZdgZdd ZdS )
ErnieModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zErnieModel.__init__Nr   r   r   r   r   rw  
  s   rw  c                   @   s   e Zd ZdgZdd ZdS )ErniePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zErniePreTrainedModel.__init__Nr   r   r   r   r   rx  
  s   rx  c                   @   s   e Zd ZdgZdd ZdS )ErnieMForInformationExtractionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z'ErnieMForInformationExtraction.__init__Nr   r   r   r   r   ry  
  s   ry  c                   @   s   e Zd ZdgZdd ZdS )ErnieMForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z ErnieMForMultipleChoice.__init__Nr   r   r   r   r   rz  
  s   rz  c                   @   s   e Zd ZdgZdd ZdS )ErnieMForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z#ErnieMForQuestionAnswering.__init__Nr   r   r   r   r   r{  
  s   r{  c                   @   s   e Zd ZdgZdd ZdS )ErnieMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z(ErnieMForSequenceClassification.__init__Nr   r   r   r   r   r|  
  s   r|  c                   @   s   e Zd ZdgZdd ZdS )ErnieMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z%ErnieMForTokenClassification.__init__Nr   r   r   r   r   r}  
  s   r}  c                   @   s   e Zd ZdgZdd ZdS )ErnieMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zErnieMModel.__init__Nr   r   r   r   r   r~  
  s   r~  c                   @   s   e Zd ZdgZdd ZdS )ErnieMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zErnieMPreTrainedModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmFoldPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zEsmFoldPreTrainedModel.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zEsmForMaskedLM.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmForProteinFoldingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zEsmForProteinFolding.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%EsmForSequenceClassification.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"EsmForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zEsmModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )EsmPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zEsmPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z"FlaubertForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    z%FlaubertForQuestionAnswering.__init__Nr   r   r   r   r   r  $  s   r  c                   @   s   e Zd ZdgZdd ZdS )"FlaubertForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    z+FlaubertForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r  +  s   r  c                   @   s   e Zd ZdgZdd ZdS )!FlaubertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    z*FlaubertForSequenceClassification.__init__Nr   r   r   r   r   r  2  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    z'FlaubertForTokenClassification.__init__Nr   r   r   r   r   r  9  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    zFlaubertModel.__init__Nr   r   r   r   r   r  @  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   J  s    z FlaubertPreTrainedModel.__init__Nr   r   r   r   r   r  G  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlaubertWithLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Q  s    z FlaubertWithLMHeadModel.__init__Nr   r   r   r   r   r  N  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   [  s    zFlavaForPreTraining.__init__Nr   r   r   r   r   r  X  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaImageCodebookr   c                 O   s   t | dg d S r   r   r   r   r   r   r   b  s    zFlavaImageCodebook.__init__Nr   r   r   r   r   r  _  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaImageModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   i  s    zFlavaImageModel.__init__Nr   r   r   r   r   r  f  s   r  c                   @   s   e Zd ZdgZdd ZdS )
FlavaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   p  s    zFlavaModel.__init__Nr   r   r   r   r   r  m  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaMultimodalModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   w  s    zFlavaMultimodalModel.__init__Nr   r   r   r   r   r  t  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ~  s    zFlavaPreTrainedModel.__init__Nr   r   r   r   r   r  {  s   r  c                   @   s   e Zd ZdgZdd ZdS )FlavaTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFlavaTextModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FNetForNextSentencePrediction.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!FNetForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&FNetForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#FNetForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	FNetLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	FNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FocalNetBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFocalNetBackbone.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FocalNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'FocalNetForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FocalNetForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'FocalNetForMaskedImageModeling.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FocalNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFocalNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FocalNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z FocalNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FSMTForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%FSMTForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	FSMTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFSMTModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PretrainedFSMTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   	  s    zPretrainedFSMTModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelBaseModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFunnelBaseModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zFunnelForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   !  s    z FunnelForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   (  s    zFunnelForPreTraining.__init__Nr   r   r   r   r   r  %  s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   /  s    z#FunnelForQuestionAnswering.__init__Nr   r   r   r   r   r  ,  s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   6  s    z(FunnelForSequenceClassification.__init__Nr   r   r   r   r   r  3  s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   =  s    z%FunnelForTokenClassification.__init__Nr   r   r   r   r   r  :  s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   D  s    zFunnelModel.__init__Nr   r   r   r   r   r  A  s   r  c                   @   s   e Zd ZdgZdd ZdS )FunnelPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   K  s    zFunnelPreTrainedModel.__init__Nr   r   r   r   r   r  H  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_funnelr<   r   r   r   r  O  s    r  c                   @   s   e Zd ZdgZdd ZdS )GitForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    zGitForCausalLM.__init__Nr   r   r   r   r   r  V  s   r  c                   @   s   e Zd ZdgZdd ZdS )GitModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   `  s    zGitModel.__init__Nr   r   r   r   r   r  ]  s   r  c                   @   s   e Zd ZdgZdd ZdS )GitPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   g  s    zGitPreTrainedModel.__init__Nr   r   r   r   r   r  d  s   r  c                   @   s   e Zd ZdgZdd ZdS )GitVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   n  s    zGitVisionModel.__init__Nr   r   r   r   r   r  k  s   r  c                   @   s   e Zd ZdgZdd ZdS )GLPNForDepthEstimationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x  s    zGLPNForDepthEstimation.__init__Nr   r   r   r   r   r  u  s   r  c                   @   s   e Zd ZdgZdd ZdS )	GLPNModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGLPNModel.__init__Nr   r   r   r   r   r  |  s   r  c                   @   s   e Zd ZdgZdd ZdS )GLPNPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGLPNPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2DoubleHeadsModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPT2DoubleHeadsModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!GPT2ForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&GPT2ForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#GPT2ForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2LMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPT2LMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	GPT2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPT2Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPT2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPT2PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_gpt2r<   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )GPTBigCodeForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTBigCodeForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )#GPTBigCodeForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,GPTBigCodeForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) GPTBigCodeForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)GPTBigCodeForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTBigCodeModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTBigCodeModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTBigCodePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"GPTBigCodePreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTNeoForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#GPTNeoForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(GPTNeoForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%GPTNeoForTokenClassification.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zGPTNeoModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTNeoPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_gpt_neor<   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTNeoXForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    z$GPTNeoXForQuestionAnswering.__init__Nr   r   r   r   r   r  #  s   r  c                   @   s   e Zd ZdgZdd ZdS ) GPTNeoXForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z)GPTNeoXForSequenceClassification.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z&GPTNeoXForTokenClassification.__init__Nr   r   r   r   r   r  1  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zGPTNeoXLayer.__init__Nr   r   r   r   r   r  8  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zGPTNeoXModel.__init__Nr   r   r   r   r   r  ?  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    zGPTNeoXPreTrainedModel.__init__Nr   r   r   r   r   r  F  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXJapaneseForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z#GPTNeoXJapaneseForCausalLM.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXJapaneseLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    zGPTNeoXJapaneseLayer.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXJapaneseModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    zGPTNeoXJapaneseModel.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTNeoXJapanesePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z'GPTNeoXJapanesePreTrainedModel.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTJForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   r  s    zGPTJForCausalLM.__init__Nr   r   r   r   r   r  o  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTJForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    z!GPTJForQuestionAnswering.__init__Nr   r   r   r   r   r  v  s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTJForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&GPTJForSequenceClassification.__init__Nr   r   r   r   r   r  }  s   r  c                   @   s   e Zd ZdgZdd ZdS )	GPTJModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTJModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTJPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTJPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )&GPTSanJapaneseForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/GPTSanJapaneseForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTSanJapaneseModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGPTSanJapaneseModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GPTSanJapanesePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&GPTSanJapanesePreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) GraphormerForGraphClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)GraphormerForGraphClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GraphormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGraphormerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GraphormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"GraphormerPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GroupViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGroupViTModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GroupViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z GroupViTPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GroupViTTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGroupViTTextModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )GroupViTVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zGroupViTVisionModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )HubertForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zHubertForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )HubertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(HubertForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )HubertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zHubertModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )HubertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zHubertPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zIBertForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zIBertForMultipleChoice.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"IBertForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'IBertForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   "  s    z$IBertForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )
IBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )  s    zIBertModel.__init__Nr   r   r   r   r   r  &  s   r  c                   @   s   e Zd ZdgZdd ZdS )IBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   0  s    zIBertPreTrainedModel.__init__Nr   r   r   r   r   r  -  s   r  c                   @   s   e Zd ZdgZdd ZdS )ImageGPTForCausalImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    z'ImageGPTForCausalImageModeling.__init__Nr   r   r   r   r   r  7  s   r  c                   @   s   e Zd ZdgZdd ZdS )ImageGPTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   A  s    z'ImageGPTForImageClassification.__init__Nr   r   r   r   r   r  >  s   r  c                   @   s   e Zd ZdgZdd ZdS )ImageGPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   H  s    zImageGPTModel.__init__Nr   r   r   r   r   r  E  s   r  c                   @   s   e Zd ZdgZdd ZdS )ImageGPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   O  s    z ImageGPTPreTrainedModel.__init__Nr   r   r   r   r   r  L  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_imagegptr<   r   r   r   r  S  s    r  c                   @   s   e Zd ZdgZdd ZdS )InformerForPredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ]  s    zInformerForPrediction.__init__Nr   r   r   r   r   r  Z  s   r  c                   @   s   e Zd ZdgZdd ZdS )InformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   d  s    zInformerModel.__init__Nr   r   r   r   r   r  a  s   r  c                   @   s   e Zd ZdgZdd ZdS )InformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   k  s    z InformerPreTrainedModel.__init__Nr   r   r   r   r   r  h  s   r  c                   @   s   e Zd ZdgZdd ZdS )JukeboxModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   u  s    zJukeboxModel.__init__Nr   r   r   r   r   r  r  s   r  c                   @   s   e Zd ZdgZdd ZdS )JukeboxPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   |  s    zJukeboxPreTrainedModel.__init__Nr   r   r   r   r   r  y  s   r  c                   @   s   e Zd ZdgZdd ZdS )JukeboxPriorr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zJukeboxPrior.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )JukeboxVQVAEr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zJukeboxVQVAE.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLayoutLMForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%LayoutLMForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!LayoutLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*LayoutLMForSequenceClassification.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )LayoutLMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'LayoutLMForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLayoutLMModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z LayoutLMPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv2ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'LayoutLMv2ForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )#LayoutLMv2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,LayoutLMv2ForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) LayoutLMv2ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)LayoutLMv2ForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLayoutLMv2Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"LayoutLMv2PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv3ForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'LayoutLMv3ForQuestionAnswering.__init__Nr   r   r   r   r   r	    s   r	  c                   @   s   e Zd ZdgZdd ZdS )#LayoutLMv3ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,LayoutLMv3ForSequenceClassification.__init__Nr   r   r   r   r   r
    s   r
  c                   @   s   e Zd ZdgZdd ZdS ) LayoutLMv3ForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)LayoutLMv3ForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv3Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLayoutLMv3Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LayoutLMv3PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"LayoutLMv3PreTrainedModel.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )LEDForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$LEDForConditionalGeneration.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )LEDForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z LEDForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LEDForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%LEDForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LEDModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   "  s    zLEDModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LEDPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )  s    zLEDPreTrainedModel.__init__Nr   r   r   r   r   r  &  s   r  c                   @   s   e Zd ZdgZdd ZdS )LevitForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   3  s    z$LevitForImageClassification.__init__Nr   r   r   r   r   r  0  s   r  c                   @   s   e Zd ZdgZdd ZdS )&LevitForImageClassificationWithTeacherr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    z/LevitForImageClassificationWithTeacher.__init__Nr   r   r   r   r   r  7  s   r  c                   @   s   e Zd ZdgZdd ZdS )
LevitModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   A  s    zLevitModel.__init__Nr   r   r   r   r   r  >  s   r  c                   @   s   e Zd ZdgZdd ZdS )LevitPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   H  s    zLevitPreTrainedModel.__init__Nr   r   r   r   r   r  E  s   r  c                   @   s   e Zd ZdgZdd ZdS )LiltForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    z!LiltForQuestionAnswering.__init__Nr   r   r   r   r   r  O  s   r  c                   @   s   e Zd ZdgZdd ZdS )LiltForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    z&LiltForSequenceClassification.__init__Nr   r   r   r   r   r  V  s   r  c                   @   s   e Zd ZdgZdd ZdS )LiltForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   `  s    z#LiltForTokenClassification.__init__Nr   r   r   r   r   r  ]  s   r  c                   @   s   e Zd ZdgZdd ZdS )	LiltModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   g  s    zLiltModel.__init__Nr   r   r   r   r   r  d  s   r  c                   @   s   e Zd ZdgZdd ZdS )LiltPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   n  s    zLiltPreTrainedModel.__init__Nr   r   r   r   r   r  k  s   r  c                   @   s   e Zd ZdgZdd ZdS )LlamaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   u  s    zLlamaForCausalLM.__init__Nr   r   r   r   r   r  r  s   r  c                   @   s   e Zd ZdgZdd ZdS )LlamaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   |  s    z'LlamaForSequenceClassification.__init__Nr   r   r   r   r   r  y  s   r  c                   @   s   e Zd ZdgZdd ZdS )
LlamaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLlamaModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LlamaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLlamaPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )LongformerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLongformerForMaskedLM.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )LongformerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$LongformerForMultipleChoice.__init__Nr   r   r   r   r   r!    s   r!  c                   @   s   e Zd ZdgZdd ZdS )LongformerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'LongformerForQuestionAnswering.__init__Nr   r   r   r   r   r"    s   r"  c                   @   s   e Zd ZdgZdd ZdS )#LongformerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,LongformerForSequenceClassification.__init__Nr   r   r   r   r   r#    s   r#  c                   @   s   e Zd ZdgZdd ZdS ) LongformerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)LongformerForTokenClassification.__init__Nr   r   r   r   r   r$    s   r$  c                   @   s   e Zd ZdgZdd ZdS )LongformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLongformerModel.__init__Nr   r   r   r   r   r%    s   r%  c                   @   s   e Zd ZdgZdd ZdS )LongformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"LongformerPreTrainedModel.__init__Nr   r   r   r   r   r&    s   r&  c                   @   s   e Zd ZdgZdd ZdS )LongformerSelfAttentionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z LongformerSelfAttention.__init__Nr   r   r   r   r   r'    s   r'  c                   @   s   e Zd ZdgZdd ZdS )LongT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLongT5EncoderModel.__init__Nr   r   r   r   r   r(    s   r(  c                   @   s   e Zd ZdgZdd ZdS )LongT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'LongT5ForConditionalGeneration.__init__Nr   r   r   r   r   r)    s   r)  c                   @   s   e Zd ZdgZdd ZdS )LongT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLongT5Model.__init__Nr   r   r   r   r   r*    s   r*  c                   @   s   e Zd ZdgZdd ZdS )LongT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLongT5PreTrainedModel.__init__Nr   r   r   r   r   r+    s   r+  c                   @   s   e Zd ZdgZdd ZdS )LukeForEntityClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$LukeForEntityClassification.__init__Nr   r   r   r   r   r,    s   r,  c                   @   s   e Zd ZdgZdd ZdS )LukeForEntityPairClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(LukeForEntityPairClassification.__init__Nr   r   r   r   r   r-    s   r-  c                   @   s   e Zd ZdgZdd ZdS )LukeForEntitySpanClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(LukeForEntitySpanClassification.__init__Nr   r   r   r   r   r.    s   r.  c                   @   s   e Zd ZdgZdd ZdS )LukeForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zLukeForMaskedLM.__init__Nr   r   r   r   r   r/     s   r/  c                   @   s   e Zd ZdgZdd ZdS )LukeForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zLukeForMultipleChoice.__init__Nr   r   r   r   r   r0    s   r0  c                   @   s   e Zd ZdgZdd ZdS )LukeForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!LukeForQuestionAnswering.__init__Nr   r   r   r   r   r1    s   r1  c                   @   s   e Zd ZdgZdd ZdS )LukeForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&LukeForSequenceClassification.__init__Nr   r   r   r   r   r2    s   r2  c                   @   s   e Zd ZdgZdd ZdS )LukeForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#LukeForTokenClassification.__init__Nr   r   r   r   r   r3    s   r3  c                   @   s   e Zd ZdgZdd ZdS )	LukeModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zLukeModel.__init__Nr   r   r   r   r   r4  #  s   r4  c                   @   s   e Zd ZdgZdd ZdS )LukePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    zLukePreTrainedModel.__init__Nr   r   r   r   r   r5  *  s   r5  c                   @   s   e Zd ZdgZdd ZdS )LxmertEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    zLxmertEncoder.__init__Nr   r   r   r   r   r6  1  s   r6  c                   @   s   e Zd ZdgZdd ZdS )LxmertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zLxmertForPreTraining.__init__Nr   r   r   r   r   r7  8  s   r7  c                   @   s   e Zd ZdgZdd ZdS )LxmertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    z#LxmertForQuestionAnswering.__init__Nr   r   r   r   r   r8  ?  s   r8  c                   @   s   e Zd ZdgZdd ZdS )LxmertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    zLxmertModel.__init__Nr   r   r   r   r   r9  F  s   r9  c                   @   s   e Zd ZdgZdd ZdS )LxmertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    zLxmertPreTrainedModel.__init__Nr   r   r   r   r   r:  M  s   r:  c                   @   s   e Zd ZdgZdd ZdS )LxmertVisualFeatureEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    z#LxmertVisualFeatureEncoder.__init__Nr   r   r   r   r   r;  T  s   r;  c                   @   s   e Zd ZdgZdd ZdS )LxmertXLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    zLxmertXLayer.__init__Nr   r   r   r   r   r<  [  s   r<  c                   @   s   e Zd ZdgZdd ZdS )M2M100ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z'M2M100ForConditionalGeneration.__init__Nr   r   r   r   r   r=  e  s   r=  c                   @   s   e Zd ZdgZdd ZdS )M2M100Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zM2M100Model.__init__Nr   r   r   r   r   r>  l  s   r>  c                   @   s   e Zd ZdgZdd ZdS )M2M100PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    zM2M100PreTrainedModel.__init__Nr   r   r   r   r   r?  s  s   r?  c                   @   s   e Zd ZdgZdd ZdS )MarianForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    zMarianForCausalLM.__init__Nr   r   r   r   r   r@  z  s   r@  c                   @   s   e Zd ZdgZdd ZdS )MarianModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMarianModel.__init__Nr   r   r   r   r   rA    s   rA  c                   @   s   e Zd ZdgZdd ZdS )MarianMTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMarianMTModel.__init__Nr   r   r   r   r   rB    s   rB  c                   @   s   e Zd ZdgZdd ZdS )MarkupLMForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%MarkupLMForQuestionAnswering.__init__Nr   r   r   r   r   rC    s   rC  c                   @   s   e Zd ZdgZdd ZdS )!MarkupLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*MarkupLMForSequenceClassification.__init__Nr   r   r   r   r   rD    s   rD  c                   @   s   e Zd ZdgZdd ZdS )MarkupLMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'MarkupLMForTokenClassification.__init__Nr   r   r   r   r   rE    s   rE  c                   @   s   e Zd ZdgZdd ZdS )MarkupLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMarkupLMModel.__init__Nr   r   r   r   r   rF    s   rF  c                   @   s   e Zd ZdgZdd ZdS )MarkupLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z MarkupLMPreTrainedModel.__init__Nr   r   r   r   r   rG    s   rG  c                   @   s   e Zd ZdgZdd ZdS )#Mask2FormerForUniversalSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,Mask2FormerForUniversalSegmentation.__init__Nr   r   r   r   r   rH    s   rH  c                   @   s   e Zd ZdgZdd ZdS )Mask2FormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMask2FormerModel.__init__Nr   r   r   r   r   rI    s   rI  c                   @   s   e Zd ZdgZdd ZdS )Mask2FormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#Mask2FormerPreTrainedModel.__init__Nr   r   r   r   r   rJ    s   rJ  c                   @   s   e Zd ZdgZdd ZdS )!MaskFormerForInstanceSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*MaskFormerForInstanceSegmentation.__init__Nr   r   r   r   r   rK    s   rK  c                   @   s   e Zd ZdgZdd ZdS )MaskFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMaskFormerModel.__init__Nr   r   r   r   r   rL    s   rL  c                   @   s   e Zd ZdgZdd ZdS )MaskFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"MaskFormerPreTrainedModel.__init__Nr   r   r   r   r   rM    s   rM  c                   @   s   e Zd ZdgZdd ZdS )MaskFormerSwinBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMaskFormerSwinBackbone.__init__Nr   r   r   r   r   rN    s   rN  c                   @   s   e Zd ZdgZdd ZdS )MBartForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMBartForCausalLM.__init__Nr   r   r   r   r   rO    s   rO  c                   @   s   e Zd ZdgZdd ZdS )MBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&MBartForConditionalGeneration.__init__Nr   r   r   r   r   rP    s   rP  c                   @   s   e Zd ZdgZdd ZdS )MBartForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"MBartForQuestionAnswering.__init__Nr   r   r   r   r   rQ    s   rQ  c                   @   s   e Zd ZdgZdd ZdS )MBartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'MBartForSequenceClassification.__init__Nr   r   r   r   r   rR    s   rR  c                   @   s   e Zd ZdgZdd ZdS )
MBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMBartModel.__init__Nr   r   r   r   r   rS    s   rS  c                   @   s   e Zd ZdgZdd ZdS )MBartPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMBartPreTrainedModel.__init__Nr   r   r   r   r   rT    s   rT  c                   @   s   e Zd ZdgZdd ZdS )MCTCTForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMCTCTForCTC.__init__Nr   r   r   r   r   rU    s   rU  c                   @   s   e Zd ZdgZdd ZdS )
MCTCTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    zMCTCTModel.__init__Nr   r   r   r   r   rV     s   rV  c                   @   s   e Zd ZdgZdd ZdS )MCTCTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    zMCTCTPreTrainedModel.__init__Nr   r   r   r   r   rW  '  s   rW  c                   @   s   e Zd ZdgZdd ZdS )MegaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    zMegaForCausalLM.__init__Nr   r   r   r   r   rX  1  s   rX  c                   @   s   e Zd ZdgZdd ZdS )MegaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zMegaForMaskedLM.__init__Nr   r   r   r   r   rY  8  s   rY  c                   @   s   e Zd ZdgZdd ZdS )MegaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zMegaForMultipleChoice.__init__Nr   r   r   r   r   rZ  ?  s   rZ  c                   @   s   e Zd ZdgZdd ZdS )MegaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z!MegaForQuestionAnswering.__init__Nr   r   r   r   r   r[  F  s   r[  c                   @   s   e Zd ZdgZdd ZdS )MegaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    z&MegaForSequenceClassification.__init__Nr   r   r   r   r   r\  M  s   r\  c                   @   s   e Zd ZdgZdd ZdS )MegaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    z#MegaForTokenClassification.__init__Nr   r   r   r   r   r]  T  s   r]  c                   @   s   e Zd ZdgZdd ZdS )	MegaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    zMegaModel.__init__Nr   r   r   r   r   r^  [  s   r^  c                   @   s   e Zd ZdgZdd ZdS )MegaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e  s    zMegaPreTrainedModel.__init__Nr   r   r   r   r   r_  b  s   r_  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    z MegatronBertForCausalLM.__init__Nr   r   r   r   r   r`  l  s   r`  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z MegatronBertForMaskedLM.__init__Nr   r   r   r   r   ra  s  s   ra  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    z&MegatronBertForMultipleChoice.__init__Nr   r   r   r   r   rb  z  s   rb  c                   @   s   e Zd ZdgZdd ZdS )%MegatronBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.MegatronBertForNextSentencePrediction.__init__Nr   r   r   r   r   rc    s   rc  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#MegatronBertForPreTraining.__init__Nr   r   r   r   r   rd    s   rd  c                   @   s   e Zd ZdgZdd ZdS ) MegatronBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)MegatronBertForQuestionAnswering.__init__Nr   r   r   r   r   re    s   re  c                   @   s   e Zd ZdgZdd ZdS )%MegatronBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.MegatronBertForSequenceClassification.__init__Nr   r   r   r   r   rf    s   rf  c                   @   s   e Zd ZdgZdd ZdS )"MegatronBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+MegatronBertForTokenClassification.__init__Nr   r   r   r   r   rg    s   rg  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMegatronBertModel.__init__Nr   r   r   r   r   rh    s   rh  c                   @   s   e Zd ZdgZdd ZdS )MegatronBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$MegatronBertPreTrainedModel.__init__Nr   r   r   r   r   ri    s   ri  c                   @   s   e Zd ZdgZdd ZdS )MgpstrForSceneTextRecognitionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&MgpstrForSceneTextRecognition.__init__Nr   r   r   r   r   rj    s   rj  c                   @   s   e Zd ZdgZdd ZdS )MgpstrModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMgpstrModel.__init__Nr   r   r   r   r   rk    s   rk  c                   @   s   e Zd ZdgZdd ZdS )MgpstrPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMgpstrPreTrainedModel.__init__Nr   r   r   r   r   rl    s   rl  c                   @   s   e Zd ZdgZdd ZdS )MMBTForClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMMBTForClassification.__init__Nr   r   r   r   r   rm    s   rm  c                   @   s   e Zd ZdgZdd ZdS )	MMBTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMMBTModel.__init__Nr   r   r   r   r   rn    s   rn  c                   @   s   e Zd ZdgZdd ZdS )ModalEmbeddingsr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zModalEmbeddings.__init__Nr   r   r   r   r   ro    s   ro  c                   @   s   e Zd ZdgZdd ZdS )MobileBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMobileBertForMaskedLM.__init__Nr   r   r   r   r   rp    s   rp  c                   @   s   e Zd ZdgZdd ZdS )MobileBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$MobileBertForMultipleChoice.__init__Nr   r   r   r   r   rq    s   rq  c                   @   s   e Zd ZdgZdd ZdS )#MobileBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,MobileBertForNextSentencePrediction.__init__Nr   r   r   r   r   rr    s   rr  c                   @   s   e Zd ZdgZdd ZdS )MobileBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!MobileBertForPreTraining.__init__Nr   r   r   r   r   rs    s   rs  c                   @   s   e Zd ZdgZdd ZdS )MobileBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'MobileBertForQuestionAnswering.__init__Nr   r   r   r   r   rt    s   rt  c                   @   s   e Zd ZdgZdd ZdS )#MobileBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,MobileBertForSequenceClassification.__init__Nr   r   r   r   r   ru    s   ru  c                   @   s   e Zd ZdgZdd ZdS ) MobileBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)MobileBertForTokenClassification.__init__Nr   r   r   r   r   rv    s   rv  c                   @   s   e Zd ZdgZdd ZdS )MobileBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMobileBertLayer.__init__Nr   r   r   r   r   rw    s   rw  c                   @   s   e Zd ZdgZdd ZdS )MobileBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMobileBertModel.__init__Nr   r   r   r   r   rx    s   rx  c                   @   s   e Zd ZdgZdd ZdS )MobileBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $  s    z"MobileBertPreTrainedModel.__init__Nr   r   r   r   r   ry  !  s   ry  c                  O   s   t tdg d S r   )r   load_tf_weights_in_mobilebertr<   r   r   r   rz  (  s    rz  c                   @   s   e Zd ZdgZdd ZdS )!MobileNetV1ForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2  s    z*MobileNetV1ForImageClassification.__init__Nr   r   r   r   r   r{  /  s   r{  c                   @   s   e Zd ZdgZdd ZdS )MobileNetV1Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9  s    zMobileNetV1Model.__init__Nr   r   r   r   r   r|  6  s   r|  c                   @   s   e Zd ZdgZdd ZdS )MobileNetV1PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   @  s    z#MobileNetV1PreTrainedModel.__init__Nr   r   r   r   r   r}  =  s   r}  c                  O   s   t tdg d S r   )r   load_tf_weights_in_mobilenet_v1r<   r   r   r   r~  D  s    r~  c                   @   s   e Zd ZdgZdd ZdS )!MobileNetV2ForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N  s    z*MobileNetV2ForImageClassification.__init__Nr   r   r   r   r   r  K  s   r  c                   @   s   e Zd ZdgZdd ZdS )"MobileNetV2ForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U  s    z+MobileNetV2ForSemanticSegmentation.__init__Nr   r   r   r   r   r  R  s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileNetV2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    zMobileNetV2Model.__init__Nr   r   r   r   r   r  Y  s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileNetV2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c  s    z#MobileNetV2PreTrainedModel.__init__Nr   r   r   r   r   r  `  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_mobilenet_v2r<   r   r   r   r  g  s    r  c                   @   s   e Zd ZdgZdd ZdS )MobileViTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q  s    z(MobileViTForImageClassification.__init__Nr   r   r   r   r   r  n  s   r  c                   @   s   e Zd ZdgZdd ZdS ) MobileViTForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x  s    z)MobileViTForSemanticSegmentation.__init__Nr   r   r   r   r   r  u  s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMobileViTModel.__init__Nr   r   r   r   r   r  |  s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!MobileViTPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!MobileViTV2ForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*MobileViTV2ForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"MobileViTV2ForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+MobileViTV2ForSemanticSegmentation.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileViTV2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMobileViTV2Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MobileViTV2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#MobileViTV2PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMPNetForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMPNetForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"MPNetForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'MPNetForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$MPNetForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )
MPNetLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMPNetLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )
MPNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMPNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MPNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMPNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MT5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMT5EncoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MT5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$MT5ForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMT5Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMT5PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zMvpForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$MvpForConditionalGeneration.__init__Nr   r   r   r   r   r  
  s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z MvpForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%MvpForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   "  s    zMvpModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )MvpPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )  s    zMvpPreTrainedModel.__init__Nr   r   r   r   r   r  &  s   r  c                   @   s   e Zd ZdgZdd ZdS )NatBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r   3  s    zNatBackbone.__init__Nr   r   r   r   r   r  0  s   r  c                   @   s   e Zd ZdgZdd ZdS )NatForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    z"NatForImageClassification.__init__Nr   r   r   r   r   r  7  s   r  c                   @   s   e Zd ZdgZdd ZdS )NatModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   A  s    zNatModel.__init__Nr   r   r   r   r   r  >  s   r  c                   @   s   e Zd ZdgZdd ZdS )NatPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   H  s    zNatPreTrainedModel.__init__Nr   r   r   r   r   r  E  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    zNezhaForMaskedLM.__init__Nr   r   r   r   r   r  O  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    zNezhaForMultipleChoice.__init__Nr   r   r   r   r   r  V  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   `  s    z'NezhaForNextSentencePrediction.__init__Nr   r   r   r   r   r  ]  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   g  s    zNezhaForPreTraining.__init__Nr   r   r   r   r   r  d  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   n  s    z"NezhaForQuestionAnswering.__init__Nr   r   r   r   r   r  k  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   u  s    z'NezhaForSequenceClassification.__init__Nr   r   r   r   r   r  r  s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   |  s    z$NezhaForTokenClassification.__init__Nr   r   r   r   r   r  y  s   r  c                   @   s   e Zd ZdgZdd ZdS )
NezhaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNezhaModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NezhaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNezhaPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NllbMoeForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(NllbMoeForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NllbMoeModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNllbMoeModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NllbMoePreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNllbMoePreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NllbMoeSparseMLPr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNllbMoeSparseMLP.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NllbMoeTop2Routerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNllbMoeTop2Router.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NystromformerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!NystromformerForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NystromformerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'NystromformerForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!NystromformerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*NystromformerForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )&NystromformerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/NystromformerForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )#NystromformerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,NystromformerForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NystromformerLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNystromformerLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NystromformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zNystromformerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )NystromformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%NystromformerPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!OneFormerForUniversalSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*OneFormerForUniversalSegmentation.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OneFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOneFormerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OneFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!OneFormerPreTrainedModel.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenLlamaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zOpenLlamaForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"OpenLlamaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+OpenLlamaForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenLlamaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOpenLlamaModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenLlamaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!OpenLlamaPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenAIGPTDoubleHeadsModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   )  s    z"OpenAIGPTDoubleHeadsModel.__init__Nr   r   r   r   r   r  &  s   r  c                   @   s   e Zd ZdgZdd ZdS )"OpenAIGPTForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   0  s    z+OpenAIGPTForSequenceClassification.__init__Nr   r   r   r   r   r  -  s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenAIGPTLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7  s    zOpenAIGPTLMHeadModel.__init__Nr   r   r   r   r   r  4  s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenAIGPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    zOpenAIGPTModel.__init__Nr   r   r   r   r   r  ;  s   r  c                   @   s   e Zd ZdgZdd ZdS )OpenAIGPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    z!OpenAIGPTPreTrainedModel.__init__Nr   r   r   r   r   r  B  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_openai_gptr<   r   r   r   r  I  s    r  c                   @   s   e Zd ZdgZdd ZdS )OPTForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    zOPTForCausalLM.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )OPTForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    z OPTForQuestionAnswering.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )OPTForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z%OPTForSequenceClassification.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )OPTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    zOPTModel.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )OPTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zOPTPreTrainedModel.__init__Nr   r   r   r   r   r  l  s   r  c                   @   s   e Zd ZdgZdd ZdS )OwlViTForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    z!OwlViTForObjectDetection.__init__Nr   r   r   r   r   r  v  s   r  c                   @   s   e Zd ZdgZdd ZdS )OwlViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOwlViTModel.__init__Nr   r   r   r   r   r  }  s   r  c                   @   s   e Zd ZdgZdd ZdS )OwlViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOwlViTPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OwlViTTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOwlViTTextModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )OwlViTVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zOwlViTVisionModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPegasusForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(PegasusForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPegasusModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPegasusPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) PegasusXForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)PegasusXForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusXModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPegasusXModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PegasusXPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z PegasusXPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )-PerceiverForImageClassificationConvProcessingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z6PerceiverForImageClassificationConvProcessing.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )&PerceiverForImageClassificationFourierr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/PerceiverForImageClassificationFourier.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )&PerceiverForImageClassificationLearnedr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z/PerceiverForImageClassificationLearned.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PerceiverForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPerceiverForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"PerceiverForMultimodalAutoencodingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+PerceiverForMultimodalAutoencoding.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PerceiverForOpticalFlowr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z PerceiverForOpticalFlow.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"PerceiverForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+PerceiverForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PerceiverLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPerceiverLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PerceiverModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zPerceiverModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )PerceiverPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!PerceiverPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"Pix2StructForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+Pix2StructForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Pix2StructPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    z"Pix2StructPreTrainedModel.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )Pix2StructTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    zPix2StructTextModel.__init__Nr   r   r   r   r   r  '  s   r  c                   @   s   e Zd ZdgZdd ZdS )Pix2StructVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   1  s    zPix2StructVisionModel.__init__Nr   r   r   r   r   r  .  s   r  c                   @   s   e Zd ZdgZdd ZdS )PLBartForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    zPLBartForCausalLM.__init__Nr   r   r   r   r   r  8  s   r  c                   @   s   e Zd ZdgZdd ZdS )PLBartForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    z'PLBartForConditionalGeneration.__init__Nr   r   r   r   r   r  ?  s   r  c                   @   s   e Zd ZdgZdd ZdS )PLBartForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z(PLBartForSequenceClassification.__init__Nr   r   r   r   r   r  F  s   r  c                   @   s   e Zd ZdgZdd ZdS )PLBartModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    zPLBartModel.__init__Nr   r   r   r   r   r  M  s   r  c                   @   s   e Zd ZdgZdd ZdS )PLBartPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    zPLBartPreTrainedModel.__init__Nr   r   r   r   r   r  T  s   r  c                   @   s   e Zd ZdgZdd ZdS ) PoolFormerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z)PoolFormerForImageClassification.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )PoolFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    zPoolFormerModel.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )PoolFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    z"PoolFormerPreTrainedModel.__init__Nr   r   r   r   r   r  l  s   r  c                   @   s   e Zd ZdgZdd ZdS )ProphetNetDecoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    zProphetNetDecoder.__init__Nr   r   r   r   r   r  v  s   r  c                   @   s   e Zd ZdgZdd ZdS )ProphetNetEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zProphetNetEncoder.__init__Nr   r   r   r   r   r  }  s   r  c                   @   s   e Zd ZdgZdd ZdS )ProphetNetForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zProphetNetForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"ProphetNetForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+ProphetNetForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ProphetNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zProphetNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ProphetNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"ProphetNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zQDQBertForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!QDQBertForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) QDQBertForNextSentencePredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)QDQBertForNextSentencePrediction.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$QDQBertForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) QDQBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)QDQBertForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&QDQBertForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zQDQBertLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zQDQBertLMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zQDQBertModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )QDQBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zQDQBertPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_qdqbertr<   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )RagModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRagModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RagPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRagPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RagSequenceForGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RagSequenceForGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RagTokenForGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRagTokenForGeneration.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )RealmEmbedderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRealmEmbedder.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmForOpenQAr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRealmForOpenQA.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmKnowledgeAugEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RealmKnowledgeAugEncoder.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   $  s    zRealmPreTrainedModel.__init__Nr   r   r   r   r   r  !  s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmReaderr   c                 O   s   t | dg d S r   r   r   r   r   r   r   +  s    zRealmReader.__init__Nr   r   r   r   r   r  (  s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmRetrieverr   c                 O   s   t | dg d S r   r   r   r   r   r   r   2  s    zRealmRetriever.__init__Nr   r   r   r   r   r  /  s   r  c                   @   s   e Zd ZdgZdd ZdS )RealmScorerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   9  s    zRealmScorer.__init__Nr   r   r   r   r   r  6  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_realmr<   r   r   r   r  =  s    r  c                   @   s   e Zd ZdgZdd ZdS )ReformerAttentionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   G  s    zReformerAttention.__init__Nr   r   r   r   r   r	  D  s   r	  c                   @   s   e Zd ZdgZdd ZdS )ReformerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   N  s    zReformerForMaskedLM.__init__Nr   r   r   r   r   r
  K  s   r
  c                   @   s   e Zd ZdgZdd ZdS )ReformerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   U  s    z%ReformerForQuestionAnswering.__init__Nr   r   r   r   r   r  R  s   r  c                   @   s   e Zd ZdgZdd ZdS )!ReformerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   \  s    z*ReformerForSequenceClassification.__init__Nr   r   r   r   r   r  Y  s   r  c                   @   s   e Zd ZdgZdd ZdS )ReformerLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   c  s    zReformerLayer.__init__Nr   r   r   r   r   r  `  s   r  c                   @   s   e Zd ZdgZdd ZdS )ReformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j  s    zReformerModel.__init__Nr   r   r   r   r   r  g  s   r  c                   @   s   e Zd ZdgZdd ZdS )ReformerModelWithLMHeadr   c                 O   s   t | dg d S r   r   r   r   r   r   r   q  s    z ReformerModelWithLMHead.__init__Nr   r   r   r   r   r  n  s   r  c                   @   s   e Zd ZdgZdd ZdS )ReformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   x  s    z ReformerPreTrainedModel.__init__Nr   r   r   r   r   r  u  s   r  c                   @   s   e Zd ZdgZdd ZdS )RegNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%RegNetForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RegNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRegNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RegNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRegNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRemBertForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRemBertForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RemBertForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$RemBertForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) RemBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)RemBertForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&RemBertForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRemBertLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRemBertModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )RemBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRemBertPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_rembertr<   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )ResNetBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zResNetBackbone.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ResNetForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%ResNetForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ResNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zResNetModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )ResNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zResNetPreTrainedModel.__init__Nr   r   r   r   r   r!    s   r!  c                   @   s   e Zd ZdgZdd ZdS )RetriBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRetriBertModel.__init__Nr   r   r   r   r   r"    s   r"  c                   @   s   e Zd ZdgZdd ZdS )RetriBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RetriBertPreTrainedModel.__init__Nr   r   r   r   r   r#    s   r#  c                   @   s   e Zd ZdgZdd ZdS )RobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRobertaForCausalLM.__init__Nr   r   r   r   r   r$    s   r$  c                   @   s   e Zd ZdgZdd ZdS )RobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRobertaForMaskedLM.__init__Nr   r   r   r   r   r%    s   r%  c                   @   s   e Zd ZdgZdd ZdS )RobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RobertaForMultipleChoice.__init__Nr   r   r   r   r   r&    s   r&  c                   @   s   e Zd ZdgZdd ZdS )RobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   %  s    z$RobertaForQuestionAnswering.__init__Nr   r   r   r   r   r'  "  s   r'  c                   @   s   e Zd ZdgZdd ZdS ) RobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ,  s    z)RobertaForSequenceClassification.__init__Nr   r   r   r   r   r(  )  s   r(  c                   @   s   e Zd ZdgZdd ZdS )RobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   3  s    z&RobertaForTokenClassification.__init__Nr   r   r   r   r   r)  0  s   r)  c                   @   s   e Zd ZdgZdd ZdS )RobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   :  s    zRobertaModel.__init__Nr   r   r   r   r   r*  7  s   r*  c                   @   s   e Zd ZdgZdd ZdS )RobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   A  s    zRobertaPreTrainedModel.__init__Nr   r   r   r   r   r+  >  s   r+  c                   @   s   e Zd ZdgZdd ZdS )RobertaPreLayerNormForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   K  s    z'RobertaPreLayerNormForCausalLM.__init__Nr   r   r   r   r   r,  H  s   r,  c                   @   s   e Zd ZdgZdd ZdS )RobertaPreLayerNormForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   R  s    z'RobertaPreLayerNormForMaskedLM.__init__Nr   r   r   r   r   r-  O  s   r-  c                   @   s   e Zd ZdgZdd ZdS )$RobertaPreLayerNormForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   Y  s    z-RobertaPreLayerNormForMultipleChoice.__init__Nr   r   r   r   r   r.  V  s   r.  c                   @   s   e Zd ZdgZdd ZdS )'RobertaPreLayerNormForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   `  s    z0RobertaPreLayerNormForQuestionAnswering.__init__Nr   r   r   r   r   r/  ]  s   r/  c                   @   s   e Zd ZdgZdd ZdS ),RobertaPreLayerNormForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   g  s    z5RobertaPreLayerNormForSequenceClassification.__init__Nr   r   r   r   r   r0  d  s   r0  c                   @   s   e Zd ZdgZdd ZdS ))RobertaPreLayerNormForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   n  s    z2RobertaPreLayerNormForTokenClassification.__init__Nr   r   r   r   r   r1  k  s   r1  c                   @   s   e Zd ZdgZdd ZdS )RobertaPreLayerNormModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   u  s    z!RobertaPreLayerNormModel.__init__Nr   r   r   r   r   r2  r  s   r2  c                   @   s   e Zd ZdgZdd ZdS )"RobertaPreLayerNormPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   |  s    z+RobertaPreLayerNormPreTrainedModel.__init__Nr   r   r   r   r   r3  y  s   r3  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertForCausalLM.__init__Nr   r   r   r   r   r4    s   r4  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertForMaskedLM.__init__Nr   r   r   r   r   r5    s   r5  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!RoCBertForMultipleChoice.__init__Nr   r   r   r   r   r6    s   r6  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertForPreTraining.__init__Nr   r   r   r   r   r7    s   r7  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$RoCBertForQuestionAnswering.__init__Nr   r   r   r   r   r8    s   r8  c                   @   s   e Zd ZdgZdd ZdS ) RoCBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)RoCBertForSequenceClassification.__init__Nr   r   r   r   r   r9    s   r9  c                   @   s   e Zd ZdgZdd ZdS )RoCBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&RoCBertForTokenClassification.__init__Nr   r   r   r   r   r:    s   r:  c                   @   s   e Zd ZdgZdd ZdS )RoCBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertLayer.__init__Nr   r   r   r   r   r;    s   r;  c                   @   s   e Zd ZdgZdd ZdS )RoCBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertModel.__init__Nr   r   r   r   r   r<    s   r<  c                   @   s   e Zd ZdgZdd ZdS )RoCBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoCBertPreTrainedModel.__init__Nr   r   r   r   r   r=    s   r=  c                  O   s   t tdg d S r   )r   load_tf_weights_in_roc_bertr<   r   r   r   r>    s    r>  c                   @   s   e Zd ZdgZdd ZdS )RoFormerForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoFormerForCausalLM.__init__Nr   r   r   r   r   r?    s   r?  c                   @   s   e Zd ZdgZdd ZdS )RoFormerForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoFormerForMaskedLM.__init__Nr   r   r   r   r   r@    s   r@  c                   @   s   e Zd ZdgZdd ZdS )RoFormerForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"RoFormerForMultipleChoice.__init__Nr   r   r   r   r   rA    s   rA  c                   @   s   e Zd ZdgZdd ZdS )RoFormerForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%RoFormerForQuestionAnswering.__init__Nr   r   r   r   r   rB    s   rB  c                   @   s   e Zd ZdgZdd ZdS )!RoFormerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*RoFormerForSequenceClassification.__init__Nr   r   r   r   r   rC    s   rC  c                   @   s   e Zd ZdgZdd ZdS )RoFormerForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'RoFormerForTokenClassification.__init__Nr   r   r   r   r   rD    s   rD  c                   @   s   e Zd ZdgZdd ZdS )RoFormerLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoFormerLayer.__init__Nr   r   r   r   r   rE    s   rE  c                   @   s   e Zd ZdgZdd ZdS )RoFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRoFormerModel.__init__Nr   r   r   r   r   rF    s   rF  c                   @   s   e Zd ZdgZdd ZdS )RoFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z RoFormerPreTrainedModel.__init__Nr   r   r   r   r   rG    s   rG  c                  O   s   t tdg d S r   )r   load_tf_weights_in_roformerr<   r   r   r   rH    s    rH  c                   @   s   e Zd ZdgZdd ZdS )RwkvForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zRwkvForCausalLM.__init__Nr   r   r   r   r   rI    s   rI  c                   @   s   e Zd ZdgZdd ZdS )	RwkvModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zRwkvModel.__init__Nr   r   r   r   r   rJ    s   rJ  c                   @   s   e Zd ZdgZdd ZdS )RwkvPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    zRwkvPreTrainedModel.__init__Nr   r   r   r   r   rK  $  s   rK  c                   @   s   e Zd ZdgZdd ZdS )SamModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   1  s    zSamModel.__init__Nr   r   r   r   r   rL  .  s   rL  c                   @   s   e Zd ZdgZdd ZdS )SamPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   8  s    zSamPreTrainedModel.__init__Nr   r   r   r   r   rM  5  s   rM  c                   @   s   e Zd ZdgZdd ZdS )SegformerDecodeHeadr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zSegformerDecodeHead.__init__Nr   r   r   r   r   rN  ?  s   rN  c                   @   s   e Zd ZdgZdd ZdS )SegformerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z(SegformerForImageClassification.__init__Nr   r   r   r   r   rO  F  s   rO  c                   @   s   e Zd ZdgZdd ZdS ) SegformerForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    z)SegformerForSemanticSegmentation.__init__Nr   r   r   r   r   rP  M  s   rP  c                   @   s   e Zd ZdgZdd ZdS )SegformerLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    zSegformerLayer.__init__Nr   r   r   r   r   rQ  T  s   rQ  c                   @   s   e Zd ZdgZdd ZdS )SegformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    zSegformerModel.__init__Nr   r   r   r   r   rR  [  s   rR  c                   @   s   e Zd ZdgZdd ZdS )SegformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   e  s    z!SegformerPreTrainedModel.__init__Nr   r   r   r   r   rS  b  s   rS  c                   @   s   e Zd ZdgZdd ZdS )	SEWForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zSEWForCTC.__init__Nr   r   r   r   r   rT  l  s   rT  c                   @   s   e Zd ZdgZdd ZdS )SEWForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z%SEWForSequenceClassification.__init__Nr   r   r   r   r   rU  s  s   rU  c                   @   s   e Zd ZdgZdd ZdS )SEWModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    zSEWModel.__init__Nr   r   r   r   r   rV  z  s   rV  c                   @   s   e Zd ZdgZdd ZdS )SEWPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSEWPreTrainedModel.__init__Nr   r   r   r   r   rW    s   rW  c                   @   s   e Zd ZdgZdd ZdS )
SEWDForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSEWDForCTC.__init__Nr   r   r   r   r   rX    s   rX  c                   @   s   e Zd ZdgZdd ZdS )SEWDForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&SEWDForSequenceClassification.__init__Nr   r   r   r   r   rY    s   rY  c                   @   s   e Zd ZdgZdd ZdS )	SEWDModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSEWDModel.__init__Nr   r   r   r   r   rZ    s   rZ  c                   @   s   e Zd ZdgZdd ZdS )SEWDPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSEWDPreTrainedModel.__init__Nr   r   r   r   r   r[    s   r[  c                   @   s   e Zd ZdgZdd ZdS )SpeechEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"SpeechEncoderDecoderModel.__init__Nr   r   r   r   r   r\    s   r\  c                   @   s   e Zd ZdgZdd ZdS )#Speech2TextForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,Speech2TextForConditionalGeneration.__init__Nr   r   r   r   r   r]    s   r]  c                   @   s   e Zd ZdgZdd ZdS )Speech2TextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSpeech2TextModel.__init__Nr   r   r   r   r   r^    s   r^  c                   @   s   e Zd ZdgZdd ZdS )Speech2TextPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#Speech2TextPreTrainedModel.__init__Nr   r   r   r   r   r_    s   r_  c                   @   s   e Zd ZdgZdd ZdS )Speech2Text2ForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Speech2Text2ForCausalLM.__init__Nr   r   r   r   r   r`    s   r`  c                   @   s   e Zd ZdgZdd ZdS )Speech2Text2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$Speech2Text2PreTrainedModel.__init__Nr   r   r   r   r   ra    s   ra  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5ForSpeechToSpeechr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"SpeechT5ForSpeechToSpeech.__init__Nr   r   r   r   r   rb    s   rb  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5ForSpeechToTextr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z SpeechT5ForSpeechToText.__init__Nr   r   r   r   r   rc    s   rc  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5ForTextToSpeechr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z SpeechT5ForTextToSpeech.__init__Nr   r   r   r   r   rd    s   rd  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5HifiGanr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSpeechT5HifiGan.__init__Nr   r   r   r   r   re    s   re  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSpeechT5Model.__init__Nr   r   r   r   r   rf    s   rf  c                   @   s   e Zd ZdgZdd ZdS )SpeechT5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z SpeechT5PreTrainedModel.__init__Nr   r   r   r   r   rg    s   rg  c                   @   s   e Zd ZdgZdd ZdS )SplinterForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSplinterForPreTraining.__init__Nr   r   r   r   r   rh    s   rh  c                   @   s   e Zd ZdgZdd ZdS )SplinterForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%SplinterForQuestionAnswering.__init__Nr   r   r   r   r   ri    s   ri  c                   @   s   e Zd ZdgZdd ZdS )SplinterLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSplinterLayer.__init__Nr   r   r   r   r   rj    s   rj  c                   @   s   e Zd ZdgZdd ZdS )SplinterModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSplinterModel.__init__Nr   r   r   r   r   rk    s   rk  c                   @   s   e Zd ZdgZdd ZdS )SplinterPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    z SplinterPreTrainedModel.__init__Nr   r   r   r   r   rl     s   rl  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    zSqueezeBertForMaskedLM.__init__Nr   r   r   r   r   rm  *  s   rm  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z%SqueezeBertForMultipleChoice.__init__Nr   r   r   r   r   rn  1  s   rn  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    z(SqueezeBertForQuestionAnswering.__init__Nr   r   r   r   r   ro  8  s   ro  c                   @   s   e Zd ZdgZdd ZdS )$SqueezeBertForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    z-SqueezeBertForSequenceClassification.__init__Nr   r   r   r   r   rp  ?  s   rp  c                   @   s   e Zd ZdgZdd ZdS )!SqueezeBertForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    z*SqueezeBertForTokenClassification.__init__Nr   r   r   r   r   rq  F  s   rq  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   P  s    zSqueezeBertModel.__init__Nr   r   r   r   r   rr  M  s   rr  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertModuler   c                 O   s   t | dg d S r   r   r   r   r   r   r   W  s    zSqueezeBertModule.__init__Nr   r   r   r   r   rs  T  s   rs  c                   @   s   e Zd ZdgZdd ZdS )SqueezeBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ^  s    z#SqueezeBertPreTrainedModel.__init__Nr   r   r   r   r   rt  [  s   rt  c                   @   s   e Zd ZdgZdd ZdS )!SwiftFormerForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z*SwiftFormerForImageClassification.__init__Nr   r   r   r   r   ru  e  s   ru  c                   @   s   e Zd ZdgZdd ZdS )SwiftFormerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zSwiftFormerModel.__init__Nr   r   r   r   r   rv  l  s   rv  c                   @   s   e Zd ZdgZdd ZdS )SwiftFormerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z#SwiftFormerPreTrainedModel.__init__Nr   r   r   r   r   rw  s  s   rw  c                   @   s   e Zd ZdgZdd ZdS )SwinBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwinBackbone.__init__Nr   r   r   r   r   rx  }  s   rx  c                   @   s   e Zd ZdgZdd ZdS )SwinForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#SwinForImageClassification.__init__Nr   r   r   r   r   ry    s   ry  c                   @   s   e Zd ZdgZdd ZdS )SwinForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#SwinForMaskedImageModeling.__init__Nr   r   r   r   r   rz    s   rz  c                   @   s   e Zd ZdgZdd ZdS )	SwinModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwinModel.__init__Nr   r   r   r   r   r{    s   r{  c                   @   s   e Zd ZdgZdd ZdS )SwinPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwinPreTrainedModel.__init__Nr   r   r   r   r   r|    s   r|  c                   @   s   e Zd ZdgZdd ZdS )Swin2SRForImageSuperResolutionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'Swin2SRForImageSuperResolution.__init__Nr   r   r   r   r   r}    s   r}  c                   @   s   e Zd ZdgZdd ZdS )Swin2SRModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwin2SRModel.__init__Nr   r   r   r   r   r~    s   r~  c                   @   s   e Zd ZdgZdd ZdS )Swin2SRPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwin2SRPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Swinv2ForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%Swinv2ForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Swinv2ForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%Swinv2ForMaskedImageModeling.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Swinv2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwinv2Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Swinv2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSwinv2PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )SwitchTransformersEncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'SwitchTransformersEncoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )*SwitchTransformersForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z3SwitchTransformersForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )SwitchTransformersModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z SwitchTransformersModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!SwitchTransformersPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*SwitchTransformersPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )SwitchTransformersSparseMLPr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$SwitchTransformersSparseMLP.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )SwitchTransformersTop1Routerr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z%SwitchTransformersTop1Router.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )T5EncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    zT5EncoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )T5ForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#T5ForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )T5Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zT5Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )T5PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zT5PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_t5r<   r   r   r   r  #  s    r  c                   @   s   e Zd ZdgZdd ZdS )"TableTransformerForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z+TableTransformerForObjectDetection.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )TableTransformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    zTableTransformerModel.__init__Nr   r   r   r   r   r  1  s   r  c                   @   s   e Zd ZdgZdd ZdS )TableTransformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    z(TableTransformerPreTrainedModel.__init__Nr   r   r   r   r   r  8  s   r  c                   @   s   e Zd ZdgZdd ZdS )TapasForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    zTapasForMaskedLM.__init__Nr   r   r   r   r   r  B  s   r  c                   @   s   e Zd ZdgZdd ZdS )TapasForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    z"TapasForQuestionAnswering.__init__Nr   r   r   r   r   r  I  s   r  c                   @   s   e Zd ZdgZdd ZdS )TapasForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z'TapasForSequenceClassification.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )
TapasModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    zTapasModel.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )TapasPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    zTapasPreTrainedModel.__init__Nr   r   r   r   r   r  ^  s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_tapasr<   r   r   r   r  e  s    r  c                   @   s   e Zd ZdgZdd ZdS )"TimeSeriesTransformerForPredictionr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    z+TimeSeriesTransformerForPrediction.__init__Nr   r   r   r   r   r  l  s   r  c                   @   s   e Zd ZdgZdd ZdS )TimeSeriesTransformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   v  s    z#TimeSeriesTransformerModel.__init__Nr   r   r   r   r   r  s  s   r  c                   @   s   e Zd ZdgZdd ZdS )$TimeSeriesTransformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   }  s    z-TimeSeriesTransformerPreTrainedModel.__init__Nr   r   r   r   r   r  z  s   r  c                   @   s   e Zd ZdgZdd ZdS )!TimesformerForVideoClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*TimesformerForVideoClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TimesformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTimesformerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TimesformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TimesformerPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TimmBackboner   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTimmBackbone.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TrajectoryTransformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#TrajectoryTransformerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )$TrajectoryTransformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-TrajectoryTransformerPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )AdaptiveEmbeddingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zAdaptiveEmbedding.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"TransfoXLForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+TransfoXLForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TransfoXLLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTransfoXLLMHeadModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TransfoXLModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTransfoXLModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TransfoXLPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!TransfoXLPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                  O   s   t tdg d S r   )r   load_tf_weights_in_transfo_xlr<   r   r   r   r    s    r  c                   @   s   e Zd ZdgZdd ZdS )TrOCRForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTrOCRForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TrOCRPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTrOCRPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) TvltForAudioVisualClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)TvltForAudioVisualClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TvltForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTvltForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	TvltModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zTvltModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )TvltPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zTvltPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zUniSpeechForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z UniSpeechForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"UniSpeechForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+UniSpeechForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zUniSpeechModel.__init__Nr   r   r   r   r   r  #  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z!UniSpeechPreTrainedModel.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )'UniSpeechSatForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7  s    z0UniSpeechSatForAudioFrameClassification.__init__Nr   r   r   r   r   r  4  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechSatForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    zUniSpeechSatForCTC.__init__Nr   r   r   r   r   r  ;  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechSatForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    z#UniSpeechSatForPreTraining.__init__Nr   r   r   r   r   r  B  s   r  c                   @   s   e Zd ZdgZdd ZdS )%UniSpeechSatForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    z.UniSpeechSatForSequenceClassification.__init__Nr   r   r   r   r   r  I  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechSatForXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    zUniSpeechSatForXVector.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechSatModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    zUniSpeechSatModel.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )UniSpeechSatPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z$UniSpeechSatPreTrainedModel.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )UperNetForSemanticSegmentationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z'UperNetForSemanticSegmentation.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )UperNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   o  s    zUperNetPreTrainedModel.__init__Nr   r   r   r   r   r  l  s   r  c                   @   s   e Zd ZdgZdd ZdS )VanForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    z"VanForImageClassification.__init__Nr   r   r   r   r   r  v  s   r  c                   @   s   e Zd ZdgZdd ZdS )VanModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zVanModel.__init__Nr   r   r   r   r   r  }  s   r  c                   @   s   e Zd ZdgZdd ZdS )VanPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zVanPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VideoMAEForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zVideoMAEForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VideoMAEForVideoClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'VideoMAEForVideoClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VideoMAEModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zVideoMAEModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VideoMAEPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z VideoMAEPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViltForImageAndTextRetrievalr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%ViltForImageAndTextRetrieval.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )"ViltForImagesAndTextClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z+ViltForImagesAndTextClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViltForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViltForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViltForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!ViltForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViltForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#ViltForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	ViltLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViltLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )	ViltModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViltModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViltPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViltPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisionEncoderDecoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"VisionEncoderDecoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisionTextDualEncoderModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#VisionTextDualEncoderModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$VisualBertForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z!VisualBertForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'VisualBertForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )$VisualBertForRegionToPhraseAlignmentr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z-VisualBertForRegionToPhraseAlignment.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertForVisualReasoningr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%VisualBertForVisualReasoning.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zVisualBertLayer.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   #  s    zVisualBertModel.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )VisualBertPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   *  s    z"VisualBertPreTrainedModel.__init__Nr   r   r   r   r   r  '  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    z"ViTForImageClassification.__init__Nr   r   r   r   r   r  1  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTForMaskedImageModelingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ;  s    z"ViTForMaskedImageModeling.__init__Nr   r   r   r   r   r  8  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   B  s    zViTModel.__init__Nr   r   r   r   r   r  ?  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   I  s    zViTPreTrainedModel.__init__Nr   r   r   r   r   r  F  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTHybridForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z(ViTHybridForImageClassification.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTHybridModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    zViTHybridModel.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTHybridPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    z!ViTHybridPreTrainedModel.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMAEForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r   k  s    zViTMAEForPreTraining.__init__Nr   r   r   r   r   r  h  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMAELayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   r  s    zViTMAELayer.__init__Nr   r   r   r   r   r  o  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMAEModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    zViTMAEModel.__init__Nr   r   r   r   r   r  v  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMAEPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViTMAEPreTrainedModel.__init__Nr   r   r   r   r   r  }  s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMSNForImageClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%ViTMSNForImageClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMSNModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViTMSNModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )ViTMSNPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zViTMSNPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )#Wav2Vec2ForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,Wav2Vec2ForAudioFrameClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWav2Vec2ForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWav2Vec2ForMaskedLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWav2Vec2ForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )!Wav2Vec2ForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z*Wav2Vec2ForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ForXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWav2Vec2ForXVector.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2Modelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWav2Vec2Model.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2PreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Wav2Vec2PreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ),Wav2Vec2ConformerForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z5Wav2Vec2ConformerForAudioFrameClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ConformerForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z Wav2Vec2ConformerForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ConformerForPreTrainingr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(Wav2Vec2ConformerForPreTraining.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )*Wav2Vec2ConformerForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z3Wav2Vec2ConformerForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ConformerForXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$Wav2Vec2ConformerForXVector.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )Wav2Vec2ConformerModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    zWav2Vec2ConformerModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) Wav2Vec2ConformerPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)Wav2Vec2ConformerPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) WavLMForAudioFrameClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)WavLMForAudioFrameClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )WavLMForCTCr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zWavLMForCTC.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )WavLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'WavLMForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )WavLMForXVectorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zWavLMForXVector.__init__Nr   r   r   r   r   r  #  s   r  c                   @   s   e Zd ZdgZdd ZdS )
WavLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    zWavLMModel.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )WavLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   4  s    zWavLMPreTrainedModel.__init__Nr   r   r   r   r   r  1  s   r  c                   @   s   e Zd ZdgZdd ZdS )WhisperForAudioClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    z&WhisperForAudioClassification.__init__Nr   r   r   r   r   r  ;  s   r  c                   @   s   e Zd ZdgZdd ZdS )WhisperForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    z(WhisperForConditionalGeneration.__init__Nr   r   r   r   r   r  B  s   r  c                   @   s   e Zd ZdgZdd ZdS )WhisperModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    zWhisperModel.__init__Nr   r   r   r   r   r  I  s   r  c                   @   s   e Zd ZdgZdd ZdS )WhisperPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    zWhisperPreTrainedModel.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )
XCLIPModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   ]  s    zXCLIPModel.__init__Nr   r   r   r   r   r  Z  s   r  c                   @   s   e Zd ZdgZdd ZdS )XCLIPPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   d  s    zXCLIPPreTrainedModel.__init__Nr   r   r   r   r   r  a  s   r  c                   @   s   e Zd ZdgZdd ZdS )XCLIPTextModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   k  s    zXCLIPTextModel.__init__Nr   r   r   r   r   r  h  s   r  c                   @   s   e Zd ZdgZdd ZdS )XCLIPVisionModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   r  s    zXCLIPVisionModel.__init__Nr   r   r   r   r   r  o  s   r  c                   @   s   e Zd ZdgZdd ZdS )XGLMForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   |  s    zXGLMForCausalLM.__init__Nr   r   r   r   r   r  y  s   r  c                   @   s   e Zd ZdgZdd ZdS )	XGLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXGLMModel.__init__Nr   r   r   r   r   r     s   r   c                   @   s   e Zd ZdgZdd ZdS )XGLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXGLMPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z XLMForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&XLMForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%XLMForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z"XLMForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMWithLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMWithLMHeadModel.__init__Nr   r   r   r   r   r	    s   r	  c                   @   s   e Zd ZdgZdd ZdS )XLMProphetNetDecoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMProphetNetDecoder.__init__Nr   r   r   r   r   r
    s   r
  c                   @   s   e Zd ZdgZdd ZdS )XLMProphetNetEncoderr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMProphetNetEncoder.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMProphetNetForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!XLMProphetNetForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )%XLMProphetNetForConditionalGenerationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z.XLMProphetNetForConditionalGeneration.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMProphetNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMProphetNetModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMProphetNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z%XLMProphetNetPreTrainedModel.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMRobertaForCausalLM.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLMRobertaForMaskedLM.__init__Nr   r   r   r   r   r     s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   
  s    z$XLMRobertaForMultipleChoice.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'XLMRobertaForQuestionAnswering.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )#XLMRobertaForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z,XLMRobertaForSequenceClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS ) XLMRobertaForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z)XLMRobertaForTokenClassification.__init__Nr   r   r   r   r   r    s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   &  s    zXLMRobertaModel.__init__Nr   r   r   r   r   r  #  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   -  s    z"XLMRobertaPreTrainedModel.__init__Nr   r   r   r   r   r  *  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaXLForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   7  s    z XLMRobertaXLForCausalLM.__init__Nr   r   r   r   r   r  4  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaXLForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r   >  s    z XLMRobertaXLForMaskedLM.__init__Nr   r   r   r   r   r  ;  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaXLForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   E  s    z&XLMRobertaXLForMultipleChoice.__init__Nr   r   r   r   r   r  B  s   r  c                   @   s   e Zd ZdgZdd ZdS ) XLMRobertaXLForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   L  s    z)XLMRobertaXLForQuestionAnswering.__init__Nr   r   r   r   r   r  I  s   r  c                   @   s   e Zd ZdgZdd ZdS )%XLMRobertaXLForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   S  s    z.XLMRobertaXLForSequenceClassification.__init__Nr   r   r   r   r   r  P  s   r  c                   @   s   e Zd ZdgZdd ZdS )"XLMRobertaXLForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r   Z  s    z+XLMRobertaXLForTokenClassification.__init__Nr   r   r   r   r   r  W  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaXLModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   a  s    zXLMRobertaXLModel.__init__Nr   r   r   r   r   r  ^  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLMRobertaXLPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   h  s    z$XLMRobertaXLPreTrainedModel.__init__Nr   r   r   r   r   r  e  s   r  c                   @   s   e Zd ZdgZdd ZdS )XLNetForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r   r  s    zXLNetForMultipleChoice.__init__Nr   r   r   r   r   r   o  s   r   c                   @   s   e Zd ZdgZdd ZdS )XLNetForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    z"XLNetForQuestionAnswering.__init__Nr   r   r   r   r   r!  v  s   r!  c                   @   s   e Zd ZdgZdd ZdS )XLNetForQuestionAnsweringSimpler   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z(XLNetForQuestionAnsweringSimple.__init__Nr   r   r   r   r   r"  }  s   r"  c                   @   s   e Zd ZdgZdd ZdS )XLNetForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z'XLNetForSequenceClassification.__init__Nr   r   r   r   r   r#    s   r#  c                   @   s   e Zd ZdgZdd ZdS )XLNetForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z$XLNetForTokenClassification.__init__Nr   r   r   r   r   r$    s   r$  c                   @   s   e Zd ZdgZdd ZdS )XLNetLMHeadModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLNetLMHeadModel.__init__Nr   r   r   r   r   r%    s   r%  c                   @   s   e Zd ZdgZdd ZdS )
XLNetModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLNetModel.__init__Nr   r   r   r   r   r&    s   r&  c                   @   s   e Zd ZdgZdd ZdS )XLNetPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXLNetPreTrainedModel.__init__Nr   r   r   r   r   r'    s   r'  c                  O   s   t tdg d S r   )r   load_tf_weights_in_xlnetr<   r   r   r   r(    s    r(  c                   @   s   e Zd ZdgZdd ZdS )XmodForCausalLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXmodForCausalLM.__init__Nr   r   r   r   r   r)    s   r)  c                   @   s   e Zd ZdgZdd ZdS )XmodForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXmodForMaskedLM.__init__Nr   r   r   r   r   r*    s   r*  c                   @   s   e Zd ZdgZdd ZdS )XmodForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXmodForMultipleChoice.__init__Nr   r   r   r   r   r+    s   r+  c                   @   s   e Zd ZdgZdd ZdS )XmodForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!XmodForQuestionAnswering.__init__Nr   r   r   r   r   r,    s   r,  c                   @   s   e Zd ZdgZdd ZdS )XmodForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&XmodForSequenceClassification.__init__Nr   r   r   r   r   r-    s   r-  c                   @   s   e Zd ZdgZdd ZdS )XmodForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z#XmodForTokenClassification.__init__Nr   r   r   r   r   r.    s   r.  c                   @   s   e Zd ZdgZdd ZdS )	XmodModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXmodModel.__init__Nr   r   r   r   r   r/    s   r/  c                   @   s   e Zd ZdgZdd ZdS )XmodPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zXmodPreTrainedModel.__init__Nr   r   r   r   r   r0    s   r0  c                   @   s   e Zd ZdgZdd ZdS )YolosForObjectDetectionr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z YolosForObjectDetection.__init__Nr   r   r   r   r   r1    s   r1  c                   @   s   e Zd ZdgZdd ZdS )
YolosModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zYolosModel.__init__Nr   r   r   r   r   r2    s   r2  c                   @   s   e Zd ZdgZdd ZdS )YolosPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zYolosPreTrainedModel.__init__Nr   r   r   r   r   r3    s   r3  c                   @   s   e Zd ZdgZdd ZdS )YosoForMaskedLMr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zYosoForMaskedLM.__init__Nr   r   r   r   r   r4    s   r4  c                   @   s   e Zd ZdgZdd ZdS )YosoForMultipleChoicer   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zYosoForMultipleChoice.__init__Nr   r   r   r   r   r5    s   r5  c                   @   s   e Zd ZdgZdd ZdS )YosoForQuestionAnsweringr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z!YosoForQuestionAnswering.__init__Nr   r   r   r   r   r6    s   r6  c                   @   s   e Zd ZdgZdd ZdS )YosoForSequenceClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    z&YosoForSequenceClassification.__init__Nr   r   r   r   r   r7    s   r7  c                   @   s   e Zd ZdgZdd ZdS )YosoForTokenClassificationr   c                 O   s   t | dg d S r   r   r   r   r   r   r      s    z#YosoForTokenClassification.__init__Nr   r   r   r   r   r8    s   r8  c                   @   s   e Zd ZdgZdd ZdS )	YosoLayerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   '  s    zYosoLayer.__init__Nr   r   r   r   r   r9  $  s   r9  c                   @   s   e Zd ZdgZdd ZdS )	YosoModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   .  s    zYosoModel.__init__Nr   r   r   r   r   r:  +  s   r:  c                   @   s   e Zd ZdgZdd ZdS )YosoPreTrainedModelr   c                 O   s   t | dg d S r   r   r   r   r   r   r   5  s    zYosoPreTrainedModel.__init__Nr   r   r   r   r   r;  2  s   r;  c                   @   s   e Zd ZdgZdd ZdS )	Adafactorr   c                 O   s   t | dg d S r   r   r   r   r   r   r   <  s    zAdafactor.__init__Nr   r   r   r   r   r<  9  s   r<  c                   @   s   e Zd ZdgZdd ZdS )AdamWr   c                 O   s   t | dg d S r   r   r   r   r   r   r   C  s    zAdamW.__init__Nr   r   r   r   r   r=  @  s   r=  c                  O   s   t tdg d S r   )r   get_constant_scheduler<   r   r   r   r>  G  s    r>  c                  O   s   t tdg d S r   )r   !get_constant_schedule_with_warmupr<   r   r   r   r?  K  s    r?  c                  O   s   t tdg d S r   )r   get_cosine_schedule_with_warmupr<   r   r   r   r@  O  s    r@  c                  O   s   t tdg d S r   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr<   r   r   r   rA  S  s    rA  c                  O   s   t tdg d S r   )r   get_inverse_sqrt_scheduler<   r   r   r   rB  W  s    rB  c                  O   s   t tdg d S r   )r   get_linear_schedule_with_warmupr<   r   r   r   rC  [  s    rC  c                  O   s   t tdg d S r   )r   )get_polynomial_decay_schedule_with_warmupr<   r   r   r   rD  _  s    rD  c                  O   s   t tdg d S r   )r   get_schedulerr<   r   r   r   rE  c  s    rE  c                   @   s   e Zd ZdgZdd ZdS )Conv1Dr   c                 O   s   t | dg d S r   r   r   r   r   r   r   j  s    zConv1D.__init__Nr   r   r   r   r   rF  g  s   rF  c                  O   s   t tdg d S r   )r   apply_chunking_to_forwardr<   r   r   r   rG  n  s    rG  c                  O   s   t tdg d S r   )r   prune_layerr<   r   r   r   rH  r  s    rH  c                   @   s   e Zd ZdgZdd ZdS )Trainerr   c                 O   s   t | dg d S r   r   r   r   r   r   r   y  s    zTrainer.__init__Nr   r   r   r   r   rI  v  s   rI  c                  O   s   t tdg d S r   )r   torch_distributed_zero_firstr<   r   r   r   rJ  }  s    rJ  c                   @   s   e Zd ZdgZdd ZdS )Seq2SeqTrainerr   c                 O   s   t | dg d S r   r   r   r   r   r   r     s    zSeq2SeqTrainer.__init__Nr   r   r   r   r   rK    s   rK  (  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+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r=   Z$ALBERT_PRETRAINED_MODEL_ARCHIVE_LISTr>   r?   r@   rA   rB   rC   rD   rE   rF   Z#ALIGN_PRETRAINED_MODEL_ARCHIVE_LISTrG   rH   rI   rJ   Z%ALTCLIP_PRETRAINED_MODEL_ARCHIVE_LISTrK   rL   rM   rN   Z;AUDIO_SPECTROGRAM_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrO   rP   rQ   Z&MODEL_FOR_AUDIO_CLASSIFICATION_MAPPINGZMODEL_FOR_AUDIO_XVECTOR_MAPPINGZMODEL_FOR_BACKBONE_MAPPINGZ'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGZMODEL_FOR_CAUSAL_LM_MAPPINGZMODEL_FOR_CTC_MAPPINGZ"MODEL_FOR_DEPTH_ESTIMATION_MAPPINGZ-MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPINGZ&MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGZ$MODEL_FOR_IMAGE_SEGMENTATION_MAPPINGZ'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPINGZ!MODEL_FOR_MASK_GENERATION_MAPPINGZ'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGZMODEL_FOR_MASKED_LM_MAPPINGZ!MODEL_FOR_MULTIPLE_CHOICE_MAPPINGZ*MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPINGZ"MODEL_FOR_OBJECT_DETECTION_MAPPINGZMODEL_FOR_PRETRAINING_MAPPINGZ$MODEL_FOR_QUESTION_ANSWERING_MAPPINGZ'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPINGZ&MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPINGZ)MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPINGZ"MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPINGZ*MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGZ&MODEL_FOR_TOKEN_CLASSIFICATION_MAPPINGZ(MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPINGZ&MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGZMODEL_FOR_VISION_2_SEQ_MAPPINGZ+MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPINGZ0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPINGZ,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGZMODEL_MAPPINGZMODEL_WITH_LM_HEAD_MAPPINGrR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   Z(AUTOFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrs   rt   ru   Z"BART_PRETRAINED_MODEL_ARCHIVE_LISTrv   rw   rx   ry   rz   r{   r|   Z"BEIT_PRETRAINED_MODEL_ARCHIVE_LISTr}   r~   r   r   r   Z"BERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Z&BIG_BIRD_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   r   r   Z-BIGBIRD_PEGASUS_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   Z$BIOGPT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z!BIT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z(BLENDERBOT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z.BLENDERBOT_SMALL_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z"BLIP_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z$BLIP_2_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z#BLOOM_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   Z)BRIDGETOWER_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z'CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   Z$CANINE_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   Z*CHINESE_CLIP_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z"CLAP_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   Z"CLIP_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   Z%CLIPSEG_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   Z%CODEGEN_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   Z.CONDITIONAL_DETR_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   Z&CONVBERT_PRETRAINED_MODEL_ARCHIVE_LISTr   r   r   r   r   r   r   r   r   Z&CONVNEXT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z(CONVNEXTV2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z$CPMANT_PRETRAINED_MODEL_ARCHIVE_LISTr	  r
  r  Z"CTRL_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z!CVT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z,DATA2VEC_AUDIO_PRETRAINED_MODEL_ARCHIVE_LISTZ+DATA2VEC_TEXT_PRETRAINED_MODEL_ARCHIVE_LISTZ-DATA2VEC_VISION_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  r  r  r  r   r!  r"  r#  r$  Z%DEBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr%  r&  r'  r(  r)  r*  Z(DEBERTA_V2_PRETRAINED_MODEL_ARCHIVE_LISTr+  r,  r-  r.  r/  r0  r1  Z2DECISION_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr2  r3  r4  r5  Z-DEFORMABLE_DETR_PRETRAINED_MODEL_ARCHIVE_LISTr6  r7  r8  Z"DEIT_PRETRAINED_MODEL_ARCHIVE_LISTr9  r:  r;  r<  r=  Z"DETA_PRETRAINED_MODEL_ARCHIVE_LISTr>  r?  r@  Z"DETR_PRETRAINED_MODEL_ARCHIVE_LISTrA  rB  rC  rD  Z#DINAT_PRETRAINED_MODEL_ARCHIVE_LISTrE  rF  rG  rH  Z(DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LISTrI  rJ  rK  rL  rM  rN  rO  Z(DONUT_SWIN_PRETRAINED_MODEL_ARCHIVE_LISTrP  rQ  Z1DPR_CONTEXT_ENCODER_PRETRAINED_MODEL_ARCHIVE_LISTZ2DPR_QUESTION_ENCODER_PRETRAINED_MODEL_ARCHIVE_LISTZ(DPR_READER_PRETRAINED_MODEL_ARCHIVE_LISTrR  rS  rT  rU  rV  rW  rX  Z!DPT_PRETRAINED_MODEL_ARCHIVE_LISTrY  rZ  r[  r\  Z-EFFICIENTFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr]  r^  r_  r`  Z*EFFICIENTNET_PRETRAINED_MODEL_ARCHIVE_LISTra  rb  rc  Z%ELECTRA_PRETRAINED_MODEL_ARCHIVE_LISTrd  re  rf  rg  rh  ri  rj  rk  rl  rm  rn  Z#ERNIE_PRETRAINED_MODEL_ARCHIVE_LISTro  rp  rq  rr  rs  rt  ru  rv  rw  rx  Z%ERNIE_M_PRETRAINED_MODEL_ARCHIVE_LISTry  rz  r{  r|  r}  r~  r  Z!ESM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z&FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z#FLAVA_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z"FNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  Z&FOCALNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z$FUNNEL_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  Z!GIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z"GLPN_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z"GPT2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z)GPT_BIGCODE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z%GPT_NEO_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z&GPT_NEOX_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z/GPT_NEOX_JAPANESE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z"GPTJ_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z-GPTSAN_JAPANESE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z(GRAPHORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z&GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z$HUBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z#IBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z&IMAGEGPT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z&INFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z%JUKEBOX_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z&LAYOUTLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r   r  r  r  Z(LAYOUTLMV2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z(LAYOUTLMV3_PRETRAINED_MODEL_ARCHIVE_LISTr	  r
  r  r  r  Z!LED_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z#LEVIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z"LILT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  Z(LONGFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr   r!  r"  r#  r$  r%  r&  r'  Z$LONGT5_PRETRAINED_MODEL_ARCHIVE_LISTr(  r)  r*  r+  Z"LUKE_PRETRAINED_MODEL_ARCHIVE_LISTr,  r-  r.  r/  r0  r1  r2  r3  r4  r5  r6  r7  r8  r9  r:  r;  r<  Z%M2M_100_PRETRAINED_MODEL_ARCHIVE_LISTr=  r>  r?  r@  rA  rB  Z&MARKUPLM_PRETRAINED_MODEL_ARCHIVE_LISTrC  rD  rE  rF  rG  Z)MASK2FORMER_PRETRAINED_MODEL_ARCHIVE_LISTrH  rI  rJ  Z(MASKFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrK  rL  rM  rN  rO  rP  rQ  rR  rS  rT  Z#MCTCT_PRETRAINED_MODEL_ARCHIVE_LISTrU  rV  rW  Z"MEGA_PRETRAINED_MODEL_ARCHIVE_LISTrX  rY  rZ  r[  r\  r]  r^  r_  Z+MEGATRON_BERT_PRETRAINED_MODEL_ARCHIVE_LISTr`  ra  rb  rc  rd  re  rf  rg  rh  ri  Z%MGP_STR_PRETRAINED_MODEL_ARCHIVE_LISTrj  rk  rl  rm  rn  ro  Z(MOBILEBERT_PRETRAINED_MODEL_ARCHIVE_LISTrp  rq  rr  rs  rt  ru  rv  rw  rx  ry  rz  Z*MOBILENET_V1_PRETRAINED_MODEL_ARCHIVE_LISTr{  r|  r}  r~  Z*MOBILENET_V2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z'MOBILEVIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z)MOBILEVITV2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z#MPNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  r  r  Z!MVP_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z!NAT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z#NEZHA_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  Z&NLLB_MOE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z+NYSTROMFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z'ONEFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z(OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z!OPT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z$OWLVIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  Z'PEGASUS_X_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z'PERCEIVER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  Z(PIX2STRUCT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z$PLBART_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z(POOLFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z(PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z%QDQBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  r  r  r  r  r   Z#REALM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z&REFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr	  r
  r  r  r  r  r  r  Z$REGNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z%REMBERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  Z$RESNET_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r   r!  Z'RETRIBERT_PRETRAINED_MODEL_ARCHIVE_LISTr"  r#  Z%ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr$  r%  r&  r'  r(  r)  r*  r+  Z2ROBERTA_PRELAYERNORM_PRETRAINED_MODEL_ARCHIVE_LISTr,  r-  r.  r/  r0  r1  r2  r3  Z&ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LISTr4  r5  r6  r7  r8  r9  r:  r;  r<  r=  r>  Z&ROFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr?  r@  rA  rB  rC  rD  rE  rF  rG  rH  Z"RWKV_PRETRAINED_MODEL_ARCHIVE_LISTrI  rJ  rK  Z!SAM_PRETRAINED_MODEL_ARCHIVE_LISTrL  rM  Z'SEGFORMER_PRETRAINED_MODEL_ARCHIVE_LISTrN  rO  rP  rQ  rR  rS  Z!SEW_PRETRAINED_MODEL_ARCHIVE_LISTrT  rU  rV  rW  Z#SEW_D_PRETRAINED_MODEL_ARCHIVE_LISTrX  rY  rZ  r[  r\  Z,SPEECH_TO_TEXT_PRETRAINED_MODEL_ARCHIVE_LISTr]  r^  r_  r`  ra  Z&SPEECHT5_PRETRAINED_MODEL_ARCHIVE_LISTrb  rc  rd  re  rf  rg  Z&SPLINTER_PRETRAINED_MODEL_ARCHIVE_LISTrh  ri  rj  rk  rl  Z)SQUEEZEBERT_PRETRAINED_MODEL_ARCHIVE_LISTrm  rn  ro  rp  rq  rr  rs  rt  Z)SWIFTFORMER_PRETRAINED_MODEL_ARCHIVE_LISTru  rv  rw  Z"SWIN_PRETRAINED_MODEL_ARCHIVE_LISTrx  ry  rz  r{  r|  Z%SWIN2SR_PRETRAINED_MODEL_ARCHIVE_LISTr}  r~  r  Z$SWINV2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z1SWITCH_TRANSFORMERS_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z T5_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z/TABLE_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z#TAPAS_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z5TIME_SERIES_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z)TIMESFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z4TRAJECTORY_TRANSFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  Z(TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z#TROCR_PRETRAINED_MODEL_ARCHIVE_LISTr  r  Z"TVLT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z'UNISPEECH_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  Z+UNISPEECH_SAT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  Z!VAN_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z&VIDEOMAE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z"VILT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  r  r  Z)VISUAL_BERT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z!VIT_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z(VIT_HYBRID_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z%VIT_MAE_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z%VIT_MSN_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  Z)WAV_2_VEC_2_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z0WAV2VEC2_CONFORMER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  Z#WAVLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  Z%WHISPER_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z#XCLIP_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  Z"XGLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r   r  Z!XLM_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r	  Z,XLM_PROPHETNET_PRETRAINED_MODEL_ARCHIVE_LISTr
  r  r  r  r  r  Z)XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z,XLM_ROBERTA_XL_PRETRAINED_MODEL_ARCHIVE_LISTr  r  r  r  r  r  r  r  Z#XLNET_PRETRAINED_MODEL_ARCHIVE_LISTr   r!  r"  r#  r$  r%  r&  r'  r(  Z"XMOD_PRETRAINED_MODEL_ARCHIVE_LISTr)  r*  r+  r,  r-  r.  r/  r0  Z#YOLOS_PRETRAINED_MODEL_ARCHIVE_LISTr1  r2  r3  Z"YOSO_PRETRAINED_MODEL_ARCHIVE_LISTr4  r5  r6  r7  r8  r9  r:  r;  r<  r=  r>  r?  r@  rA  rB  rC  rD  rE  rF  rG  rH  rI  rJ  rK  r   r   r   r   <module>   s

  