
    sg                       d dl mZmZ  G d de      Z G d de      Z G d de      Z G d	 d
e      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d! d"e      Z G d# d$e      Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Z G d1 d2e      Z G d3 d4e      Z G d5 d6e      Z G d7 d8e      Z G d9 d:e      Z G d; d<e      Z G d= d>e      Z  G d? d@e      Z! G dA dBe      Z" G dC dDe      Z# G dE dFe      Z$ G dG dHe      Z% G dI dJe      Z& G dK dLe      Z' G dM dNe      Z( G dO dPe      Z) G dQ dRe      Z* G dS dTe      Z+ G dU dVe      Z, G dW dXe      Z- G dY dZe      Z. G d[ d\e      Z/ G d] d^e      Z0 G d_ d`e      Z1 G da dbe      Z2 G dc dde      Z3 G de dfe      Z4 G dg dhe      Z5 G di dje      Z6 G dk dle      Z7 G dm dne      Z8 G do dpe      Z9 G dq dre      Z: G ds dte      Z; G du dve      Z< G dw dxe      Z= G dy dze      Z> G d{ d|e      Z? G d} d~e      Z@ G d de      ZA G d de      ZB G d de      ZC G d de      ZD G d de      ZE G d de      ZF G d de      ZG G d de      ZH G d de      ZI G d de      ZJ G d de      ZK G d de      ZL G d de      ZM G d de      ZN G d de      ZO G d de      ZP G d de      ZQd ZRdZS G d de      ZT G d de      ZU G d de      ZV G d de      ZW G d de      ZX G d de      ZY G d de      ZZ G d de      Z[ G d de      Z\d Z] G d de      Z^ G d de      Z_ G d de      Z` G d de      Za G d de      Zb G d de      Zc G d de      Zd G dĄ de      Ze G dƄ de      Zf G dȄ de      Zg G dʄ de      ZhdZidZjdZkdZldZmdZndZodZpdZqdZrdZsdZtdZudZvdZwdZxdZydZzdZ{dZ|dZ}dZ~dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ G d̄ de      Z G d΄ de      Z G dЄ de      Z G d҄ de      Z G dԄ de      Z G dք de      Z G d؄ de      Z G dڄ de      Z G d܄ de      Z G dބ de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  de      Z G d de      Z G d de      Z G d de      Z G d d	e      Z G d
 de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  d!e      Z G d" d#e      Z G d$ d%e      Z G d& d'e      Z G d( d)e      Z G d* d+e      Z G d, d-e      Z G d. d/e      Z G d0 d1e      Z G d2 d3e      Z G d4 d5e      Z G d6 d7e      Z G d8 d9e      Z G d: d;e      Z G d< d=e      Z G d> d?e      Z G d@ dAe      Z G dB dCe      Z G dD dEe      Z G dF dGe      Z G dH dIe      Z G dJ dKe      Z G dL dMe      Z G dN dOe      Z G dP dQe      Z G dR dSe      Z G dT dUe      Z G dV dWe      Z G dX dYe      Z G dZ d[e      Zېd\ Z G d] d^e      Z G d_ d`e      Z G da dbe      Zߐdc Z G dd dee      Z G df dge      Z G dh die      Z G dj dke      Z G dl dme      Z G dn doe      Z G dp dqe      Z G dr dse      Z G dt due      Zdv Z G dw dxe      Z G dy dze      Z G d{ d|e      Z G d} d~e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z  G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dÄ de      Z G dń de      Z G dǄ de      Z G dɄ de      Z G d˄ de      Z G d̈́ de      Z G dτ de      Z G dф de      Z G dӄ de      Z G dՄ de      Z G dׄ de      Z G dل de      Z G dۄ de      Z G d݄ de      Z G d߄ de      Z G d de      Z  G d de      Z! G d de      Z" G d de      Z# G d de      Z$ G d de      Z% G d de      Z& G d de      Z' G d de      Z( G d de      Z) G d de      Z* G d de      Z+ G d de      Z, G d de      Z- G d de      Z. G d d e      Z/d Z0 G d de      Z1 G d de      Z2 G d de      Z3 G d d	e      Z4 G d
 de      Z5 G d de      Z6 G d de      Z7 G d de      Z8 G d de      Z9 G d de      Z: G d de      Z; G d de      Z< G d de      Z= G d de      Z> G d de      Z? G d  d!e      Z@ G d" d#e      ZA G d$ d%e      ZB G d& d'e      ZC G d( d)e      ZD G d* d+e      ZE G d, d-e      ZF G d. d/e      ZG G d0 d1e      ZH G d2 d3e      ZI G d4 d5e      ZJ G d6 d7e      ZK G d8 d9e      ZL G d: d;e      ZM G d< d=e      ZN G d> d?e      ZO G d@ dAe      ZP G dB dCe      ZQ G dD dEe      ZR G dF dGe      ZS G dH dIe      ZT G dJ dKe      ZU G dL dMe      ZV G dN dOe      ZW G dP dQe      ZX G dR dSe      ZY G dT dUe      ZZ G dV dWe      Z[ G dX dYe      Z\ G dZ d[e      Z] G d\ d]e      Z^ G d^ d_e      Z_ G d` dae      Z` G db dce      Za G dd dee      Zb G df dge      Zcdh Zd G di dje      Ze G dk dle      Zf G dm dne      Zg G do dpe      Zh G dq dre      Zi G ds dte      Zj G du dve      Zk G dw dxe      Zl G dy dze      Zm G d{ d|e      Zn G d} d~e      Zo G d de      Zp G d de      Zq G d de      Zr G d de      Zs G d de      Zt G d de      Zu G d de      Zv G d de      Zw G d de      Zx G d de      Zy G d de      Zz G d de      Z{ G d de      Z| G d de      Z} G d de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dÄ de      Z G dń de      Z G dǄ de      Z G dɄ de      Z G d˄ de      Z G d̈́ de      Z G dτ de      Z G dф de      Z G dӄ de      Z G dՄ de      Z G dׄ de      Z G dل de      Z G dۄ de      Z G d݄ de      Z G d߄ de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d	 d
e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d! d"e      Z G d# d$e      Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Z G d1 d2e      Z G d3 d4e      Z G d5 d6e      Z G d7 d8e      Z G d9 d:e      Z G d; d<e      Z G d= d>e      Z G d? d@e      Z G dA dBe      Z G dC dDe      Z G dE dFe      Z G dG dHe      Z G dI dJe      Z G dK dLe      Z G dM dNe      Z G dO dPe      Z G dQ dRe      Z G dS dTe      Z G dU dVe      Z G dW dXe      Z G dY dZe      Z G d[ d\e      Z G d] d^e      Z G d_ d`e      Z G da dbe      Z G dc dde      Z G de dfe      Z G dg dhe      Z G di dje      Z G dk dle      Zdm Z G dn doe      Z G dp dqe      Z G dr dse      Z G dt due      Z G dv dwe      Z G dx dye      Z G dz d{e      Zd| Z G d} d~e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z  G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dĄ de      Z G dƄ de      Z G dȄ de      Z G dʄ de      Z G d̄ de      Z G d΄ de      Z G dЄ de      Z G d҄ de      Z G dԄ de      Z G dք de      Z G d؄ de      Z G dڄ de      Z G d܄ de      Z  G dބ de      Z! G d de      Z" G d de      Z# G d de      Z$ G d de      Z% G d de      Z& G d de      Z' G d de      Z( G d de      Z) G d de      Z* G d de      Z+ G d de      Z, G d de      Z- G d de      Z. G d de      Z/ G d de      Z0 G d de      Z1 G d  de      Z2 G d de      Z3 G d de      Z4 G d de      Z5 G d d	e      Z6 G d
 de      Z7 G d de      Z8 G d de      Z9d Z: G d de      Z; G d de      Z< G d de      Z= G d de      Z> G d de      Z? G d de      Z@ G d de      ZA G d d e      ZB G d! d"e      ZC G d# d$e      ZD G d% d&e      ZE G d' d(e      ZF G d) d*e      ZG G d+ d,e      ZH G d- d.e      ZI G d/ d0e      ZJ G d1 d2e      ZK G d3 d4e      ZL G d5 d6e      ZM G d7 d8e      ZN G d9 d:e      ZO G d; d<e      ZP G d= d>e      ZQ G d? d@e      ZR G dA dBe      ZS G dC dDe      ZT G dE dFe      ZU G dG dHe      ZV G dI dJe      ZW G dK dLe      ZX G dM dNe      ZY G dO dPe      ZZ G dQ dRe      Z[ G dS dTe      Z\ G dU dVe      Z] G dW dXe      Z^ G dY dZe      Z_ G d[ d\e      Z` G d] d^e      Za G d_ d`e      Zb G da dbe      Zc G dc dde      Zd G de dfe      Ze G dg dhe      Zf G di dje      Zg G dk dle      Zh G dm dne      Zi G do dpe      Zj G dq dre      Zk G ds dte      Zl G du dve      Zm G dw dxe      Zn G dy dze      Zo G d{ d|e      Zp G d} d~e      Zq G d de      Zr G d de      Zs G d de      Zt G d de      Zu G d de      Zv G d de      Zw G d de      Zx G d de      Zy G d de      Zz G d de      Z{ G d de      Z| G d de      Z} G d de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dĄ de      Z G dƄ de      Z G dȄ de      Z G dʄ de      Z G d̄ de      Z G d΄ de      Z G dЄ de      Z G d҄ de      Z G dԄ de      Z G dք de      Z G d؄ de      Z G dڄ de      Z G d܄ de      Z G dބ de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d  de      Z G d de      Z G d de      Z G d de      Z G d d	e      Z G d
 de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  d!e      Z G d" d#e      Z G d$ d%e      Z G d& d'e      Z G d( d)e      Z G d* d+e      Z G d, d-e      Z G d. d/e      Z G d0 d1e      Z G d2 d3e      Z G d4 d5e      Z G d6 d7e      Z G d8 d9e      Z G d: d;e      Z G d< d=e      Z G d> d?e      Z G d@ dAe      Z G dB dCe      Z G dD dEe      Z G dF dGe      Z G dH dIe      Z G dJ dKe      Z G dL dMe      Z G dN dOe      Z G dP dQe      Z G dR dSe      Z G dT dUe      Z G dV dWe      Z G dX dYe      Z G dZ d[e      Z G d\ d]e      Z G d^ d_e      Z G d` dae      Z G db dce      Z G dd dee      Z G df dge      Z G dh die      Z G dj dke      Z G dl dme      Z G dn doe      Z G dp dqe      Z G dr dse      Z G dt due      Z G dv dwe      Zdx Z G dy dze      Z G d{ d|e      Z G d} d~e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z  G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dÄ de      Z G dń de      Z G dǄ de      Z G dɄ de      Z G d˄ de      Z G d̈́ de      Z G dτ de      Z G dф de      Z G dӄ de      Z G dՄ de      Z G dׄ de      Z  G dل de      Z! G dۄ de      Z" G d݄ de      Z# G d߄ de      Z$ G d de      Z% G d de      Z& G d de      Z' G d de      Z( G d de      Z) G d de      Z* G d de      Z+ G d de      Z, G d de      Z- G d de      Z. G d de      Z/ G d de      Z0 G d de      Z1 G d de      Z2 G d de      Z3 G d d e      Z4 G d de      Z5 G d de      Z6 G d de      Z7 G d de      Z8 G d	 d
e      Z9 G d de      Z: G d de      Z; G d de      Z< G d de      Z= G d de      Z> G d de      Z? G d de      Z@ G d de      ZA G d de      ZB G d de      ZC G d d e      ZD G d! d"e      ZE G d# d$e      ZF G d% d&e      ZG G d' d(e      ZH G d) d*e      ZI G d+ d,e      ZJ G d- d.e      ZK G d/ d0e      ZL G d1 d2e      ZM G d3 d4e      ZN G d5 d6e      ZO G d7 d8e      ZP G d9 d:e      ZQ G d; d<e      ZR G d= d>e      ZS G d? d@e      ZT G dA dBe      ZU G dC dDe      ZV G dE dFe      ZW G dG dHe      ZX G dI dJe      ZY G dK dLe      ZZ G dM dNe      Z[ G dO dPe      Z\ G dQ dRe      Z] G dS dTe      Z^ G dU dVe      Z_ G dW dXe      Z` G dY dZe      Za G d[ d\e      Zb G d] d^e      Zc G d_ d`e      Zd G da dbe      Ze G dc dde      Zf G de dfe      Zg G dg dhe      Zh G di dje      Zi G dk dle      Zj G dm dne      Zk G do dpe      Zl G dq dre      Zm G ds dte      Zn G du dve      Zo G dw dxe      Zp G dy dze      Zq G d{ d|e      Zr G d} d~e      Zs G d de      Zt G d de      Zu G d de      Zv G d de      Zw G d de      Zx G d de      Zy G d de      Zz G d de      Z{ G d de      Z| G d de      Z} G d de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dÄ de      Zdń Z G dƄ de      Z G dȄ de      Z G dʄ de      Zd̄ Z G d̈́ de      Z G dτ de      Z G dф de      Z G dӄ de      ZdՄ Z G dք de      Z G d؄ de      Z G dڄ de      Z G d܄ de      Z G dބ de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  de      Z G d de      Z G d de      Z G d de      Z G d d	e      Z G d
 de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d  d!e      Z G d" d#e      Z G d$ d%e      Z G d& d'e      Z G d( d)e      Z G d* d+e      Z G d, d-e      Z G d. d/e      Z G d0 d1e      Z G d2 d3e      Z G d4 d5e      Z G d6 d7e      Z G d8 d9e      Z G d: d;e      Z G d< d=e      Z G d> d?e      Z G d@ dAe      Z G dB dCe      Z G dD dEe      Z G dF dGe      Z G dH dIe      Z G dJ dKe      Z G dL dMe      Z G dN dOe      Z G dP dQe      Z G dR dSe      Z G dT dUe      Z G dV dWe      Z G dX dYe      Z G dZ d[e      Z G d\ d]e      Z G d^ d_e      Z G d` dae      Z G db dce      Z G dd dee      Z G df dge      Z G dh die      Z G dj dke      Z G dl dme      Z G dn doe      Z G dp dqe      Z G dr dse      Z G dt due      Z G dv dwe      Z G dx dye      Z G dz d{e      Z G d| d}e      Z G d~ de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z  G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z	 G d de      Z
 G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dÄ de      Z G dń de      Z G dǄ de      Z G dɄ de      Z G d˄ de      Z G d̈́ de      Z G dτ de      Z G dф de      Z G dӄ de      Z  G dՄ de      Z! G dׄ de      Z" G dل de      Z# G dۄ de      Z$ G d݄ de      Z% G d߄ de      Z& G d de      Z' G d de      Z( G d de      Z) G d de      Z* G d de      Z+ G d de      Z, G d de      Z- G d de      Z. G d de      Z/ G d de      Z0 G d de      Z1 G d de      Z2 G d de      Z3 G d de      Z4 G d de      Z5 G d d e      Z6 G d de      Z7 G d de      Z8 G d de      Z9 G d de      Z: G d	 d
e      Z; G d de      Z< G d de      Z= G d de      Z> G d de      Z? G d de      Z@ G d de      ZA G d de      ZB G d de      ZC G d de      ZD G d de      ZE G d d e      ZF G d! d"e      ZG G d# d$e      ZH G d% d&e      ZI G d' d(e      ZJ G d) d*e      ZK G d+ d,e      ZL G d- d.e      ZM G d/ d0e      ZN G d1 d2e      ZO G d3 d4e      ZP G d5 d6e      ZQ G d7 d8e      ZR G d9 d:e      ZS G d; d<e      ZT G d= d>e      ZU G d? d@e      ZV G dA dBe      ZW G dC dDe      ZX G dE dFe      ZY G dG dHe      ZZ G dI dJe      Z[ G dK dLe      Z\ G dM dNe      Z] G dO dPe      Z^ G dQ dRe      Z_ G dS dTe      Z` G dU dVe      Za G dW dXe      Zb G dY dZe      Zc G d[ d\e      Zd G d] d^e      Ze G d_ d`e      Zf G da dbe      Zg G dc dde      Zh G de dfe      Zi G dg dhe      Zj G di dje      Zk G dk dle      Zl G dm dne      Zm G do dpe      Zn G dq dre      Zo G ds dte      Zp G du dve      Zq G dw dxe      Zr G dy dze      Zs G d{ d|e      Zt G d} d~e      Zu G d de      Zv G d de      Zw G d de      Zx G d de      Zy G d de      Zz G d de      Z{ G d de      Z| G d de      Z} G d de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G dĄ de      Z G dƄ de      Z G dȄ de      Z G dʄ de      Z G d̄ de      Z G d΄ de      Z G dЄ de      Z G d҄ de      Z G dԄ de      Z G dք de      Zd؄ Z G dل de      Z G dۄ de      Z G d݄ de      Z G d߄ de      Z G d de      Z G d de      Z G d de      Z G d de      Zd Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G 	d  	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	d	e      Z G 	d
 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d  	d!e      Z G 	d" 	d#e      Z G 	d$ 	d%e      Z G 	d& 	d'e      Z G 	d( 	d)e      Z G 	d* 	d+e      Z G 	d, 	d-e      Z G 	d. 	d/e      Z G 	d0 	d1e      Z G 	d2 	d3e      Z G 	d4 	d5e      Z G 	d6 	d7e      Z G 	d8 	d9e      Z G 	d: 	d;e      Z G 	d< 	d=e      Z G 	d> 	d?e      Z G 	d@ 	dAe      Z G 	dB 	dCe      Z G 	dD 	dEe      Z G 	dF 	dGe      Z G 	dH 	dIe      Z G 	dJ 	dKe      Z G 	dL 	dMe      Z G 	dN 	dOe      Z G 	dP 	dQe      Z G 	dR 	dSe      Z G 	dT 	dUe      Z G 	dV 	dWe      Z G 	dX 	dYe      Z G 	dZ 	d[e      Z G 	d\ 	d]e      Z G 	d^ 	d_e      Z G 	d` 	dae      Z G 	db 	dce      Z G 	dd 	dee      Z G 	df 	dge      Z G 	dh 	die      Z G 	dj 	dke      Z G 	dl 	dme      Z G 	dn 	doe      Z G 	dp 	dqe      Z G 	dr 	dse      Z G 	dt 	due      Z G 	dv 	dwe      Z G 	dx 	dye      Z G 	dz 	d{e      Z G 	d| 	d}e      Z G 	d~ 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z  G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z	 G 	d 	de      Z
 G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z G 	d 	de      Z	dĄ Z G 	dń 	de      Z G 	dǄ 	de      Z G 	dɄ 	de      Z G 	d˄ 	de      Z G 	d̈́ 	de      Z G 	dτ 	de      Z  G 	dф 	de      Z! G 	dӄ 	de      Z"	dՄ Z# G 	dք 	de      Z$ G 	d؄ 	de      Z% G 	dڄ 	de      Z& G 	d܄ 	de      Z' G 	dބ 	de      Z( G 	d 	de      Z) G 	d 	de      Z* G 	d 	de      Z+ G 	d 	de      Z, G 	d 	de      Z- G 	d 	de      Z. G 	d 	de      Z/ G 	d 	de      Z0 G 	d 	de      Z1 G 	d 	de      Z2 G 	d 	de      Z3 G 	d 	de      Z4 G 	d 	de      Z5 G 	d 	de      Z6 G 	d 	de      Z7 G 	d 	de      Z8 G 
d  
de      Z9 G 
d 
de      Z: G 
d 
de      Z; G 
d 
de      Z< G 
d 
d	e      Z= G 
d
 
de      Z> G 
d 
de      Z? G 
d 
de      Z@ G 
d 
de      ZA G 
d 
de      ZB G 
d 
de      ZC G 
d 
de      ZD G 
d 
de      ZE G 
d 
de      ZF G 
d 
de      ZG G 
d 
de      ZH G 
d  
d!e      ZI G 
d" 
d#e      ZJ G 
d$ 
d%e      ZK G 
d& 
d'e      ZL G 
d( 
d)e      ZM G 
d* 
d+e      ZN G 
d, 
d-e      ZO G 
d. 
d/e      ZP G 
d0 
d1e      ZQ G 
d2 
d3e      ZR G 
d4 
d5e      ZS G 
d6 
d7e      ZT G 
d8 
d9e      ZU G 
d: 
d;e      ZV G 
d< 
d=e      ZW G 
d> 
d?e      ZX G 
d@ 
dAe      ZY G 
dB 
dCe      ZZ G 
dD 
dEe      Z[ G 
dF 
dGe      Z\ G 
dH 
dIe      Z] G 
dJ 
dKe      Z^ G 
dL 
dMe      Z_ G 
dN 
dOe      Z` G 
dP 
dQe      Za G 
dR 
dSe      Zb G 
dT 
dUe      Zc G 
dV 
dWe      Zd G 
dX 
dYe      Ze G 
dZ 
d[e      Zf G 
d\ 
d]e      Zg G 
d^ 
d_e      Zh G 
d` 
dae      Zi G 
db 
dce      Zj G 
dd 
dee      Zk G 
df 
dge      Zl G 
dh 
die      Zm G 
dj 
dke      Zn G 
dl 
dme      Zo G 
dn 
doe      Zp G 
dp 
dqe      Zq G 
dr 
dse      Zr G 
dt 
due      Zs G 
dv 
dwe      Zt G 
dx 
dye      Zu G 
dz 
d{e      Zv G 
d| 
d}e      Zw G 
d~ 
de      Zx G 
d 
de      Zy G 
d 
de      Zz G 
d 
de      Z{ G 
d 
de      Z| G 
d 
de      Z} G 
d 
de      Z~ G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
dĄ 
de      Z G 
dƄ 
de      Z G 
dȄ 
de      Z G 
dʄ 
de      Z G 
d̄ 
de      Z G 
d΄ 
de      Z G 
dЄ 
de      Z G 
d҄ 
de      Z G 
dԄ 
de      Z G 
dք 
de      Z G 
d؄ 
de      Z G 
dڄ 
de      Z G 
d܄ 
de      Z G 
dބ 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G 
d 
de      Z G d  de      Z G d de      Z G d de      Z G d de      Z G d d	e      Zd
 Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d! d"e      Z G d# d$e      Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Z G d1 d2e      Z G d3 d4e      Z G d5 d6e      Z G d7 d8e      Z G d9 d:e      Z G d; d<e      Z G d= d>e      Zؐd? Zِd@ ZڐdA ZېdB ZܐdC ZݐdD ZސdE ZߐdF ZdG Z G dH dIe      ZdJ ZdK Z G dL dMe      ZdN Z G dO dPe      Zy(Q     )DummyObjectrequires_backendsc                       e Zd ZdgZd Zy)PyTorchBenchmarktorchc                     t        | dg       y Nr   r   selfargskwargss      V/var/www/html/venv/lib/python3.12/site-packages/transformers/utils/dummy_pt_objects.py__init__zPyTorchBenchmark.__init__       $	*    N__name__
__module____qualname__	_backendsr    r   r   r   r          	I+r   r   )	metaclassc                       e Zd ZdgZd Zy)PyTorchBenchmarkArgumentsr   c                     t        | dg       y r	   r
   r   s      r   r   z"PyTorchBenchmarkArguments.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)Cacher   c                     t        | dg       y r	   r
   r   s      r   r   zCache.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)CacheConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zCacheConfig.__init__   r   r   Nr   r   r   r   r"   r"      r   r   r"   c                       e Zd ZdgZd Zy)DynamicCacher   c                     t        | dg       y r	   r
   r   s      r   r   zDynamicCache.__init__$   r   r   Nr   r   r   r   r%   r%   !   r   r   r%   c                       e Zd ZdgZd Zy)EncoderDecoderCacher   c                     t        | dg       y r	   r
   r   s      r   r   zEncoderDecoderCache.__init__+   r   r   Nr   r   r   r   r(   r(   (   r   r   r(   c                       e Zd ZdgZd Zy)HQQQuantizedCacher   c                     t        | dg       y r	   r
   r   s      r   r   zHQQQuantizedCache.__init__2   r   r   Nr   r   r   r   r+   r+   /   r   r   r+   c                       e Zd ZdgZd Zy)HybridCacher   c                     t        | dg       y r	   r
   r   s      r   r   zHybridCache.__init__9   r   r   Nr   r   r   r   r.   r.   6   r   r   r.   c                       e Zd ZdgZd Zy)
MambaCacher   c                     t        | dg       y r	   r
   r   s      r   r   zMambaCache.__init__@   r   r   Nr   r   r   r   r1   r1   =   r   r   r1   c                       e Zd ZdgZd Zy)OffloadedCacher   c                     t        | dg       y r	   r
   r   s      r   r   zOffloadedCache.__init__G   r   r   Nr   r   r   r   r4   r4   D   r   r   r4   c                       e Zd ZdgZd Zy)OffloadedStaticCacher   c                     t        | dg       y r	   r
   r   s      r   r   zOffloadedStaticCache.__init__N   r   r   Nr   r   r   r   r7   r7   K   r   r   r7   c                       e Zd ZdgZd Zy)QuantizedCacher   c                     t        | dg       y r	   r
   r   s      r   r   zQuantizedCache.__init__U   r   r   Nr   r   r   r   r:   r:   R   r   r   r:   c                       e Zd ZdgZd Zy)QuantizedCacheConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zQuantizedCacheConfig.__init__\   r   r   Nr   r   r   r   r=   r=   Y   r   r   r=   c                       e Zd ZdgZd Zy)QuantoQuantizedCacher   c                     t        | dg       y r	   r
   r   s      r   r   zQuantoQuantizedCache.__init__c   r   r   Nr   r   r   r   r@   r@   `   r   r   r@   c                       e Zd ZdgZd Zy)	SinkCacher   c                     t        | dg       y r	   r
   r   s      r   r   zSinkCache.__init__j   r   r   Nr   r   r   r   rC   rC   g   r   r   rC   c                       e Zd ZdgZd Zy)SlidingWindowCacher   c                     t        | dg       y r	   r
   r   s      r   r   zSlidingWindowCache.__init__q   r   r   Nr   r   r   r   rF   rF   n   r   r   rF   c                       e Zd ZdgZd Zy)StaticCacher   c                     t        | dg       y r	   r
   r   s      r   r   zStaticCache.__init__x   r   r   Nr   r   r   r   rI   rI   u   r   r   rI   c                       e Zd ZdgZd Zy)GlueDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   zGlueDataset.__init__   r   r   Nr   r   r   r   rL   rL   |   r   r   rL   c                       e Zd ZdgZd Zy)GlueDataTrainingArgumentsr   c                     t        | dg       y r	   r
   r   s      r   r   z"GlueDataTrainingArguments.__init__   r   r   Nr   r   r   r   rO   rO      r   r   rO   c                       e Zd ZdgZd Zy)LineByLineTextDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   zLineByLineTextDataset.__init__   r   r   Nr   r   r   r   rR   rR      r   r   rR   c                       e Zd ZdgZd Zy)LineByLineWithRefDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   z!LineByLineWithRefDataset.__init__   r   r   Nr   r   r   r   rU   rU      r   r   rU   c                       e Zd ZdgZd Zy)LineByLineWithSOPTextDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   z%LineByLineWithSOPTextDataset.__init__   r   r   Nr   r   r   r   rX   rX      r   r   rX   c                       e Zd ZdgZd Zy)SquadDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   zSquadDataset.__init__   r   r   Nr   r   r   r   r[   r[      r   r   r[   c                       e Zd ZdgZd Zy)SquadDataTrainingArgumentsr   c                     t        | dg       y r	   r
   r   s      r   r   z#SquadDataTrainingArguments.__init__   r   r   Nr   r   r   r   r^   r^      r   r   r^   c                       e Zd ZdgZd Zy)TextDatasetr   c                     t        | dg       y r	   r
   r   s      r   r   zTextDataset.__init__   r   r   Nr   r   r   r   ra   ra      r   r   ra   c                       e Zd ZdgZd Zy)$TextDatasetForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z-TextDatasetForNextSentencePrediction.__init__   r   r   Nr   r   r   r   rd   rd      r   r   rd   c                       e Zd ZdgZd Zy)#AlternatingCodebooksLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z,AlternatingCodebooksLogitsProcessor.__init__   r   r   Nr   r   r   r   rg   rg      r   r   rg   c                       e Zd ZdgZd Zy)BayesianDetectorConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zBayesianDetectorConfig.__init__   r   r   Nr   r   r   r   rj   rj      r   r   rj   c                       e Zd ZdgZd Zy)BayesianDetectorModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBayesianDetectorModel.__init__   r   r   Nr   r   r   r   rm   rm      r   r   rm   c                       e Zd ZdgZd Zy)
BeamScorerr   c                     t        | dg       y r	   r
   r   s      r   r   zBeamScorer.__init__   r   r   Nr   r   r   r   rp   rp      r   r   rp   c                       e Zd ZdgZd Zy)BeamSearchScorerr   c                     t        | dg       y r	   r
   r   s      r   r   zBeamSearchScorer.__init__   r   r   Nr   r   r   r   rs   rs      r   r   rs   c                       e Zd ZdgZd Zy)%ClassifierFreeGuidanceLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z.ClassifierFreeGuidanceLogitsProcessor.__init__   r   r   Nr   r   r   r   rv   rv      r   r   rv   c                       e Zd ZdgZd Zy)ConstrainedBeamSearchScorerr   c                     t        | dg       y r	   r
   r   s      r   r   z$ConstrainedBeamSearchScorer.__init__   r   r   Nr   r   r   r   ry   ry      r   r   ry   c                       e Zd ZdgZd Zy)
Constraintr   c                     t        | dg       y r	   r
   r   s      r   r   zConstraint.__init__   r   r   Nr   r   r   r   r|   r|      r   r   r|   c                       e Zd ZdgZd Zy)ConstraintListStater   c                     t        | dg       y r	   r
   r   s      r   r   zConstraintListState.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)DisjunctiveConstraintr   c                     t        | dg       y r	   r
   r   s      r   r   zDisjunctiveConstraint.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)#EncoderNoRepeatNGramLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z,EncoderNoRepeatNGramLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)'EncoderRepetitionPenaltyLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z0EncoderRepetitionPenaltyLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)EosTokenCriteriar   c                     t        | dg       y r	   r
   r   s      r   r   zEosTokenCriteria.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)EpsilonLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zEpsilonLogitsWarper.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)EtaLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zEtaLogitsWarper.__init__   r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)ExponentialDecayLengthPenaltyr   c                     t        | dg       y r	   r
   r   s      r   r   z&ExponentialDecayLengthPenalty.__init__'  r   r   Nr   r   r   r   r   r   $  r   r   r   c                       e Zd ZdgZd Zy)ForcedBOSTokenLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z&ForcedBOSTokenLogitsProcessor.__init__.  r   r   Nr   r   r   r   r   r   +  r   r   r   c                       e Zd ZdgZd Zy)ForcedEOSTokenLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z&ForcedEOSTokenLogitsProcessor.__init__5  r   r   Nr   r   r   r   r   r   2  r   r   r   c                       e Zd ZdgZd Zy)GenerationMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zGenerationMixin.__init__<  r   r   Nr   r   r   r   r   r   9  r   r   r   c                       e Zd ZdgZd Zy)HammingDiversityLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z(HammingDiversityLogitsProcessor.__init__C  r   r   Nr   r   r   r   r   r   @  r   r   r   c                       e Zd ZdgZd Zy)InfNanRemoveLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z$InfNanRemoveLogitsProcessor.__init__J  r   r   Nr   r   r   r   r   r   G  r   r   r   c                       e Zd ZdgZd Zy)LogitNormalizationr   c                     t        | dg       y r	   r
   r   s      r   r   zLogitNormalization.__init__Q  r   r   Nr   r   r   r   r   r   N  r   r   r   c                       e Zd ZdgZd Zy)LogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zLogitsProcessor.__init__X  r   r   Nr   r   r   r   r   r   U  r   r   r   c                       e Zd ZdgZd Zy)LogitsProcessorListr   c                     t        | dg       y r	   r
   r   s      r   r   zLogitsProcessorList.__init___  r   r   Nr   r   r   r   r   r   \  r   r   r   c                       e Zd ZdgZd Zy)LogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zLogitsWarper.__init__f  r   r   Nr   r   r   r   r   r   c  r   r   r   c                       e Zd ZdgZd Zy)MaxLengthCriteriar   c                     t        | dg       y r	   r
   r   s      r   r   zMaxLengthCriteria.__init__m  r   r   Nr   r   r   r   r   r   j  r   r   r   c                       e Zd ZdgZd Zy)MaxTimeCriteriar   c                     t        | dg       y r	   r
   r   s      r   r   zMaxTimeCriteria.__init__t  r   r   Nr   r   r   r   r   r   q  r   r   r   c                       e Zd ZdgZd Zy)MinLengthLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z!MinLengthLogitsProcessor.__init__{  r   r   Nr   r   r   r   r   r   x  r   r   r   c                       e Zd ZdgZd Zy)!MinNewTokensLengthLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z*MinNewTokensLengthLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)MinPLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zMinPLogitsWarper.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)NoBadWordsLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z"NoBadWordsLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)NoRepeatNGramLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z%NoRepeatNGramLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)PhrasalConstraintr   c                     t        | dg       y r	   r
   r   s      r   r   zPhrasalConstraint.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy) PrefixConstrainedLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z)PrefixConstrainedLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy) RepetitionPenaltyLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z)RepetitionPenaltyLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)SequenceBiasLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z$SequenceBiasLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)StoppingCriteriar   c                     t        | dg       y r	   r
   r   s      r   r   zStoppingCriteria.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)StoppingCriteriaListr   c                     t        | dg       y r	   r
   r   s      r   r   zStoppingCriteriaList.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)StopStringCriteriar   c                     t        | dg       y r	   r
   r   s      r   r   zStopStringCriteria.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)$SuppressTokensAtBeginLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z-SuppressTokensAtBeginLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)SuppressTokensLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z&SuppressTokensLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)SynthIDTextWatermarkDetectorr   c                     t        | dg       y r	   r
   r   s      r   r   z%SynthIDTextWatermarkDetector.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)SynthIDTextWatermarkingConfigr   c                     t        | dg       y r	   r
   r   s      r   r   z&SynthIDTextWatermarkingConfig.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)#SynthIDTextWatermarkLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z,SynthIDTextWatermarkLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)TemperatureLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   z TemperatureLogitsWarper.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)TopKLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zTopKLogitsWarper.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)TopPLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zTopPLogitsWarper.__init__   r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)TypicalLogitsWarperr   c                     t        | dg       y r	   r
   r   s      r   r   zTypicalLogitsWarper.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy).UnbatchedClassifierFreeGuidanceLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z7UnbatchedClassifierFreeGuidanceLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)WatermarkDetectorr   c                     t        | dg       y r	   r
   r   s      r   r   zWatermarkDetector.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)WatermarkLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z!WatermarkLogitsProcessor.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)WhisperTimeStampLogitsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   z(WhisperTimeStampLogitsProcessor.__init__#  r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)$TorchExportableModuleWithStaticCacher   c                     t        | dg       y r	   r
   r   s      r   r   z-TorchExportableModuleWithStaticCache.__init__*  r   r   Nr   r   r   r   r  r  '  r   r   r  c                  &    t        t        dg       y r	   )r   convert_and_export_with_cacher   r   s     r   r  r  .      3gY?r   Nc                       e Zd ZdgZd Zy)PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPreTrainedModel.__init__8  r   r   Nr   r   r   r   r
  r
  5  r   r   r
  c                       e Zd ZdgZd Zy)AlbertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zAlbertForMaskedLM.__init__?  r   r   Nr   r   r   r   r  r  <  r   r   r  c                       e Zd ZdgZd Zy)AlbertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z AlbertForMultipleChoice.__init__F  r   r   Nr   r   r   r   r  r  C  r   r   r  c                       e Zd ZdgZd Zy)AlbertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zAlbertForPreTraining.__init__M  r   r   Nr   r   r   r   r  r  J  r   r   r  c                       e Zd ZdgZd Zy)AlbertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#AlbertForQuestionAnswering.__init__T  r   r   Nr   r   r   r   r  r  Q  r   r   r  c                       e Zd ZdgZd Zy)AlbertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(AlbertForSequenceClassification.__init__[  r   r   Nr   r   r   r   r  r  X  r   r   r  c                       e Zd ZdgZd Zy)AlbertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%AlbertForTokenClassification.__init__b  r   r   Nr   r   r   r   r  r  _  r   r   r  c                       e Zd ZdgZd Zy)AlbertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlbertModel.__init__i  r   r   Nr   r   r   r   r  r  f  r   r   r  c                       e Zd ZdgZd Zy)AlbertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlbertPreTrainedModel.__init__p  r   r   Nr   r   r   r   r"  r"  m  r   r   r"  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_albertr  s     r   r%  r%  t      /';r   c                       e Zd ZdgZd Zy)
AlignModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlignModel.__init__{  r   r   Nr   r   r   r   r(  r(  x  r   r   r(  c                       e Zd ZdgZd Zy)AlignPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlignPreTrainedModel.__init__  r   r   Nr   r   r   r   r+  r+    r   r   r+  c                       e Zd ZdgZd Zy)AlignTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlignTextModel.__init__  r   r   Nr   r   r   r   r.  r.    r   r   r.  c                       e Zd ZdgZd Zy)AlignVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAlignVisionModel.__init__  r   r   Nr   r   r   r   r1  r1    r   r   r1  c                       e Zd ZdgZd Zy)AltCLIPModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAltCLIPModel.__init__  r   r   Nr   r   r   r   r4  r4    r   r   r4  c                       e Zd ZdgZd Zy)AltCLIPPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAltCLIPPreTrainedModel.__init__  r   r   Nr   r   r   r   r7  r7    r   r   r7  c                       e Zd ZdgZd Zy)AltCLIPTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAltCLIPTextModel.__init__  r   r   Nr   r   r   r   r:  r:    r   r   r:  c                       e Zd ZdgZd Zy)AltCLIPVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAltCLIPVisionModel.__init__  r   r   Nr   r   r   r   r=  r=    r   r   r=  c                       e Zd ZdgZd Zy)ASTForAudioClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"ASTForAudioClassification.__init__  r   r   Nr   r   r   r   r@  r@    r   r   r@  c                       e Zd ZdgZd Zy)ASTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zASTModel.__init__  r   r   Nr   r   r   r   rC  rC    r   r   rC  c                       e Zd ZdgZd Zy)ASTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zASTPreTrainedModel.__init__  r   r   Nr   r   r   r   rF  rF    r   r   rF  c                       e Zd ZdgZd Zy)AutoBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zAutoBackbone.__init__I  r   r   Nr   r   r   r   rI  rI  F  r   r   rI  c                       e Zd ZdgZd Zy)	AutoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModel.__init__P  r   r   Nr   r   r   r   rL  rL  M  r   r   rL  c                       e Zd ZdgZd Zy)AutoModelForAudioClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoModelForAudioClassification.__init__W  r   r   Nr   r   r   r   rO  rO  T  r   r   rO  c                       e Zd ZdgZd Zy)$AutoModelForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-AutoModelForAudioFrameClassification.__init__^  r   r   Nr   r   r   r   rR  rR  [  r   r   rR  c                       e Zd ZdgZd Zy)AutoModelForAudioXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   z!AutoModelForAudioXVector.__init__e  r   r   Nr   r   r   r   rU  rU  b  r   r   rU  c                       e Zd ZdgZd Zy)AutoModelForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelForCausalLM.__init__l  r   r   Nr   r   r   r   rX  rX  i  r   r   rX  c                       e Zd ZdgZd Zy)AutoModelForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelForCTC.__init__s  r   r   Nr   r   r   r   r[  r[  p  r   r   r[  c                       e Zd ZdgZd Zy)AutoModelForDepthEstimationr   c                     t        | dg       y r	   r
   r   s      r   r   z$AutoModelForDepthEstimation.__init__z  r   r   Nr   r   r   r   r^  r^  w  r   r   r^  c                       e Zd ZdgZd Zy)%AutoModelForDocumentQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z.AutoModelForDocumentQuestionAnswering.__init__  r   r   Nr   r   r   r   ra  ra  ~  r   r   ra  c                       e Zd ZdgZd Zy)AutoModelForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoModelForImageClassification.__init__  r   r   Nr   r   r   r   rd  rd    r   r   rd  c                       e Zd ZdgZd Zy)AutoModelForImageSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoModelForImageSegmentation.__init__  r   r   Nr   r   r   r   rg  rg    r   r   rg  c                       e Zd ZdgZd Zy)AutoModelForImageTextToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z$AutoModelForImageTextToText.__init__  r   r   Nr   r   r   r   rj  rj    r   r   rj  c                       e Zd ZdgZd Zy)AutoModelForImageToImager   c                     t        | dg       y r	   r
   r   s      r   r   z!AutoModelForImageToImage.__init__  r   r   Nr   r   r   r   rm  rm    r   r   rm  c                       e Zd ZdgZd Zy) AutoModelForInstanceSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z)AutoModelForInstanceSegmentation.__init__  r   r   Nr   r   r   r   rp  rp    r   r   rp  c                       e Zd ZdgZd Zy)AutoModelForKeypointDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoModelForKeypointDetection.__init__  r   r   Nr   r   r   r   rs  rs    r   r   rs  c                       e Zd ZdgZd Zy)AutoModelForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoModelForMaskedImageModeling.__init__  r   r   Nr   r   r   r   rv  rv    r   r   rv  c                       e Zd ZdgZd Zy)AutoModelForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelForMaskedLM.__init__  r   r   Nr   r   r   r   ry  ry    r   r   ry  c                       e Zd ZdgZd Zy)AutoModelForMaskGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoModelForMaskGeneration.__init__  r   r   Nr   r   r   r   r|  r|    r   r   r|  c                       e Zd ZdgZd Zy)AutoModelForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoModelForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"AutoModelForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z+AutoModelForNextSentencePrediction.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z$AutoModelForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z AutoModelForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoModelForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) AutoModelForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z)AutoModelForSemanticSegmentation.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForSeq2SeqLMr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelForSeq2SeqLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"AutoModelForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+AutoModelForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForSpeechSeq2Seqr   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoModelForSpeechSeq2Seq.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"AutoModelForTableQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z+AutoModelForTableQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForTextEncodingr   c                     t        | dg       y r	   r
   r   s      r   r   z!AutoModelForTextEncoding.__init__  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)AutoModelForTextToSpectrogramr   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoModelForTextToSpectrogram.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForTextToWaveformr   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoModelForTextToWaveform.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)AutoModelForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoModelForTokenClassification.__init__"  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!AutoModelForUniversalSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z*AutoModelForUniversalSegmentation.__init__)  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)AutoModelForVideoClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoModelForVideoClassification.__init__0  r   r   Nr   r   r   r   r  r  -  r   r   r  c                       e Zd ZdgZd Zy)AutoModelForVision2Seqr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelForVision2Seq.__init__7  r   r   Nr   r   r   r   r  r  4  r   r   r  c                       e Zd ZdgZd Zy)#AutoModelForVisualQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z,AutoModelForVisualQuestionAnswering.__init__>  r   r   Nr   r   r   r   r  r  ;  r   r   r  c                       e Zd ZdgZd Zy)'AutoModelForZeroShotImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0AutoModelForZeroShotImageClassification.__init__E  r   r   Nr   r   r   r   r  r  B  r   r   r  c                       e Zd ZdgZd Zy)#AutoModelForZeroShotObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z,AutoModelForZeroShotObjectDetection.__init__L  r   r   Nr   r   r   r   r  r  I  r   r   r  c                       e Zd ZdgZd Zy)AutoModelWithLMHeadr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModelWithLMHead.__init__S  r   r   Nr   r   r   r   r  r  P  r   r   r  c                       e Zd ZdgZd Zy)AutoformerForPredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z AutoformerForPrediction.__init__Z  r   r   Nr   r   r   r   r  r  W  r   r   r  c                       e Zd ZdgZd Zy)AutoformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoformerModel.__init__a  r   r   Nr   r   r   r   r  r  ^  r   r   r  c                       e Zd ZdgZd Zy)AutoformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoformerPreTrainedModel.__init__h  r   r   Nr   r   r   r   r  r  e  r   r   r  c                       e Zd ZdgZd Zy)BarkCausalModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkCausalModel.__init__o  r   r   Nr   r   r   r   r  r  l  r   r   r  c                       e Zd ZdgZd Zy)BarkCoarseModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkCoarseModel.__init__v  r   r   Nr   r   r   r   r  r  s  r   r   r  c                       e Zd ZdgZd Zy)BarkFineModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkFineModel.__init__}  r   r   Nr   r   r   r   r  r  z  r   r   r  c                       e Zd ZdgZd Zy)	BarkModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BarkPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BarkSemanticModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBarkSemanticModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBartForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z%BartForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!BartForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&BartForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	BartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBartModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBartPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BartPretrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBartPretrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PretrainedBartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPretrainedBartModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BeitBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zBeitBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BeitForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#BeitForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BeitForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z#BeitForMaskedImageModeling.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BeitForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z$BeitForSemanticSegmentation.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	BeitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBeitModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BeitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBeitPreTrainedModel.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)BertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBertForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BertForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z&BertForNextSentencePrediction.__init__	  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)BertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zBertForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!BertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&BertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#BertForTokenClassification.__init__%  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)BertLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBertLMHeadModel.__init__,  r   r   Nr   r   r   r   r  r  )  r   r   r  c                       e Zd ZdgZd Zy)	BertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBertModel.__init__3  r   r   Nr   r   r   r   r  r  0  r   r   r  c                       e Zd ZdgZd Zy)BertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBertPreTrainedModel.__init__:  r   r   Nr   r   r   r   r  r  7  r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_bertr  s     r   r!  r!  >      -y9r   c                       e Zd ZdgZd Zy)BertGenerationDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   zBertGenerationDecoder.__init__E  r   r   Nr   r   r   r   r$  r$  B  r   r   r$  c                       e Zd ZdgZd Zy)BertGenerationEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zBertGenerationEncoder.__init__L  r   r   Nr   r   r   r   r'  r'  I  r   r   r'  c                       e Zd ZdgZd Zy)BertGenerationPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&BertGenerationPreTrainedModel.__init__S  r   r   Nr   r   r   r   r*  r*  P  r   r   r*  c                  &    t        t        dg       y r	   )r   "load_tf_weights_in_bert_generationr  s     r   r-  r-  W  s    87)Dr   c                       e Zd ZdgZd Zy)BigBirdForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdForCausalLM.__init__^  r   r   Nr   r   r   r   r/  r/  [  r   r   r/  c                       e Zd ZdgZd Zy)BigBirdForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdForMaskedLM.__init__e  r   r   Nr   r   r   r   r2  r2  b  r   r   r2  c                       e Zd ZdgZd Zy)BigBirdForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!BigBirdForMultipleChoice.__init__l  r   r   Nr   r   r   r   r5  r5  i  r   r   r5  c                       e Zd ZdgZd Zy)BigBirdForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdForPreTraining.__init__s  r   r   Nr   r   r   r   r8  r8  p  r   r   r8  c                       e Zd ZdgZd Zy)BigBirdForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$BigBirdForQuestionAnswering.__init__z  r   r   Nr   r   r   r   r;  r;  w  r   r   r;  c                       e Zd ZdgZd Zy) BigBirdForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)BigBirdForSequenceClassification.__init__  r   r   Nr   r   r   r   r>  r>  ~  r   r   r>  c                       e Zd ZdgZd Zy)BigBirdForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&BigBirdForTokenClassification.__init__  r   r   Nr   r   r   r   rA  rA    r   r   rA  c                       e Zd ZdgZd Zy)BigBirdModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdModel.__init__  r   r   Nr   r   r   r   rD  rD    r   r   rD  c                       e Zd ZdgZd Zy)BigBirdPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdPreTrainedModel.__init__  r   r   Nr   r   r   r   rG  rG    r   r   rG  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_big_birdr  s     r   rJ  rJ        1G9=r   c                       e Zd ZdgZd Zy)BigBirdPegasusForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"BigBirdPegasusForCausalLM.__init__  r   r   Nr   r   r   r   rM  rM    r   r   rM  c                       e Zd ZdgZd Zy)&BigBirdPegasusForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/BigBirdPegasusForConditionalGeneration.__init__  r   r   Nr   r   r   r   rP  rP    r   r   rP  c                       e Zd ZdgZd Zy)"BigBirdPegasusForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z+BigBirdPegasusForQuestionAnswering.__init__  r   r   Nr   r   r   r   rS  rS    r   r   rS  c                       e Zd ZdgZd Zy)'BigBirdPegasusForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0BigBirdPegasusForSequenceClassification.__init__  r   r   Nr   r   r   r   rV  rV    r   r   rV  c                       e Zd ZdgZd Zy)BigBirdPegasusModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBigBirdPegasusModel.__init__  r   r   Nr   r   r   r   rY  rY    r   r   rY  c                       e Zd ZdgZd Zy)BigBirdPegasusPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&BigBirdPegasusPreTrainedModel.__init__  r   r   Nr   r   r   r   r\  r\    r   r   r\  c                       e Zd ZdgZd Zy)BioGptForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBioGptForCausalLM.__init__  r   r   Nr   r   r   r   r_  r_    r   r   r_  c                       e Zd ZdgZd Zy)BioGptForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(BioGptForSequenceClassification.__init__  r   r   Nr   r   r   r   rb  rb    r   r   rb  c                       e Zd ZdgZd Zy)BioGptForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%BioGptForTokenClassification.__init__  r   r   Nr   r   r   r   re  re    r   r   re  c                       e Zd ZdgZd Zy)BioGptModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBioGptModel.__init__  r   r   Nr   r   r   r   rh  rh    r   r   rh  c                       e Zd ZdgZd Zy)BioGptPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBioGptPreTrainedModel.__init__  r   r   Nr   r   r   r   rk  rk    r   r   rk  c                       e Zd ZdgZd Zy)BitBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zBitBackbone.__init__  r   r   Nr   r   r   r   rn  rn    r   r   rn  c                       e Zd ZdgZd Zy)BitForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"BitForImageClassification.__init__  r   r   Nr   r   r   r   rq  rq    r   r   rq  c                       e Zd ZdgZd Zy)BitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBitModel.__init__  r   r   Nr   r   r   r   rt  rt    r   r   rt  c                       e Zd ZdgZd Zy)BitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBitPreTrainedModel.__init__  r   r   Nr   r   r   r   rw  rw     r   r   rw  c                       e Zd ZdgZd Zy)BlenderbotForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBlenderbotForCausalLM.__init__
  r   r   Nr   r   r   r   rz  rz    r   r   rz  c                       e Zd ZdgZd Zy)"BlenderbotForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+BlenderbotForConditionalGeneration.__init__  r   r   Nr   r   r   r   r}  r}    r   r   r}  c                       e Zd ZdgZd Zy)BlenderbotModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlenderbotModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BlenderbotPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"BlenderbotPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BlenderbotSmallForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z#BlenderbotSmallForCausalLM.__init__&  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)'BlenderbotSmallForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z0BlenderbotSmallForConditionalGeneration.__init__-  r   r   Nr   r   r   r   r  r  *  r   r   r  c                       e Zd ZdgZd Zy)BlenderbotSmallModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlenderbotSmallModel.__init__4  r   r   Nr   r   r   r   r  r  1  r   r   r  c                       e Zd ZdgZd Zy)BlenderbotSmallPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'BlenderbotSmallPreTrainedModel.__init__;  r   r   Nr   r   r   r   r  r  8  r   r   r  c                       e Zd ZdgZd Zy)BlipForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z%BlipForConditionalGeneration.__init__B  r   r   Nr   r   r   r   r  r  ?  r   r   r  c                       e Zd ZdgZd Zy)BlipForImageTextRetrievalr   c                     t        | dg       y r	   r
   r   s      r   r   z"BlipForImageTextRetrieval.__init__I  r   r   Nr   r   r   r   r  r  F  r   r   r  c                       e Zd ZdgZd Zy)BlipForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!BlipForQuestionAnswering.__init__P  r   r   Nr   r   r   r   r  r  M  r   r   r  c                       e Zd ZdgZd Zy)	BlipModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlipModel.__init__W  r   r   Nr   r   r   r   r  r  T  r   r   r  c                       e Zd ZdgZd Zy)BlipPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlipPreTrainedModel.__init__^  r   r   Nr   r   r   r   r  r  [  r   r   r  c                       e Zd ZdgZd Zy)BlipTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlipTextModel.__init__e  r   r   Nr   r   r   r   r  r  b  r   r   r  c                       e Zd ZdgZd Zy)BlipVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlipVisionModel.__init__l  r   r   Nr   r   r   r   r  r  i  r   r   r  c                       e Zd ZdgZd Zy)Blip2ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z&Blip2ForConditionalGeneration.__init__s  r   r   Nr   r   r   r   r  r  p  r   r   r  c                       e Zd ZdgZd Zy)Blip2ForImageTextRetrievalr   c                     t        | dg       y r	   r
   r   s      r   r   z#Blip2ForImageTextRetrieval.__init__z  r   r   Nr   r   r   r   r  r  w  r   r   r  c                       e Zd ZdgZd Zy)
Blip2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlip2Model.__init__  r   r   Nr   r   r   r   r  r  ~  r   r   r  c                       e Zd ZdgZd Zy)Blip2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlip2PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Blip2QFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlip2QFormerModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Blip2TextModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z%Blip2TextModelWithProjection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Blip2VisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBlip2VisionModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Blip2VisionModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z'Blip2VisionModelWithProjection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BloomForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBloomForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BloomForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"BloomForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BloomForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'BloomForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BloomForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$BloomForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
BloomModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBloomModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BloomPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBloomPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!BridgeTowerForContrastiveLearningr   c                     t        | dg       y r	   r
   r   s      r   r   z*BridgeTowerForContrastiveLearning.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)#BridgeTowerForImageAndTextRetrievalr   c                     t        | dg       y r	   r
   r   s      r   r   z,BridgeTowerForImageAndTextRetrieval.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BridgeTowerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zBridgeTowerForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BridgeTowerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBridgeTowerModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BridgeTowerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#BridgeTowerPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BrosForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#BrosForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	BrosModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBrosModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BrosPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zBrosPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)BrosProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zBrosProcessor.__init__  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)!BrosSpadeEEForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*BrosSpadeEEForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!BrosSpadeELForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*BrosSpadeELForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)CamembertForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zCamembertForCausalLM.__init__"  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)CamembertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zCamembertForMaskedLM.__init__)  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)CamembertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z#CamembertForMultipleChoice.__init__0  r   r   Nr   r   r   r   r  r  -  r   r   r  c                       e Zd ZdgZd Zy)CamembertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z&CamembertForQuestionAnswering.__init__7  r   r   Nr   r   r   r   r  r  4  r   r   r  c                       e Zd ZdgZd Zy)"CamembertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+CamembertForSequenceClassification.__init__>  r   r   Nr   r   r   r   r  r  ;  r   r   r  c                       e Zd ZdgZd Zy)CamembertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(CamembertForTokenClassification.__init__E  r   r   Nr   r   r   r   r  r  B  r   r   r  c                       e Zd ZdgZd Zy)CamembertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCamembertModel.__init__L  r   r   Nr   r   r   r   r  r  I  r   r   r  c                       e Zd ZdgZd Zy)CamembertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!CamembertPreTrainedModel.__init__S  r   r   Nr   r   r   r   r  r  P  r   r   r  c                       e Zd ZdgZd Zy)CanineForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z CanineForMultipleChoice.__init__Z  r   r   Nr   r   r   r   r
  r
  W  r   r   r
  c                       e Zd ZdgZd Zy)CanineForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#CanineForQuestionAnswering.__init__a  r   r   Nr   r   r   r   r  r  ^  r   r   r  c                       e Zd ZdgZd Zy)CanineForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(CanineForSequenceClassification.__init__h  r   r   Nr   r   r   r   r  r  e  r   r   r  c                       e Zd ZdgZd Zy)CanineForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%CanineForTokenClassification.__init__o  r   r   Nr   r   r   r   r  r  l  r   r   r  c                       e Zd ZdgZd Zy)CanineModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCanineModel.__init__v  r   r   Nr   r   r   r   r  r  s  r   r   r  c                       e Zd ZdgZd Zy)CaninePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCaninePreTrainedModel.__init__}  r   r   Nr   r   r   r   r  r  z  r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_caniner  s     r   r  r    r&  r   c                       e Zd ZdgZd Zy)!ChameleonForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*ChameleonForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ChameleonModelr   c                     t        | dg       y r	   r
   r   s      r   r   zChameleonModel.__init__  r   r   Nr   r   r   r   r!  r!    r   r   r!  c                       e Zd ZdgZd Zy)ChameleonPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!ChameleonPreTrainedModel.__init__  r   r   Nr   r   r   r   r$  r$    r   r   r$  c                       e Zd ZdgZd Zy)ChameleonProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zChameleonProcessor.__init__  r   r   Nr   r   r   r   r'  r'    r   r   r'  c                       e Zd ZdgZd Zy)ChameleonVQVAEr   c                     t        | dg       y r	   r
   r   s      r   r   zChameleonVQVAE.__init__  r   r   Nr   r   r   r   r*  r*    r   r   r*  c                       e Zd ZdgZd Zy)ChineseCLIPModelr   c                     t        | dg       y r	   r
   r   s      r   r   zChineseCLIPModel.__init__  r   r   Nr   r   r   r   r-  r-    r   r   r-  c                       e Zd ZdgZd Zy)ChineseCLIPPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#ChineseCLIPPreTrainedModel.__init__  r   r   Nr   r   r   r   r0  r0    r   r   r0  c                       e Zd ZdgZd Zy)ChineseCLIPTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zChineseCLIPTextModel.__init__  r   r   Nr   r   r   r   r3  r3    r   r   r3  c                       e Zd ZdgZd Zy)ChineseCLIPVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zChineseCLIPVisionModel.__init__  r   r   Nr   r   r   r   r6  r6    r   r   r6  c                       e Zd ZdgZd Zy)ClapAudioModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClapAudioModel.__init__  r   r   Nr   r   r   r   r9  r9    r   r   r9  c                       e Zd ZdgZd Zy)ClapAudioModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z%ClapAudioModelWithProjection.__init__  r   r   Nr   r   r   r   r<  r<    r   r   r<  c                       e Zd ZdgZd Zy)ClapFeatureExtractorr   c                     t        | dg       y r	   r
   r   s      r   r   zClapFeatureExtractor.__init__  r   r   Nr   r   r   r   r?  r?    r   r   r?  c                       e Zd ZdgZd Zy)	ClapModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClapModel.__init__  r   r   Nr   r   r   r   rB  rB    r   r   rB  c                       e Zd ZdgZd Zy)ClapPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClapPreTrainedModel.__init__  r   r   Nr   r   r   r   rE  rE    r   r   rE  c                       e Zd ZdgZd Zy)ClapTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClapTextModel.__init__  r   r   Nr   r   r   r   rH  rH    r   r   rH  c                       e Zd ZdgZd Zy)ClapTextModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z$ClapTextModelWithProjection.__init__  r   r   Nr   r   r   r   rK  rK    r   r   rK  c                       e Zd ZdgZd Zy)CLIPForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#CLIPForImageClassification.__init__  r   r   Nr   r   r   r   rN  rN    r   r   rN  c                       e Zd ZdgZd Zy)	CLIPModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPModel.__init__  r   r   Nr   r   r   r   rQ  rQ    r   r   rQ  c                       e Zd ZdgZd Zy)CLIPPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPPreTrainedModel.__init__  r   r   Nr   r   r   r   rT  rT    r   r   rT  c                       e Zd ZdgZd Zy)CLIPTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPTextModel.__init__  r   r   Nr   r   r   r   rW  rW  
  r   r   rW  c                       e Zd ZdgZd Zy)CLIPTextModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z$CLIPTextModelWithProjection.__init__  r   r   Nr   r   r   r   rZ  rZ    r   r   rZ  c                       e Zd ZdgZd Zy)CLIPVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPVisionModel.__init__  r   r   Nr   r   r   r   r]  r]    r   r   r]  c                       e Zd ZdgZd Zy)CLIPVisionModelWithProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   z&CLIPVisionModelWithProjection.__init__"  r   r   Nr   r   r   r   r`  r`    r   r   r`  c                       e Zd ZdgZd Zy)CLIPSegForImageSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z$CLIPSegForImageSegmentation.__init__)  r   r   Nr   r   r   r   rc  rc  &  r   r   rc  c                       e Zd ZdgZd Zy)CLIPSegModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPSegModel.__init__0  r   r   Nr   r   r   r   rf  rf  -  r   r   rf  c                       e Zd ZdgZd Zy)CLIPSegPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPSegPreTrainedModel.__init__7  r   r   Nr   r   r   r   ri  ri  4  r   r   ri  c                       e Zd ZdgZd Zy)CLIPSegTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPSegTextModel.__init__>  r   r   Nr   r   r   r   rl  rl  ;  r   r   rl  c                       e Zd ZdgZd Zy)CLIPSegVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPSegVisionModel.__init__E  r   r   Nr   r   r   r   ro  ro  B  r   r   ro  c                       e Zd ZdgZd Zy)ClvpDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   zClvpDecoder.__init__L  r   r   Nr   r   r   r   rr  rr  I  r   r   rr  c                       e Zd ZdgZd Zy)ClvpEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zClvpEncoder.__init__S  r   r   Nr   r   r   r   ru  ru  P  r   r   ru  c                       e Zd ZdgZd Zy)ClvpForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zClvpForCausalLM.__init__Z  r   r   Nr   r   r   r   rx  rx  W  r   r   rx  c                       e Zd ZdgZd Zy)	ClvpModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClvpModel.__init__a  r   r   Nr   r   r   r   r{  r{  ^  r   r   r{  c                       e Zd ZdgZd Zy)!ClvpModelForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*ClvpModelForConditionalGeneration.__init__h  r   r   Nr   r   r   r   r~  r~  e  r   r   r~  c                       e Zd ZdgZd Zy)ClvpPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zClvpPreTrainedModel.__init__o  r   r   Nr   r   r   r   r  r  l  r   r   r  c                       e Zd ZdgZd Zy)CodeGenForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zCodeGenForCausalLM.__init__v  r   r   Nr   r   r   r   r  r  s  r   r   r  c                       e Zd ZdgZd Zy)CodeGenModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCodeGenModel.__init__}  r   r   Nr   r   r   r   r  r  z  r   r   r  c                       e Zd ZdgZd Zy)CodeGenPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCodeGenPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)CohereForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zCohereForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)CohereModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCohereModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)CoherePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCoherePreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!ConditionalDetrForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z*ConditionalDetrForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConditionalDetrForSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ConditionalDetrForSegmentation.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConditionalDetrModelr   c                     t        | dg       y r	   r
   r   s      r   r   zConditionalDetrModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConditionalDetrPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'ConditionalDetrPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zConvBertForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z"ConvBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%ConvBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!ConvBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*ConvBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ConvBertForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zConvBertModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z ConvBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_convbertr  s     r   r  r    rK  r   c                       e Zd ZdgZd Zy)ConvNextBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zConvNextBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvNextForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ConvNextForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvNextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zConvNextModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ConvNextPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z ConvNextPreTrainedModel.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)ConvNextV2Backboner   c                     t        | dg       y r	   r
   r   s      r   r   zConvNextV2Backbone.__init__	  r   r   Nr   r   r   r   r  r  
	  r   r   r  c                       e Zd ZdgZd Zy) ConvNextV2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)ConvNextV2ForImageClassification.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)ConvNextV2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zConvNextV2Model.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)ConvNextV2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"ConvNextV2PreTrainedModel.__init__"	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)CpmAntForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zCpmAntForCausalLM.__init__)	  r   r   Nr   r   r   r   r  r  &	  r   r   r  c                       e Zd ZdgZd Zy)CpmAntModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCpmAntModel.__init__0	  r   r   Nr   r   r   r   r  r  -	  r   r   r  c                       e Zd ZdgZd Zy)CpmAntPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCpmAntPreTrainedModel.__init__7	  r   r   Nr   r   r   r   r  r  4	  r   r   r  c                       e Zd ZdgZd Zy)CTRLForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&CTRLForSequenceClassification.__init__>	  r   r   Nr   r   r   r   r  r  ;	  r   r   r  c                       e Zd ZdgZd Zy)CTRLLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCTRLLMHeadModel.__init__E	  r   r   Nr   r   r   r   r  r  B	  r   r   r  c                       e Zd ZdgZd Zy)	CTRLModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCTRLModel.__init__L	  r   r   Nr   r   r   r   r  r  I	  r   r   r  c                       e Zd ZdgZd Zy)CTRLPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCTRLPreTrainedModel.__init__S	  r   r   Nr   r   r   r   r  r  P	  r   r   r  c                       e Zd ZdgZd Zy)CvtForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"CvtForImageClassification.__init__Z	  r   r   Nr   r   r   r   r  r  W	  r   r   r  c                       e Zd ZdgZd Zy)CvtModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCvtModel.__init__a	  r   r   Nr   r   r   r   r  r  ^	  r   r   r  c                       e Zd ZdgZd Zy)CvtPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zCvtPreTrainedModel.__init__h	  r   r   Nr   r   r   r   r  r  e	  r   r   r  c                       e Zd ZdgZd Zy)DacModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDacModel.__init__o	  r   r   Nr   r   r   r   r  r  l	  r   r   r  c                       e Zd ZdgZd Zy)DacPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDacPreTrainedModel.__init__v	  r   r   Nr   r   r   r   r  r  s	  r   r   r  c                       e Zd ZdgZd Zy)(Data2VecAudioForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z1Data2VecAudioForAudioFrameClassification.__init__}	  r   r   Nr   r   r   r   r  r  z	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecAudioForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zData2VecAudioForCTC.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)&Data2VecAudioForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z/Data2VecAudioForSequenceClassification.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecAudioForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   z Data2VecAudioForXVector.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecAudioModelr   c                     t        | dg       y r	   r
   r   s      r   r   zData2VecAudioModel.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecAudioPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%Data2VecAudioPreTrainedModel.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecTextForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z Data2VecTextForCausalLM.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecTextForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z Data2VecTextForMaskedLM.__init__	  r   r   Nr   r   r   r   r
  r
  	  r   r   r
  c                       e Zd ZdgZd Zy)Data2VecTextForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z&Data2VecTextForMultipleChoice.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy) Data2VecTextForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z)Data2VecTextForQuestionAnswering.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)%Data2VecTextForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.Data2VecTextForSequenceClassification.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)"Data2VecTextForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+Data2VecTextForTokenClassification.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zData2VecTextModel.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)Data2VecTextPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$Data2VecTextPreTrainedModel.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)$Data2VecVisionForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-Data2VecVisionForImageClassification.__init__	  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)%Data2VecVisionForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z.Data2VecVisionForSemanticSegmentation.__init__	  r   r   Nr   r   r   r   r"  r"  	  r   r   r"  c                       e Zd ZdgZd Zy)Data2VecVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zData2VecVisionModel.__init__	  r   r   Nr   r   r   r   r%  r%  	  r   r   r%  c                       e Zd ZdgZd Zy)Data2VecVisionPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&Data2VecVisionPreTrainedModel.__init__	  r   r   Nr   r   r   r   r(  r(  	  r   r   r(  c                       e Zd ZdgZd Zy)DbrxForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zDbrxForCausalLM.__init__	  r   r   Nr   r   r   r   r+  r+  	  r   r   r+  c                       e Zd ZdgZd Zy)	DbrxModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDbrxModel.__init__
  r   r   Nr   r   r   r   r.  r.  	  r   r   r.  c                       e Zd ZdgZd Zy)DbrxPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDbrxPreTrainedModel.__init__	
  r   r   Nr   r   r   r   r1  r1  
  r   r   r1  c                       e Zd ZdgZd Zy)DebertaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zDebertaForMaskedLM.__init__
  r   r   Nr   r   r   r   r4  r4  
  r   r   r4  c                       e Zd ZdgZd Zy)DebertaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$DebertaForQuestionAnswering.__init__
  r   r   Nr   r   r   r   r7  r7  
  r   r   r7  c                       e Zd ZdgZd Zy) DebertaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)DebertaForSequenceClassification.__init__
  r   r   Nr   r   r   r   r:  r:  
  r   r   r:  c                       e Zd ZdgZd Zy)DebertaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&DebertaForTokenClassification.__init__%
  r   r   Nr   r   r   r   r=  r=  "
  r   r   r=  c                       e Zd ZdgZd Zy)DebertaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDebertaModel.__init__,
  r   r   Nr   r   r   r   r@  r@  )
  r   r   r@  c                       e Zd ZdgZd Zy)DebertaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDebertaPreTrainedModel.__init__3
  r   r   Nr   r   r   r   rC  rC  0
  r   r   rC  c                       e Zd ZdgZd Zy)DebertaV2ForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zDebertaV2ForMaskedLM.__init__:
  r   r   Nr   r   r   r   rF  rF  7
  r   r   rF  c                       e Zd ZdgZd Zy)DebertaV2ForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z#DebertaV2ForMultipleChoice.__init__A
  r   r   Nr   r   r   r   rI  rI  >
  r   r   rI  c                       e Zd ZdgZd Zy)DebertaV2ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z&DebertaV2ForQuestionAnswering.__init__H
  r   r   Nr   r   r   r   rL  rL  E
  r   r   rL  c                       e Zd ZdgZd Zy)"DebertaV2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+DebertaV2ForSequenceClassification.__init__O
  r   r   Nr   r   r   r   rO  rO  L
  r   r   rO  c                       e Zd ZdgZd Zy)DebertaV2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(DebertaV2ForTokenClassification.__init__V
  r   r   Nr   r   r   r   rR  rR  S
  r   r   rR  c                       e Zd ZdgZd Zy)DebertaV2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zDebertaV2Model.__init__]
  r   r   Nr   r   r   r   rU  rU  Z
  r   r   rU  c                       e Zd ZdgZd Zy)DebertaV2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!DebertaV2PreTrainedModel.__init__d
  r   r   Nr   r   r   r   rX  rX  a
  r   r   rX  c                       e Zd ZdgZd Zy)DecisionTransformerGPT2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   z%DecisionTransformerGPT2Model.__init__k
  r   r   Nr   r   r   r   r[  r[  h
  r   r   r[  c                       e Zd ZdgZd Zy)&DecisionTransformerGPT2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z/DecisionTransformerGPT2PreTrainedModel.__init__r
  r   r   Nr   r   r   r   r^  r^  o
  r   r   r^  c                       e Zd ZdgZd Zy)DecisionTransformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!DecisionTransformerModel.__init__y
  r   r   Nr   r   r   r   ra  ra  v
  r   r   ra  c                       e Zd ZdgZd Zy)"DecisionTransformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z+DecisionTransformerPreTrainedModel.__init__
  r   r   Nr   r   r   r   rd  rd  }
  r   r   rd  c                       e Zd ZdgZd Zy) DeformableDetrForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z)DeformableDetrForObjectDetection.__init__
  r   r   Nr   r   r   r   rg  rg  
  r   r   rg  c                       e Zd ZdgZd Zy)DeformableDetrModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDeformableDetrModel.__init__
  r   r   Nr   r   r   r   rj  rj  
  r   r   rj  c                       e Zd ZdgZd Zy)DeformableDetrPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&DeformableDetrPreTrainedModel.__init__
  r   r   Nr   r   r   r   rm  rm  
  r   r   rm  c                       e Zd ZdgZd Zy)DeiTForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#DeiTForImageClassification.__init__
  r   r   Nr   r   r   r   rp  rp  
  r   r   rp  c                       e Zd ZdgZd Zy)%DeiTForImageClassificationWithTeacherr   c                     t        | dg       y r	   r
   r   s      r   r   z.DeiTForImageClassificationWithTeacher.__init__
  r   r   Nr   r   r   r   rs  rs  
  r   r   rs  c                       e Zd ZdgZd Zy)DeiTForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z#DeiTForMaskedImageModeling.__init__
  r   r   Nr   r   r   r   rv  rv  
  r   r   rv  c                       e Zd ZdgZd Zy)	DeiTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDeiTModel.__init__
  r   r   Nr   r   r   r   ry  ry  
  r   r   ry  c                       e Zd ZdgZd Zy)DeiTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDeiTPreTrainedModel.__init__
  r   r   Nr   r   r   r   r|  r|  
  r   r   r|  c                       e Zd ZdgZd Zy)DetaForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   zDetaForObjectDetection.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)	DetaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDetaModel.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)DetaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDetaPreTrainedModel.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)%EfficientFormerForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.EfficientFormerForImageClassification.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)0EfficientFormerForImageClassificationWithTeacherr   c                     t        | dg       y r	   r
   r   s      r   r   z9EfficientFormerForImageClassificationWithTeacher.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)EfficientFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEfficientFormerModel.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)EfficientFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'EfficientFormerPreTrainedModel.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)ErnieMForInformationExtractionr   c                     t        | dg       y r	   r
   r   s      r   r   z'ErnieMForInformationExtraction.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)ErnieMForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z ErnieMForMultipleChoice.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)ErnieMForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#ErnieMForQuestionAnswering.__init__
  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)ErnieMForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(ErnieMForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ErnieMForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%ErnieMForTokenClassification.__init__  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)ErnieMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieMModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ErnieMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieMPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)&GPTSanJapaneseForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/GPTSanJapaneseForConditionalGeneration.__init__!  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTSanJapaneseModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTSanJapaneseModel.__init__(  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)GPTSanJapanesePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&GPTSanJapanesePreTrainedModel.__init__/  r   r   Nr   r   r   r   r  r  ,  r   r   r  c                       e Zd ZdgZd Zy) GraphormerForGraphClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)GraphormerForGraphClassification.__init__6  r   r   Nr   r   r   r   r  r  3  r   r   r  c                       e Zd ZdgZd Zy)GraphormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGraphormerModel.__init__=  r   r   Nr   r   r   r   r  r  :  r   r   r  c                       e Zd ZdgZd Zy)GraphormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"GraphormerPreTrainedModel.__init__D  r   r   Nr   r   r   r   r  r  A  r   r   r  c                       e Zd ZdgZd Zy)JukeboxModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJukeboxModel.__init__K  r   r   Nr   r   r   r   r  r  H  r   r   r  c                       e Zd ZdgZd Zy)JukeboxPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJukeboxPreTrainedModel.__init__R  r   r   Nr   r   r   r   r  r  O  r   r   r  c                       e Zd ZdgZd Zy)JukeboxPriorr   c                     t        | dg       y r	   r
   r   s      r   r   zJukeboxPrior.__init__Y  r   r   Nr   r   r   r   r  r  V  r   r   r  c                       e Zd ZdgZd Zy)JukeboxVQVAEr   c                     t        | dg       y r	   r
   r   s      r   r   zJukeboxVQVAE.__init__`  r   r   Nr   r   r   r   r  r  ]  r   r   r  c                       e Zd ZdgZd Zy)MCTCTForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zMCTCTForCTC.__init__g  r   r   Nr   r   r   r   r  r  d  r   r   r  c                       e Zd ZdgZd Zy)
MCTCTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMCTCTModel.__init__n  r   r   Nr   r   r   r   r  r  k  r   r   r  c                       e Zd ZdgZd Zy)MCTCTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMCTCTPreTrainedModel.__init__u  r   r   Nr   r   r   r   r  r  r  r   r   r  c                       e Zd ZdgZd Zy)MegaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMegaForCausalLM.__init__|  r   r   Nr   r   r   r   r  r  y  r   r   r  c                       e Zd ZdgZd Zy)MegaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMegaForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MegaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zMegaForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MegaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!MegaForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MegaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&MegaForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MegaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#MegaForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	MegaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMegaModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MegaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMegaPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MMBTForClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   zMMBTForClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	MMBTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMMBTModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ModalEmbeddingsr   c                     t        | dg       y r	   r
   r   s      r   r   zModalEmbeddings.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NatBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zNatBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NatForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"NatForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NatModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNatModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NatPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNatPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zNezhaForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zNezhaForMultipleChoice.__init__  r   r   Nr   r   r   r   r   r     r   r   r   c                       e Zd ZdgZd Zy)NezhaForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z'NezhaForNextSentencePrediction.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zNezhaForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"NezhaForQuestionAnswering.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)NezhaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'NezhaForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$NezhaForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
NezhaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNezhaModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NezhaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNezhaPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)OpenLlamaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zOpenLlamaForCausalLM.__init__$  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)"OpenLlamaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+OpenLlamaForSequenceClassification.__init__+  r   r   Nr   r   r   r   r  r  (  r   r   r  c                       e Zd ZdgZd Zy)OpenLlamaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOpenLlamaModel.__init__2  r   r   Nr   r   r   r   r  r  /  r   r   r  c                       e Zd ZdgZd Zy)OpenLlamaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!OpenLlamaPreTrainedModel.__init__9  r   r   Nr   r   r   r   r!  r!  6  r   r   r!  c                       e Zd ZdgZd Zy)QDQBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zQDQBertForMaskedLM.__init__@  r   r   Nr   r   r   r   r$  r$  =  r   r   r$  c                       e Zd ZdgZd Zy)QDQBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!QDQBertForMultipleChoice.__init__G  r   r   Nr   r   r   r   r'  r'  D  r   r   r'  c                       e Zd ZdgZd Zy) QDQBertForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z)QDQBertForNextSentencePrediction.__init__N  r   r   Nr   r   r   r   r*  r*  K  r   r   r*  c                       e Zd ZdgZd Zy)QDQBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$QDQBertForQuestionAnswering.__init__U  r   r   Nr   r   r   r   r-  r-  R  r   r   r-  c                       e Zd ZdgZd Zy) QDQBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)QDQBertForSequenceClassification.__init__\  r   r   Nr   r   r   r   r0  r0  Y  r   r   r0  c                       e Zd ZdgZd Zy)QDQBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&QDQBertForTokenClassification.__init__c  r   r   Nr   r   r   r   r3  r3  `  r   r   r3  c                       e Zd ZdgZd Zy)QDQBertLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQDQBertLMHeadModel.__init__j  r   r   Nr   r   r   r   r6  r6  g  r   r   r6  c                       e Zd ZdgZd Zy)QDQBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQDQBertModel.__init__q  r   r   Nr   r   r   r   r9  r9  n  r   r   r9  c                       e Zd ZdgZd Zy)QDQBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQDQBertPreTrainedModel.__init__x  r   r   Nr   r   r   r   r<  r<  u  r   r   r<  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_qdqbertr  s     r   r?  r?  |      07)<r   c                       e Zd ZdgZd Zy)RealmEmbedderr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmEmbedder.__init__  r   r   Nr   r   r   r   rB  rB    r   r   rB  c                       e Zd ZdgZd Zy)RealmForOpenQAr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmForOpenQA.__init__  r   r   Nr   r   r   r   rE  rE    r   r   rE  c                       e Zd ZdgZd Zy)RealmKnowledgeAugEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   z!RealmKnowledgeAugEncoder.__init__  r   r   Nr   r   r   r   rH  rH    r   r   rH  c                       e Zd ZdgZd Zy)RealmPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmPreTrainedModel.__init__  r   r   Nr   r   r   r   rK  rK    r   r   rK  c                       e Zd ZdgZd Zy)RealmReaderr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmReader.__init__  r   r   Nr   r   r   r   rN  rN    r   r   rN  c                       e Zd ZdgZd Zy)RealmRetrieverr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmRetriever.__init__  r   r   Nr   r   r   r   rQ  rQ    r   r   rQ  c                       e Zd ZdgZd Zy)RealmScorerr   c                     t        | dg       y r	   r
   r   s      r   r   zRealmScorer.__init__  r   r   Nr   r   r   r   rT  rT    r   r   rT  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_realmr  s     r   rW  rW        .	:r   c                       e Zd ZdgZd Zy)RetriBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRetriBertModel.__init__  r   r   Nr   r   r   r   rZ  rZ    r   r   rZ  c                       e Zd ZdgZd Zy)RetriBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!RetriBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r]  r]    r   r   r]  c                       e Zd ZdgZd Zy)Speech2Text2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z Speech2Text2ForCausalLM.__init__  r   r   Nr   r   r   r   r`  r`    r   r   r`  c                       e Zd ZdgZd Zy)Speech2Text2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$Speech2Text2PreTrainedModel.__init__  r   r   Nr   r   r   r   rc  rc    r   r   rc  c                       e Zd ZdgZd Zy)TrajectoryTransformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#TrajectoryTransformerModel.__init__  r   r   Nr   r   r   r   rf  rf    r   r   rf  c                       e Zd ZdgZd Zy)$TrajectoryTransformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z-TrajectoryTransformerPreTrainedModel.__init__  r   r   Nr   r   r   r   ri  ri    r   r   ri  c                       e Zd ZdgZd Zy)AdaptiveEmbeddingr   c                     t        | dg       y r	   r
   r   s      r   r   zAdaptiveEmbedding.__init__  r   r   Nr   r   r   r   rl  rl    r   r   rl  c                       e Zd ZdgZd Zy)"TransfoXLForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+TransfoXLForSequenceClassification.__init__  r   r   Nr   r   r   r   ro  ro    r   r   ro  c                       e Zd ZdgZd Zy)TransfoXLLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTransfoXLLMHeadModel.__init__  r   r   Nr   r   r   r   rr  rr    r   r   rr  c                       e Zd ZdgZd Zy)TransfoXLModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTransfoXLModel.__init__  r   r   Nr   r   r   r   ru  ru    r   r   ru  c                       e Zd ZdgZd Zy)TransfoXLPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!TransfoXLPreTrainedModel.__init__  r   r   Nr   r   r   r   rx  rx    r   r   rx  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_transfo_xlr  s     r   r{  r{    r  r   c                       e Zd ZdgZd Zy) TvltForAudioVisualClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)TvltForAudioVisualClassification.__init__	  r   r   Nr   r   r   r   r}  r}    r   r   r}  c                       e Zd ZdgZd Zy)TvltForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zTvltForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	TvltModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTvltModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)TvltPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTvltPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)VanForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"VanForImageClassification.__init__%  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)VanModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVanModel.__init__,  r   r   Nr   r   r   r   r  r  )  r   r   r  c                       e Zd ZdgZd Zy)VanPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVanPreTrainedModel.__init__3  r   r   Nr   r   r   r   r  r  0  r   r   r  c                       e Zd ZdgZd Zy)ViTHybridForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(ViTHybridForImageClassification.__init__:  r   r   Nr   r   r   r   r  r  7  r   r   r  c                       e Zd ZdgZd Zy)ViTHybridModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTHybridModel.__init__A  r   r   Nr   r   r   r   r  r  >  r   r   r  c                       e Zd ZdgZd Zy)ViTHybridPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!ViTHybridPreTrainedModel.__init__H  r   r   Nr   r   r   r   r  r  E  r   r   r  c                       e Zd ZdgZd Zy)XLMProphetNetDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMProphetNetDecoder.__init__O  r   r   Nr   r   r   r   r  r  L  r   r   r  c                       e Zd ZdgZd Zy)XLMProphetNetEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMProphetNetEncoder.__init__V  r   r   Nr   r   r   r   r  r  S  r   r   r  c                       e Zd ZdgZd Zy)XLMProphetNetForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z!XLMProphetNetForCausalLM.__init__]  r   r   Nr   r   r   r   r  r  Z  r   r   r  c                       e Zd ZdgZd Zy)%XLMProphetNetForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z.XLMProphetNetForConditionalGeneration.__init__d  r   r   Nr   r   r   r   r  r  a  r   r   r  c                       e Zd ZdgZd Zy)XLMProphetNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMProphetNetModel.__init__k  r   r   Nr   r   r   r   r  r  h  r   r   r  c                       e Zd ZdgZd Zy)XLMProphetNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%XLMProphetNetPreTrainedModel.__init__r  r   r   Nr   r   r   r   r  r  o  r   r   r  c                       e Zd ZdgZd Zy)DepthAnythingForDepthEstimationr   c                     t        | dg       y r	   r
   r   s      r   r   z(DepthAnythingForDepthEstimation.__init__y  r   r   Nr   r   r   r   r  r  v  r   r   r  c                       e Zd ZdgZd Zy)DepthAnythingPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%DepthAnythingPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r  }  r   r   r  c                       e Zd ZdgZd Zy)DetrForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   zDetrForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DetrForSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   zDetrForSegmentation.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	DetrModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDetrModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DetrPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDetrPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DinatBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zDinatBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DinatForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$DinatForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
DinatModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDinatModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DinatPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDinatPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Dinov2Backboner   c                     t        | dg       y r	   r
   r   s      r   r   zDinov2Backbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Dinov2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%Dinov2ForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Dinov2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zDinov2Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Dinov2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDinov2PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DistilBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zDistilBertForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DistilBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$DistilBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DistilBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'DistilBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)#DistilBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,DistilBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) DistilBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)DistilBertForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DistilBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDistilBertModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DistilBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"DistilBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DonutSwinModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDonutSwinModel.__init__  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)DonutSwinPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!DonutSwinPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DPRContextEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zDPRContextEncoder.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DPRPretrainedContextEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   z$DPRPretrainedContextEncoder.__init__!  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)DPRPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDPRPreTrainedModel.__init__(  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)DPRPretrainedQuestionEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   z%DPRPretrainedQuestionEncoder.__init__/  r   r   Nr   r   r   r   r  r  ,  r   r   r  c                       e Zd ZdgZd Zy)DPRPretrainedReaderr   c                     t        | dg       y r	   r
   r   s      r   r   zDPRPretrainedReader.__init__6  r   r   Nr   r   r   r   r  r  3  r   r   r  c                       e Zd ZdgZd Zy)DPRQuestionEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zDPRQuestionEncoder.__init__=  r   r   Nr   r   r   r   r  r  :  r   r   r  c                       e Zd ZdgZd Zy)	DPRReaderr   c                     t        | dg       y r	   r
   r   s      r   r   zDPRReader.__init__D  r   r   Nr   r   r   r   r  r  A  r   r   r  c                       e Zd ZdgZd Zy)DPTForDepthEstimationr   c                     t        | dg       y r	   r
   r   s      r   r   zDPTForDepthEstimation.__init__K  r   r   Nr   r   r   r   r  r  H  r   r   r  c                       e Zd ZdgZd Zy)DPTForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z#DPTForSemanticSegmentation.__init__R  r   r   Nr   r   r   r   r
  r
  O  r   r   r
  c                       e Zd ZdgZd Zy)DPTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDPTModel.__init__Y  r   r   Nr   r   r   r   r  r  V  r   r   r  c                       e Zd ZdgZd Zy)DPTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDPTPreTrainedModel.__init__`  r   r   Nr   r   r   r   r  r  ]  r   r   r  c                       e Zd ZdgZd Zy)"EfficientNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+EfficientNetForImageClassification.__init__g  r   r   Nr   r   r   r   r  r  d  r   r   r  c                       e Zd ZdgZd Zy)EfficientNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEfficientNetModel.__init__n  r   r   Nr   r   r   r   r  r  k  r   r   r  c                       e Zd ZdgZd Zy)EfficientNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$EfficientNetPreTrainedModel.__init__u  r   r   Nr   r   r   r   r  r  r  r   r   r  c                       e Zd ZdgZd Zy)ElectraForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zElectraForCausalLM.__init__|  r   r   Nr   r   r   r   r  r  y  r   r   r  c                       e Zd ZdgZd Zy)ElectraForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zElectraForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ElectraForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!ElectraForMultipleChoice.__init__  r   r   Nr   r   r   r   r"  r"    r   r   r"  c                       e Zd ZdgZd Zy)ElectraForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zElectraForPreTraining.__init__  r   r   Nr   r   r   r   r%  r%    r   r   r%  c                       e Zd ZdgZd Zy)ElectraForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$ElectraForQuestionAnswering.__init__  r   r   Nr   r   r   r   r(  r(    r   r   r(  c                       e Zd ZdgZd Zy) ElectraForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)ElectraForSequenceClassification.__init__  r   r   Nr   r   r   r   r+  r+    r   r   r+  c                       e Zd ZdgZd Zy)ElectraForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&ElectraForTokenClassification.__init__  r   r   Nr   r   r   r   r.  r.    r   r   r.  c                       e Zd ZdgZd Zy)ElectraModelr   c                     t        | dg       y r	   r
   r   s      r   r   zElectraModel.__init__  r   r   Nr   r   r   r   r1  r1    r   r   r1  c                       e Zd ZdgZd Zy)ElectraPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zElectraPreTrainedModel.__init__  r   r   Nr   r   r   r   r4  r4    r   r   r4  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_electrar  s     r   r7  r7    r@  r   c                       e Zd ZdgZd Zy)EncodecModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEncodecModel.__init__  r   r   Nr   r   r   r   r9  r9    r   r   r9  c                       e Zd ZdgZd Zy)EncodecPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEncodecPreTrainedModel.__init__  r   r   Nr   r   r   r   r<  r<    r   r   r<  c                       e Zd ZdgZd Zy)EncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEncoderDecoderModel.__init__  r   r   Nr   r   r   r   r?  r?    r   r   r?  c                       e Zd ZdgZd Zy)ErnieForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieForCausalLM.__init__  r   r   Nr   r   r   r   rB  rB    r   r   rB  c                       e Zd ZdgZd Zy)ErnieForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieForMaskedLM.__init__  r   r   Nr   r   r   r   rE  rE    r   r   rE  c                       e Zd ZdgZd Zy)ErnieForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zErnieForMultipleChoice.__init__  r   r   Nr   r   r   r   rH  rH    r   r   rH  c                       e Zd ZdgZd Zy)ErnieForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z'ErnieForNextSentencePrediction.__init__  r   r   Nr   r   r   r   rK  rK    r   r   rK  c                       e Zd ZdgZd Zy)ErnieForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieForPreTraining.__init__  r   r   Nr   r   r   r   rN  rN    r   r   rN  c                       e Zd ZdgZd Zy)ErnieForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"ErnieForQuestionAnswering.__init__  r   r   Nr   r   r   r   rQ  rQ    r   r   rQ  c                       e Zd ZdgZd Zy)ErnieForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ErnieForSequenceClassification.__init__  r   r   Nr   r   r   r   rT  rT    r   r   rT  c                       e Zd ZdgZd Zy)ErnieForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$ErnieForTokenClassification.__init__  r   r   Nr   r   r   r   rW  rW    r   r   rW  c                       e Zd ZdgZd Zy)
ErnieModelr   c                     t        | dg       y r	   r
   r   s      r   r   zErnieModel.__init__  r   r   Nr   r   r   r   rZ  rZ  	  r   r   rZ  c                       e Zd ZdgZd Zy)ErniePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zErniePreTrainedModel.__init__  r   r   Nr   r   r   r   r]  r]    r   r   r]  c                       e Zd ZdgZd Zy)EsmFoldPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEsmFoldPreTrainedModel.__init__  r   r   Nr   r   r   r   r`  r`    r   r   r`  c                       e Zd ZdgZd Zy)EsmForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zEsmForMaskedLM.__init__!  r   r   Nr   r   r   r   rc  rc    r   r   rc  c                       e Zd ZdgZd Zy)EsmForProteinFoldingr   c                     t        | dg       y r	   r
   r   s      r   r   zEsmForProteinFolding.__init__(  r   r   Nr   r   r   r   rf  rf  %  r   r   rf  c                       e Zd ZdgZd Zy)EsmForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%EsmForSequenceClassification.__init__/  r   r   Nr   r   r   r   ri  ri  ,  r   r   ri  c                       e Zd ZdgZd Zy)EsmForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"EsmForTokenClassification.__init__6  r   r   Nr   r   r   r   rl  rl  3  r   r   rl  c                       e Zd ZdgZd Zy)EsmModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEsmModel.__init__=  r   r   Nr   r   r   r   ro  ro  :  r   r   ro  c                       e Zd ZdgZd Zy)EsmPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEsmPreTrainedModel.__init__D  r   r   Nr   r   r   r   rr  rr  A  r   r   rr  c                       e Zd ZdgZd Zy)FalconForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFalconForCausalLM.__init__K  r   r   Nr   r   r   r   ru  ru  H  r   r   ru  c                       e Zd ZdgZd Zy)FalconForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#FalconForQuestionAnswering.__init__R  r   r   Nr   r   r   r   rx  rx  O  r   r   rx  c                       e Zd ZdgZd Zy)FalconForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(FalconForSequenceClassification.__init__Y  r   r   Nr   r   r   r   r{  r{  V  r   r   r{  c                       e Zd ZdgZd Zy)FalconForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%FalconForTokenClassification.__init__`  r   r   Nr   r   r   r   r~  r~  ]  r   r   r~  c                       e Zd ZdgZd Zy)FalconModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFalconModel.__init__g  r   r   Nr   r   r   r   r  r  d  r   r   r  c                       e Zd ZdgZd Zy)FalconPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFalconPreTrainedModel.__init__n  r   r   Nr   r   r   r   r  r  k  r   r   r  c                       e Zd ZdgZd Zy)FalconMambaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFalconMambaForCausalLM.__init__u  r   r   Nr   r   r   r   r  r  r  r   r   r  c                       e Zd ZdgZd Zy)FalconMambaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFalconMambaModel.__init__|  r   r   Nr   r   r   r   r  r  y  r   r   r  c                       e Zd ZdgZd Zy)FalconMambaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#FalconMambaPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FastSpeech2ConformerHifiGanr   c                     t        | dg       y r	   r
   r   s      r   r   z$FastSpeech2ConformerHifiGan.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FastSpeech2ConformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"FastSpeech2ConformerModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)#FastSpeech2ConformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z,FastSpeech2ConformerPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FastSpeech2ConformerWithHifiGanr   c                     t        | dg       y r	   r
   r   s      r   r   z(FastSpeech2ConformerWithHifiGan.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z"FlaubertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%FlaubertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"FlaubertForQuestionAnsweringSimpler   c                     t        | dg       y r	   r
   r   s      r   r   z+FlaubertForQuestionAnsweringSimple.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!FlaubertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*FlaubertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlaubertForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlaubertModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaubertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlaubertWithLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FlaubertWithLMHeadModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaImageCodebookr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaImageCodebook.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaImageModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaImageModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
FlavaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaMultimodalModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaMultimodalModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FlavaTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFlavaTextModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FNetForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFNetForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FNetForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zFNetForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FNetForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z&FNetForNextSentencePrediction.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FNetForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zFNetForPreTraining.__init__$  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)FNetForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!FNetForQuestionAnswering.__init__+  r   r   Nr   r   r   r   r  r  (  r   r   r  c                       e Zd ZdgZd Zy)FNetForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&FNetForSequenceClassification.__init__2  r   r   Nr   r   r   r   r  r  /  r   r   r  c                       e Zd ZdgZd Zy)FNetForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#FNetForTokenClassification.__init__9  r   r   Nr   r   r   r   r  r  6  r   r   r  c                       e Zd ZdgZd Zy)	FNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFNetModel.__init__@  r   r   Nr   r   r   r   r  r  =  r   r   r  c                       e Zd ZdgZd Zy)FNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFNetPreTrainedModel.__init__G  r   r   Nr   r   r   r   r  r  D  r   r   r  c                       e Zd ZdgZd Zy)FocalNetBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zFocalNetBackbone.__init__N  r   r   Nr   r   r   r   r  r  K  r   r   r  c                       e Zd ZdgZd Zy)FocalNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'FocalNetForImageClassification.__init__U  r   r   Nr   r   r   r   r  r  R  r   r   r  c                       e Zd ZdgZd Zy)FocalNetForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z'FocalNetForMaskedImageModeling.__init__\  r   r   Nr   r   r   r   r  r  Y  r   r   r  c                       e Zd ZdgZd Zy)FocalNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFocalNetModel.__init__c  r   r   Nr   r   r   r   r  r  `  r   r   r  c                       e Zd ZdgZd Zy)FocalNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z FocalNetPreTrainedModel.__init__j  r   r   Nr   r   r   r   r  r  g  r   r   r  c                       e Zd ZdgZd Zy)FSMTForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z%FSMTForConditionalGeneration.__init__q  r   r   Nr   r   r   r   r  r  n  r   r   r  c                       e Zd ZdgZd Zy)	FSMTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFSMTModel.__init__x  r   r   Nr   r   r   r   r  r  u  r   r   r  c                       e Zd ZdgZd Zy)PretrainedFSMTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPretrainedFSMTModel.__init__  r   r   Nr   r   r   r   r  r  |  r   r   r  c                       e Zd ZdgZd Zy)FunnelBaseModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFunnelBaseModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFunnelForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z FunnelForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zFunnelForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#FunnelForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(FunnelForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%FunnelForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFunnelModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FunnelPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFunnelPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_funnelr  s     r   r  r    r&  r   c                       e Zd ZdgZd Zy)FuyuForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zFuyuForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)FuyuPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFuyuPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GemmaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGemmaForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GemmaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'GemmaForSequenceClassification.__init__  r   r   Nr   r   r   r   r"  r"    r   r   r"  c                       e Zd ZdgZd Zy)GemmaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$GemmaForTokenClassification.__init__  r   r   Nr   r   r   r   r%  r%    r   r   r%  c                       e Zd ZdgZd Zy)
GemmaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGemmaModel.__init__  r   r   Nr   r   r   r   r(  r(    r   r   r(  c                       e Zd ZdgZd Zy)GemmaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGemmaPreTrainedModel.__init__  r   r   Nr   r   r   r   r+  r+    r   r   r+  c                       e Zd ZdgZd Zy)Gemma2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGemma2ForCausalLM.__init__  r   r   Nr   r   r   r   r.  r.    r   r   r.  c                       e Zd ZdgZd Zy)Gemma2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(Gemma2ForSequenceClassification.__init__  r   r   Nr   r   r   r   r1  r1    r   r   r1  c                       e Zd ZdgZd Zy)Gemma2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%Gemma2ForTokenClassification.__init__  r   r   Nr   r   r   r   r4  r4    r   r   r4  c                       e Zd ZdgZd Zy)Gemma2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zGemma2Model.__init__  r   r   Nr   r   r   r   r7  r7    r   r   r7  c                       e Zd ZdgZd Zy)Gemma2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGemma2PreTrainedModel.__init__  r   r   Nr   r   r   r   r:  r:    r   r   r:  c                       e Zd ZdgZd Zy)GitForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGitForCausalLM.__init__  r   r   Nr   r   r   r   r=  r=    r   r   r=  c                       e Zd ZdgZd Zy)GitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGitModel.__init__$  r   r   Nr   r   r   r   r@  r@  !  r   r   r@  c                       e Zd ZdgZd Zy)GitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGitPreTrainedModel.__init__+  r   r   Nr   r   r   r   rC  rC  (  r   r   rC  c                       e Zd ZdgZd Zy)GitVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGitVisionModel.__init__2  r   r   Nr   r   r   r   rF  rF  /  r   r   rF  c                       e Zd ZdgZd Zy)GlmForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGlmForCausalLM.__init__9  r   r   Nr   r   r   r   rI  rI  6  r   r   rI  c                       e Zd ZdgZd Zy)GlmForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%GlmForSequenceClassification.__init__@  r   r   Nr   r   r   r   rL  rL  =  r   r   rL  c                       e Zd ZdgZd Zy)GlmForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"GlmForTokenClassification.__init__G  r   r   Nr   r   r   r   rO  rO  D  r   r   rO  c                       e Zd ZdgZd Zy)GlmModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGlmModel.__init__N  r   r   Nr   r   r   r   rR  rR  K  r   r   rR  c                       e Zd ZdgZd Zy)GlmPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGlmPreTrainedModel.__init__U  r   r   Nr   r   r   r   rU  rU  R  r   r   rU  c                       e Zd ZdgZd Zy)GLPNForDepthEstimationr   c                     t        | dg       y r	   r
   r   s      r   r   zGLPNForDepthEstimation.__init__\  r   r   Nr   r   r   r   rX  rX  Y  r   r   rX  c                       e Zd ZdgZd Zy)	GLPNModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGLPNModel.__init__c  r   r   Nr   r   r   r   r[  r[  `  r   r   r[  c                       e Zd ZdgZd Zy)GLPNPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGLPNPreTrainedModel.__init__j  r   r   Nr   r   r   r   r^  r^  g  r   r   r^  c                       e Zd ZdgZd Zy)GPT2DoubleHeadsModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPT2DoubleHeadsModel.__init__q  r   r   Nr   r   r   r   ra  ra  n  r   r   ra  c                       e Zd ZdgZd Zy)GPT2ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!GPT2ForQuestionAnswering.__init__x  r   r   Nr   r   r   r   rd  rd  u  r   r   rd  c                       e Zd ZdgZd Zy)GPT2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&GPT2ForSequenceClassification.__init__  r   r   Nr   r   r   r   rg  rg  |  r   r   rg  c                       e Zd ZdgZd Zy)GPT2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#GPT2ForTokenClassification.__init__  r   r   Nr   r   r   r   rj  rj    r   r   rj  c                       e Zd ZdgZd Zy)GPT2LMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPT2LMHeadModel.__init__  r   r   Nr   r   r   r   rm  rm    r   r   rm  c                       e Zd ZdgZd Zy)	GPT2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPT2Model.__init__  r   r   Nr   r   r   r   rp  rp    r   r   rp  c                       e Zd ZdgZd Zy)GPT2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPT2PreTrainedModel.__init__  r   r   Nr   r   r   r   rs  rs    r   r   rs  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_gpt2r  s     r   rv  rv    r"  r   c                       e Zd ZdgZd Zy)GPTBigCodeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTBigCodeForCausalLM.__init__  r   r   Nr   r   r   r   rx  rx    r   r   rx  c                       e Zd ZdgZd Zy)#GPTBigCodeForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,GPTBigCodeForSequenceClassification.__init__  r   r   Nr   r   r   r   r{  r{    r   r   r{  c                       e Zd ZdgZd Zy) GPTBigCodeForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)GPTBigCodeForTokenClassification.__init__  r   r   Nr   r   r   r   r~  r~    r   r   r~  c                       e Zd ZdgZd Zy)GPTBigCodeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTBigCodeModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTBigCodePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"GPTBigCodePreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#GPTNeoForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(GPTNeoForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%GPTNeoForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_gpt_neor  s     r   r  r    r@  r   c                       e Zd ZdgZd Zy)GPTNeoXForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoXForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$GPTNeoXForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) GPTNeoXForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)GPTNeoXForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&GPTNeoXForTokenClassification.__init__  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoXModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoXPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXJapaneseForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z#GPTNeoXJapaneseForCausalLM.__init__!  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXJapaneseModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTNeoXJapaneseModel.__init__(  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)GPTNeoXJapanesePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'GPTNeoXJapanesePreTrainedModel.__init__/  r   r   Nr   r   r   r   r  r  ,  r   r   r  c                       e Zd ZdgZd Zy)GPTJForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTJForCausalLM.__init__6  r   r   Nr   r   r   r   r  r  3  r   r   r  c                       e Zd ZdgZd Zy)GPTJForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!GPTJForQuestionAnswering.__init__=  r   r   Nr   r   r   r   r  r  :  r   r   r  c                       e Zd ZdgZd Zy)GPTJForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&GPTJForSequenceClassification.__init__D  r   r   Nr   r   r   r   r  r  A  r   r   r  c                       e Zd ZdgZd Zy)	GPTJModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTJModel.__init__K  r   r   Nr   r   r   r   r  r  H  r   r   r  c                       e Zd ZdgZd Zy)GPTJPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGPTJPreTrainedModel.__init__R  r   r   Nr   r   r   r   r  r  O  r   r   r  c                       e Zd ZdgZd Zy)GraniteForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGraniteForCausalLM.__init__Y  r   r   Nr   r   r   r   r  r  V  r   r   r  c                       e Zd ZdgZd Zy)GraniteModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGraniteModel.__init__`  r   r   Nr   r   r   r   r  r  ]  r   r   r  c                       e Zd ZdgZd Zy)GranitePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGranitePreTrainedModel.__init__g  r   r   Nr   r   r   r   r  r  d  r   r   r  c                       e Zd ZdgZd Zy)GraniteMoeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zGraniteMoeForCausalLM.__init__n  r   r   Nr   r   r   r   r  r  k  r   r   r  c                       e Zd ZdgZd Zy)GraniteMoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGraniteMoeModel.__init__u  r   r   Nr   r   r   r   r  r  r  r   r   r  c                       e Zd ZdgZd Zy)GraniteMoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"GraniteMoePreTrainedModel.__init__|  r   r   Nr   r   r   r   r  r  y  r   r   r  c                       e Zd ZdgZd Zy)GroundingDinoForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z(GroundingDinoForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroundingDinoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGroundingDinoModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroundingDinoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%GroundingDinoPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroupViTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGroupViTModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroupViTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z GroupViTPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroupViTTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGroupViTTextModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)GroupViTVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zGroupViTVisionModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HieraBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zHieraBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HieraForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$HieraForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HieraForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zHieraForPreTraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
HieraModelr   c                     t        | dg       y r	   r
   r   s      r   r   zHieraModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HieraPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zHieraPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HubertForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zHubertForCTC.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HubertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(HubertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HubertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zHubertModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)HubertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zHubertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zIBertForMaskedLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zIBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)IBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"IBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'IBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$IBertForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
IBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIBertModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)IdeficsForVisionText2Textr   c                     t        | dg       y r	   r
   r   s      r   r   z"IdeficsForVisionText2Text.__init__$  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)IdeficsModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIdeficsModel.__init__+  r   r   Nr   r   r   r   r  r  (  r   r   r  c                       e Zd ZdgZd Zy)IdeficsPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIdeficsPreTrainedModel.__init__2  r   r   Nr   r   r   r   r"  r"  /  r   r   r"  c                       e Zd ZdgZd Zy)IdeficsProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zIdeficsProcessor.__init__9  r   r   Nr   r   r   r   r%  r%  6  r   r   r%  c                       e Zd ZdgZd Zy) Idefics2ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)Idefics2ForConditionalGeneration.__init__@  r   r   Nr   r   r   r   r(  r(  =  r   r   r(  c                       e Zd ZdgZd Zy)Idefics2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zIdefics2Model.__init__G  r   r   Nr   r   r   r   r+  r+  D  r   r   r+  c                       e Zd ZdgZd Zy)Idefics2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z Idefics2PreTrainedModel.__init__N  r   r   Nr   r   r   r   r.  r.  K  r   r   r.  c                       e Zd ZdgZd Zy)Idefics2Processorr   c                     t        | dg       y r	   r
   r   s      r   r   zIdefics2Processor.__init__U  r   r   Nr   r   r   r   r1  r1  R  r   r   r1  c                       e Zd ZdgZd Zy) Idefics3ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)Idefics3ForConditionalGeneration.__init__\  r   r   Nr   r   r   r   r4  r4  Y  r   r   r4  c                       e Zd ZdgZd Zy)Idefics3Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zIdefics3Model.__init__c  r   r   Nr   r   r   r   r7  r7  `  r   r   r7  c                       e Zd ZdgZd Zy)Idefics3PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z Idefics3PreTrainedModel.__init__j  r   r   Nr   r   r   r   r:  r:  g  r   r   r:  c                       e Zd ZdgZd Zy)Idefics3Processorr   c                     t        | dg       y r	   r
   r   s      r   r   zIdefics3Processor.__init__q  r   r   Nr   r   r   r   r=  r=  n  r   r   r=  c                       e Zd ZdgZd Zy)IJepaForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$IJepaForImageClassification.__init__x  r   r   Nr   r   r   r   r@  r@  u  r   r   r@  c                       e Zd ZdgZd Zy)
IJepaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIJepaModel.__init__  r   r   Nr   r   r   r   rC  rC  |  r   r   rC  c                       e Zd ZdgZd Zy)IJepaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zIJepaPreTrainedModel.__init__  r   r   Nr   r   r   r   rF  rF    r   r   rF  c                       e Zd ZdgZd Zy)ImageGPTForCausalImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z'ImageGPTForCausalImageModeling.__init__  r   r   Nr   r   r   r   rI  rI    r   r   rI  c                       e Zd ZdgZd Zy)ImageGPTForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ImageGPTForImageClassification.__init__  r   r   Nr   r   r   r   rL  rL    r   r   rL  c                       e Zd ZdgZd Zy)ImageGPTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zImageGPTModel.__init__  r   r   Nr   r   r   r   rO  rO    r   r   rO  c                       e Zd ZdgZd Zy)ImageGPTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z ImageGPTPreTrainedModel.__init__  r   r   Nr   r   r   r   rR  rR    r   r   rR  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_imagegptr  s     r   rU  rU    rK  r   c                       e Zd ZdgZd Zy)InformerForPredictionr   c                     t        | dg       y r	   r
   r   s      r   r   zInformerForPrediction.__init__  r   r   Nr   r   r   r   rW  rW    r   r   rW  c                       e Zd ZdgZd Zy)InformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zInformerModel.__init__  r   r   Nr   r   r   r   rZ  rZ    r   r   rZ  c                       e Zd ZdgZd Zy)InformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z InformerPreTrainedModel.__init__  r   r   Nr   r   r   r   r]  r]    r   r   r]  c                       e Zd ZdgZd Zy)$InstructBlipForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z-InstructBlipForConditionalGeneration.__init__  r   r   Nr   r   r   r   r`  r`    r   r   r`  c                       e Zd ZdgZd Zy)InstructBlipPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$InstructBlipPreTrainedModel.__init__  r   r   Nr   r   r   r   rc  rc    r   r   rc  c                       e Zd ZdgZd Zy)InstructBlipQFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!InstructBlipQFormerModel.__init__  r   r   Nr   r   r   r   rf  rf    r   r   rf  c                       e Zd ZdgZd Zy)InstructBlipVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   z InstructBlipVisionModel.__init__  r   r   Nr   r   r   r   ri  ri    r   r   ri  c                       e Zd ZdgZd Zy))InstructBlipVideoForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z2InstructBlipVideoForConditionalGeneration.__init__  r   r   Nr   r   r   r   rl  rl    r   r   rl  c                       e Zd ZdgZd Zy) InstructBlipVideoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z)InstructBlipVideoPreTrainedModel.__init__  r   r   Nr   r   r   r   ro  ro    r   r   ro  c                       e Zd ZdgZd Zy)InstructBlipVideoQFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&InstructBlipVideoQFormerModel.__init__  r   r   Nr   r   r   r   rr  rr    r   r   rr  c                       e Zd ZdgZd Zy)InstructBlipVideoVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%InstructBlipVideoVisionModel.__init__  r   r   Nr   r   r   r   ru  ru    r   r   ru  c                       e Zd ZdgZd Zy)JambaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zJambaForCausalLM.__init__  r   r   Nr   r   r   r   rx  rx    r   r   rx  c                       e Zd ZdgZd Zy)JambaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'JambaForSequenceClassification.__init__  r   r   Nr   r   r   r   r{  r{    r   r   r{  c                       e Zd ZdgZd Zy)
JambaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJambaModel.__init__  r   r   Nr   r   r   r   r~  r~    r   r   r~  c                       e Zd ZdgZd Zy)JambaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJambaPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)JetMoeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zJetMoeForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)JetMoeForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(JetMoeForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)JetMoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJetMoeModel.__init__$  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)JetMoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zJetMoePreTrainedModel.__init__+  r   r   Nr   r   r   r   r  r  (  r   r   r  c                       e Zd ZdgZd Zy)Kosmos2ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(Kosmos2ForConditionalGeneration.__init__2  r   r   Nr   r   r   r   r  r  /  r   r   r  c                       e Zd ZdgZd Zy)Kosmos2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zKosmos2Model.__init__9  r   r   Nr   r   r   r   r  r  6  r   r   r  c                       e Zd ZdgZd Zy)Kosmos2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zKosmos2PreTrainedModel.__init__@  r   r   Nr   r   r   r   r  r  =  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zLayoutLMForMaskedLM.__init__G  r   r   Nr   r   r   r   r  r  D  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%LayoutLMForQuestionAnswering.__init__N  r   r   Nr   r   r   r   r  r  K  r   r   r  c                       e Zd ZdgZd Zy)!LayoutLMForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*LayoutLMForSequenceClassification.__init__U  r   r   Nr   r   r   r   r  r  R  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'LayoutLMForTokenClassification.__init__\  r   r   Nr   r   r   r   r  r  Y  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLayoutLMModel.__init__c  r   r   Nr   r   r   r   r  r  `  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z LayoutLMPreTrainedModel.__init__j  r   r   Nr   r   r   r   r  r  g  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv2ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'LayoutLMv2ForQuestionAnswering.__init__q  r   r   Nr   r   r   r   r  r  n  r   r   r  c                       e Zd ZdgZd Zy)#LayoutLMv2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,LayoutLMv2ForSequenceClassification.__init__x  r   r   Nr   r   r   r   r  r  u  r   r   r  c                       e Zd ZdgZd Zy) LayoutLMv2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)LayoutLMv2ForTokenClassification.__init__  r   r   Nr   r   r   r   r  r  |  r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zLayoutLMv2Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"LayoutLMv2PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv3ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'LayoutLMv3ForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)#LayoutLMv3ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,LayoutLMv3ForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) LayoutLMv3ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)LayoutLMv3ForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv3Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zLayoutLMv3Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LayoutLMv3PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"LayoutLMv3PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LEDForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z$LEDForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LEDForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z LEDForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LEDForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%LEDForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LEDModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLEDModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LEDPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLEDPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LevitForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$LevitForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)&LevitForImageClassificationWithTeacherr   c                     t        | dg       y r	   r
   r   s      r   r   z/LevitForImageClassificationWithTeacher.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
LevitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLevitModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LevitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLevitPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LiltForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!LiltForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LiltForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&LiltForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LiltForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#LiltForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	LiltModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLiltModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LiltPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLiltPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LlamaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zLlamaForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LlamaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"LlamaForQuestionAnswering.__init__   r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)LlamaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'LlamaForSequenceClassification.__init__'  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)LlamaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$LlamaForTokenClassification.__init__.  r   r   Nr   r   r   r   r  r  +  r   r   r  c                       e Zd ZdgZd Zy)
LlamaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLlamaModel.__init__5  r   r   Nr   r   r   r   r  r  2  r   r   r  c                       e Zd ZdgZd Zy)LlamaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLlamaPreTrainedModel.__init__<  r   r   Nr   r   r   r   r	  r	  9  r   r   r	  c                       e Zd ZdgZd Zy)LlavaForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z&LlavaForConditionalGeneration.__init__C  r   r   Nr   r   r   r   r	  r	  @  r   r   r	  c                       e Zd ZdgZd Zy)LlavaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLlavaPreTrainedModel.__init__J  r   r   Nr   r   r   r   r	  r	  G  r   r   r	  c                       e Zd ZdgZd Zy)!LlavaNextForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*LlavaNextForConditionalGeneration.__init__Q  r   r   Nr   r   r   r   r	  r	  N  r   r   r	  c                       e Zd ZdgZd Zy)LlavaNextPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!LlavaNextPreTrainedModel.__init__X  r   r   Nr   r   r   r   r	  r	  U  r   r   r	  c                       e Zd ZdgZd Zy)&LlavaNextVideoForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/LlavaNextVideoForConditionalGeneration.__init___  r   r   Nr   r   r   r   r	  r	  \  r   r   r	  c                       e Zd ZdgZd Zy)LlavaNextVideoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&LlavaNextVideoPreTrainedModel.__init__f  r   r   Nr   r   r   r   r	  r	  c  r   r   r	  c                       e Zd ZdgZd Zy)&LlavaOnevisionForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/LlavaOnevisionForConditionalGeneration.__init__m  r   r   Nr   r   r   r   r	  r	  j  r   r   r	  c                       e Zd ZdgZd Zy)LlavaOnevisionPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&LlavaOnevisionPreTrainedModel.__init__t  r   r   Nr   r   r   r   r	  r	  q  r   r   r	  c                       e Zd ZdgZd Zy)LongformerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zLongformerForMaskedLM.__init__{  r   r   Nr   r   r   r   r	  r	  x  r   r   r	  c                       e Zd ZdgZd Zy)LongformerForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$LongformerForMultipleChoice.__init__  r   r   Nr   r   r   r   r 	  r 	    r   r   r 	  c                       e Zd ZdgZd Zy)LongformerForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'LongformerForQuestionAnswering.__init__  r   r   Nr   r   r   r   r#	  r#	    r   r   r#	  c                       e Zd ZdgZd Zy)#LongformerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,LongformerForSequenceClassification.__init__  r   r   Nr   r   r   r   r&	  r&	    r   r   r&	  c                       e Zd ZdgZd Zy) LongformerForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)LongformerForTokenClassification.__init__  r   r   Nr   r   r   r   r)	  r)	    r   r   r)	  c                       e Zd ZdgZd Zy)LongformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLongformerModel.__init__  r   r   Nr   r   r   r   r,	  r,	    r   r   r,	  c                       e Zd ZdgZd Zy)LongformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"LongformerPreTrainedModel.__init__  r   r   Nr   r   r   r   r/	  r/	    r   r   r/	  c                       e Zd ZdgZd Zy)LongT5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLongT5EncoderModel.__init__  r   r   Nr   r   r   r   r2	  r2	    r   r   r2	  c                       e Zd ZdgZd Zy)LongT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z'LongT5ForConditionalGeneration.__init__  r   r   Nr   r   r   r   r5	  r5	    r   r   r5	  c                       e Zd ZdgZd Zy)LongT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zLongT5Model.__init__  r   r   Nr   r   r   r   r8	  r8	    r   r   r8	  c                       e Zd ZdgZd Zy)LongT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLongT5PreTrainedModel.__init__  r   r   Nr   r   r   r   r;	  r;	    r   r   r;	  c                       e Zd ZdgZd Zy)LukeForEntityClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$LukeForEntityClassification.__init__  r   r   Nr   r   r   r   r>	  r>	    r   r   r>	  c                       e Zd ZdgZd Zy)LukeForEntityPairClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(LukeForEntityPairClassification.__init__  r   r   Nr   r   r   r   rA	  rA	    r   r   rA	  c                       e Zd ZdgZd Zy)LukeForEntitySpanClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(LukeForEntitySpanClassification.__init__  r   r   Nr   r   r   r   rD	  rD	    r   r   rD	  c                       e Zd ZdgZd Zy)LukeForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zLukeForMaskedLM.__init__  r   r   Nr   r   r   r   rG	  rG	    r   r   rG	  c                       e Zd ZdgZd Zy)LukeForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zLukeForMultipleChoice.__init__  r   r   Nr   r   r   r   rJ	  rJ	    r   r   rJ	  c                       e Zd ZdgZd Zy)LukeForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!LukeForQuestionAnswering.__init__  r   r   Nr   r   r   r   rM	  rM	    r   r   rM	  c                       e Zd ZdgZd Zy)LukeForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&LukeForSequenceClassification.__init__  r   r   Nr   r   r   r   rP	  rP	    r   r   rP	  c                       e Zd ZdgZd Zy)LukeForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#LukeForTokenClassification.__init__  r   r   Nr   r   r   r   rS	  rS	    r   r   rS	  c                       e Zd ZdgZd Zy)	LukeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLukeModel.__init__   r   r   Nr   r   r   r   rV	  rV	    r   r   rV	  c                       e Zd ZdgZd Zy)LukePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLukePreTrainedModel.__init__  r   r   Nr   r   r   r   rY	  rY	    r   r   rY	  c                       e Zd ZdgZd Zy)LxmertEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zLxmertEncoder.__init__  r   r   Nr   r   r   r   r\	  r\	    r   r   r\	  c                       e Zd ZdgZd Zy)LxmertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zLxmertForPreTraining.__init__  r   r   Nr   r   r   r   r_	  r_	    r   r   r_	  c                       e Zd ZdgZd Zy)LxmertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z#LxmertForQuestionAnswering.__init__  r   r   Nr   r   r   r   rb	  rb	    r   r   rb	  c                       e Zd ZdgZd Zy)LxmertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLxmertModel.__init__#  r   r   Nr   r   r   r   re	  re	     r   r   re	  c                       e Zd ZdgZd Zy)LxmertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLxmertPreTrainedModel.__init__*  r   r   Nr   r   r   r   rh	  rh	  '  r   r   rh	  c                       e Zd ZdgZd Zy)LxmertVisualFeatureEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   z#LxmertVisualFeatureEncoder.__init__1  r   r   Nr   r   r   r   rk	  rk	  .  r   r   rk	  c                       e Zd ZdgZd Zy)M2M100ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z'M2M100ForConditionalGeneration.__init__8  r   r   Nr   r   r   r   rn	  rn	  5  r   r   rn	  c                       e Zd ZdgZd Zy)M2M100Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zM2M100Model.__init__?  r   r   Nr   r   r   r   rq	  rq	  <  r   r   rq	  c                       e Zd ZdgZd Zy)M2M100PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zM2M100PreTrainedModel.__init__F  r   r   Nr   r   r   r   rt	  rt	  C  r   r   rt	  c                       e Zd ZdgZd Zy)MambaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMambaForCausalLM.__init__M  r   r   Nr   r   r   r   rw	  rw	  J  r   r   rw	  c                       e Zd ZdgZd Zy)
MambaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMambaModel.__init__T  r   r   Nr   r   r   r   rz	  rz	  Q  r   r   rz	  c                       e Zd ZdgZd Zy)MambaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMambaPreTrainedModel.__init__[  r   r   Nr   r   r   r   r}	  r}	  X  r   r   r}	  c                       e Zd ZdgZd Zy)Mamba2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMamba2ForCausalLM.__init__b  r   r   Nr   r   r   r   r	  r	  _  r   r   r	  c                       e Zd ZdgZd Zy)Mamba2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zMamba2Model.__init__i  r   r   Nr   r   r   r   r	  r	  f  r   r   r	  c                       e Zd ZdgZd Zy)Mamba2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMamba2PreTrainedModel.__init__p  r   r   Nr   r   r   r   r	  r	  m  r   r   r	  c                       e Zd ZdgZd Zy)MarianForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMarianForCausalLM.__init__w  r   r   Nr   r   r   r   r	  r	  t  r   r   r	  c                       e Zd ZdgZd Zy)MarianModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMarianModel.__init__~  r   r   Nr   r   r   r   r	  r	  {  r   r   r	  c                       e Zd ZdgZd Zy)MarianMTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMarianMTModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MarianPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMarianPreTrainedModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MarkupLMForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%MarkupLMForQuestionAnswering.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)!MarkupLMForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*MarkupLMForSequenceClassification.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MarkupLMForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'MarkupLMForTokenClassification.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MarkupLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMarkupLMModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MarkupLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z MarkupLMPreTrainedModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)#Mask2FormerForUniversalSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z,Mask2FormerForUniversalSegmentation.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)Mask2FormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMask2FormerModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)Mask2FormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#Mask2FormerPreTrainedModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)!MaskFormerForInstanceSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z*MaskFormerForInstanceSegmentation.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MaskFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMaskFormerModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MaskFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"MaskFormerPreTrainedModel.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MaskFormerSwinBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zMaskFormerSwinBackbone.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MBartForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMBartForCausalLM.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MBartForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z&MBartForConditionalGeneration.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MBartForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"MBartForQuestionAnswering.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MBartForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'MBartForSequenceClassification.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)
MBartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMBartModel.__init__  r   r   Nr   r   r   r   r	  r	     r   r   r	  c                       e Zd ZdgZd Zy)MBartPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMBartPreTrainedModel.__init__
  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z MegatronBertForCausalLM.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z MegatronBertForMaskedLM.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z&MegatronBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)%MegatronBertForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z.MegatronBertForNextSentencePrediction.__init__&  r   r   Nr   r   r   r   r	  r	  #  r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z#MegatronBertForPreTraining.__init__-  r   r   Nr   r   r   r   r	  r	  *  r   r   r	  c                       e Zd ZdgZd Zy) MegatronBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z)MegatronBertForQuestionAnswering.__init__4  r   r   Nr   r   r   r   r	  r	  1  r   r   r	  c                       e Zd ZdgZd Zy)%MegatronBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.MegatronBertForSequenceClassification.__init__;  r   r   Nr   r   r   r   r	  r	  8  r   r   r	  c                       e Zd ZdgZd Zy)"MegatronBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+MegatronBertForTokenClassification.__init__B  r   r   Nr   r   r   r   r	  r	  ?  r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMegatronBertModel.__init__I  r   r   Nr   r   r   r   r	  r	  F  r   r   r	  c                       e Zd ZdgZd Zy)MegatronBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$MegatronBertPreTrainedModel.__init__P  r   r   Nr   r   r   r   r	  r	  M  r   r   r	  c                       e Zd ZdgZd Zy)MgpstrForSceneTextRecognitionr   c                     t        | dg       y r	   r
   r   s      r   r   z&MgpstrForSceneTextRecognition.__init__W  r   r   Nr   r   r   r   r	  r	  T  r   r   r	  c                       e Zd ZdgZd Zy)MgpstrModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMgpstrModel.__init__^  r   r   Nr   r   r   r   r	  r	  [  r   r   r	  c                       e Zd ZdgZd Zy)MgpstrPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMgpstrPreTrainedModel.__init__e  r   r   Nr   r   r   r   r	  r	  b  r   r   r	  c                       e Zd ZdgZd Zy)	MimiModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMimiModel.__init__l  r   r   Nr   r   r   r   r	  r	  i  r   r   r	  c                       e Zd ZdgZd Zy)MimiPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMimiPreTrainedModel.__init__s  r   r   Nr   r   r   r   r	  r	  p  r   r   r	  c                       e Zd ZdgZd Zy)MistralForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMistralForCausalLM.__init__z  r   r   Nr   r   r   r   r	  r	  w  r   r   r	  c                       e Zd ZdgZd Zy)MistralForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$MistralForQuestionAnswering.__init__  r   r   Nr   r   r   r   r	  r	  ~  r   r   r	  c                       e Zd ZdgZd Zy) MistralForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)MistralForSequenceClassification.__init__  r   r   Nr   r   r   r   r	  r	    r   r   r	  c                       e Zd ZdgZd Zy)MistralForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&MistralForTokenClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MistralModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMistralModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MistralPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMistralPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MixtralForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMixtralForCausalLM.__init__  r   r   Nr   r   r   r   r

  r

    r   r   r

  c                       e Zd ZdgZd Zy)MixtralForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$MixtralForQuestionAnswering.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy) MixtralForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)MixtralForSequenceClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MixtralForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&MixtralForTokenClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MixtralModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMixtralModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MixtralPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMixtralPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MllamaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMllamaForCausalLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MllamaForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z'MllamaForConditionalGeneration.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MllamaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMllamaPreTrainedModel.__init__  r   r   Nr   r   r   r   r"
  r"
    r   r   r"
  c                       e Zd ZdgZd Zy)MllamaProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zMllamaProcessor.__init__  r   r   Nr   r   r   r   r%
  r%
    r   r   r%
  c                       e Zd ZdgZd Zy)MllamaTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMllamaTextModel.__init__  r   r   Nr   r   r   r   r(
  r(
    r   r   r(
  c                       e Zd ZdgZd Zy)MllamaVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMllamaVisionModel.__init__  r   r   Nr   r   r   r   r+
  r+
    r   r   r+
  c                       e Zd ZdgZd Zy)MobileBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileBertForMaskedLM.__init__  r   r   Nr   r   r   r   r.
  r.
    r   r   r.
  c                       e Zd ZdgZd Zy)MobileBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$MobileBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r1
  r1
    r   r   r1
  c                       e Zd ZdgZd Zy)#MobileBertForNextSentencePredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z,MobileBertForNextSentencePrediction.__init__  r   r   Nr   r   r   r   r4
  r4
    r   r   r4
  c                       e Zd ZdgZd Zy)MobileBertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z!MobileBertForPreTraining.__init__  r   r   Nr   r   r   r   r7
  r7
  
  r   r   r7
  c                       e Zd ZdgZd Zy)MobileBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'MobileBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r:
  r:
    r   r   r:
  c                       e Zd ZdgZd Zy)#MobileBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,MobileBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r=
  r=
    r   r   r=
  c                       e Zd ZdgZd Zy) MobileBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)MobileBertForTokenClassification.__init__"  r   r   Nr   r   r   r   r@
  r@
    r   r   r@
  c                       e Zd ZdgZd Zy)MobileBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileBertModel.__init__)  r   r   Nr   r   r   r   rC
  rC
  &  r   r   rC
  c                       e Zd ZdgZd Zy)MobileBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"MobileBertPreTrainedModel.__init__0  r   r   Nr   r   r   r   rF
  rF
  -  r   r   rF
  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_mobilebertr  s     r   rI
  rI
  4  r  r   c                       e Zd ZdgZd Zy)!MobileNetV1ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*MobileNetV1ForImageClassification.__init__;  r   r   Nr   r   r   r   rK
  rK
  8  r   r   rK
  c                       e Zd ZdgZd Zy)MobileNetV1Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileNetV1Model.__init__B  r   r   Nr   r   r   r   rN
  rN
  ?  r   r   rN
  c                       e Zd ZdgZd Zy)MobileNetV1PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#MobileNetV1PreTrainedModel.__init__I  r   r   Nr   r   r   r   rQ
  rQ
  F  r   r   rQ
  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_mobilenet_v1r  s     r   rT
  rT
  M      5yAr   c                       e Zd ZdgZd Zy)!MobileNetV2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*MobileNetV2ForImageClassification.__init__T  r   r   Nr   r   r   r   rW
  rW
  Q  r   r   rW
  c                       e Zd ZdgZd Zy)"MobileNetV2ForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z+MobileNetV2ForSemanticSegmentation.__init__[  r   r   Nr   r   r   r   rZ
  rZ
  X  r   r   rZ
  c                       e Zd ZdgZd Zy)MobileNetV2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileNetV2Model.__init__b  r   r   Nr   r   r   r   r]
  r]
  _  r   r   r]
  c                       e Zd ZdgZd Zy)MobileNetV2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#MobileNetV2PreTrainedModel.__init__i  r   r   Nr   r   r   r   r`
  r`
  f  r   r   r`
  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_mobilenet_v2r  s     r   rc
  rc
  m  rU
  r   c                       e Zd ZdgZd Zy)MobileViTForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(MobileViTForImageClassification.__init__t  r   r   Nr   r   r   r   re
  re
  q  r   r   re
  c                       e Zd ZdgZd Zy) MobileViTForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z)MobileViTForSemanticSegmentation.__init__{  r   r   Nr   r   r   r   rh
  rh
  x  r   r   rh
  c                       e Zd ZdgZd Zy)MobileViTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileViTModel.__init__  r   r   Nr   r   r   r   rk
  rk
    r   r   rk
  c                       e Zd ZdgZd Zy)MobileViTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!MobileViTPreTrainedModel.__init__  r   r   Nr   r   r   r   rn
  rn
    r   r   rn
  c                       e Zd ZdgZd Zy)!MobileViTV2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*MobileViTV2ForImageClassification.__init__  r   r   Nr   r   r   r   rq
  rq
    r   r   rq
  c                       e Zd ZdgZd Zy)"MobileViTV2ForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z+MobileViTV2ForSemanticSegmentation.__init__  r   r   Nr   r   r   r   rt
  rt
    r   r   rt
  c                       e Zd ZdgZd Zy)MobileViTV2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zMobileViTV2Model.__init__  r   r   Nr   r   r   r   rw
  rw
    r   r   rw
  c                       e Zd ZdgZd Zy)MobileViTV2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#MobileViTV2PreTrainedModel.__init__  r   r   Nr   r   r   r   rz
  rz
    r   r   rz
  c                       e Zd ZdgZd Zy)MoshiForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMoshiForCausalLM.__init__  r   r   Nr   r   r   r   r}
  r}
    r   r   r}
  c                       e Zd ZdgZd Zy)MoshiForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z&MoshiForConditionalGeneration.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)
MoshiModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMoshiModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MoshiPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMoshiPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMPNetForMaskedLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zMPNetForMultipleChoice.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"MPNetForQuestionAnswering.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'MPNetForSequenceClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$MPNetForTokenClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)
MPNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMPNetModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MPNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMPNetPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMptForCausalLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z MptForQuestionAnswering.__init__   r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%MptForSequenceClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"MptForTokenClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMptModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MptPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMptPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MraForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMraForMaskedLM.__init__#  r   r   Nr   r   r   r   r
  r
     r   r   r
  c                       e Zd ZdgZd Zy)MraForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zMraForMultipleChoice.__init__*  r   r   Nr   r   r   r   r
  r
  '  r   r   r
  c                       e Zd ZdgZd Zy)MraForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z MraForQuestionAnswering.__init__1  r   r   Nr   r   r   r   r
  r
  .  r   r   r
  c                       e Zd ZdgZd Zy)MraForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%MraForSequenceClassification.__init__8  r   r   Nr   r   r   r   r
  r
  5  r   r   r
  c                       e Zd ZdgZd Zy)MraForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"MraForTokenClassification.__init__?  r   r   Nr   r   r   r   r
  r
  <  r   r   r
  c                       e Zd ZdgZd Zy)MraModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMraModel.__init__F  r   r   Nr   r   r   r   r
  r
  C  r   r   r
  c                       e Zd ZdgZd Zy)MraPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMraPreTrainedModel.__init__M  r   r   Nr   r   r   r   r
  r
  J  r   r   r
  c                       e Zd ZdgZd Zy)MT5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMT5EncoderModel.__init__T  r   r   Nr   r   r   r   r
  r
  Q  r   r   r
  c                       e Zd ZdgZd Zy)MT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z$MT5ForConditionalGeneration.__init__[  r   r   Nr   r   r   r   r
  r
  X  r   r   r
  c                       e Zd ZdgZd Zy)MT5ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z MT5ForQuestionAnswering.__init__b  r   r   Nr   r   r   r   r
  r
  _  r   r   r
  c                       e Zd ZdgZd Zy)MT5ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%MT5ForSequenceClassification.__init__i  r   r   Nr   r   r   r   r
  r
  f  r   r   r
  c                       e Zd ZdgZd Zy)MT5ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"MT5ForTokenClassification.__init__p  r   r   Nr   r   r   r   r
  r
  m  r   r   r
  c                       e Zd ZdgZd Zy)MT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zMT5Model.__init__w  r   r   Nr   r   r   r   r
  r
  t  r   r   r
  c                       e Zd ZdgZd Zy)MT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMT5PreTrainedModel.__init__~  r   r   Nr   r   r   r   r
  r
  {  r   r   r
  c                       e Zd ZdgZd Zy)MusicgenForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMusicgenForCausalLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy) MusicgenForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)MusicgenForConditionalGeneration.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMusicgenModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z MusicgenPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zMusicgenProcessor.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenMelodyForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"MusicgenMelodyForCausalLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)&MusicgenMelodyForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z/MusicgenMelodyForConditionalGeneration.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenMelodyModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMusicgenMelodyModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MusicgenMelodyPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&MusicgenMelodyPreTrainedModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MvpForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zMvpForCausalLM.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MvpForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z$MvpForConditionalGeneration.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MvpForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z MvpForQuestionAnswering.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MvpForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%MvpForSequenceClassification.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)MvpModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMvpModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)MvpPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMvpPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NemotronForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zNemotronForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NemotronForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%NemotronForQuestionAnswering.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)!NemotronForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*NemotronForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NemotronForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'NemotronForTokenClassification.__init__  r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)NemotronModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNemotronModel.__init__
  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NemotronPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z NemotronPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NllbMoeForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(NllbMoeForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NllbMoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNllbMoeModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)NllbMoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNllbMoePreTrainedModel.__init__&  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)NllbMoeSparseMLPr   c                     t        | dg       y r	   r
   r   s      r   r   zNllbMoeSparseMLP.__init__-  r   r   Nr   r   r   r   r"  r"  *  r   r   r"  c                       e Zd ZdgZd Zy)NllbMoeTop2Routerr   c                     t        | dg       y r	   r
   r   s      r   r   zNllbMoeTop2Router.__init__4  r   r   Nr   r   r   r   r%  r%  1  r   r   r%  c                       e Zd ZdgZd Zy)NystromformerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z!NystromformerForMaskedLM.__init__;  r   r   Nr   r   r   r   r(  r(  8  r   r   r(  c                       e Zd ZdgZd Zy)NystromformerForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z'NystromformerForMultipleChoice.__init__B  r   r   Nr   r   r   r   r+  r+  ?  r   r   r+  c                       e Zd ZdgZd Zy)!NystromformerForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z*NystromformerForQuestionAnswering.__init__I  r   r   Nr   r   r   r   r.  r.  F  r   r   r.  c                       e Zd ZdgZd Zy)&NystromformerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z/NystromformerForSequenceClassification.__init__P  r   r   Nr   r   r   r   r1  r1  M  r   r   r1  c                       e Zd ZdgZd Zy)#NystromformerForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,NystromformerForTokenClassification.__init__W  r   r   Nr   r   r   r   r4  r4  T  r   r   r4  c                       e Zd ZdgZd Zy)NystromformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zNystromformerModel.__init__^  r   r   Nr   r   r   r   r7  r7  [  r   r   r7  c                       e Zd ZdgZd Zy)NystromformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%NystromformerPreTrainedModel.__init__e  r   r   Nr   r   r   r   r:  r:  b  r   r   r:  c                       e Zd ZdgZd Zy)OlmoForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoForCausalLM.__init__l  r   r   Nr   r   r   r   r=  r=  i  r   r   r=  c                       e Zd ZdgZd Zy)	OlmoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoModel.__init__s  r   r   Nr   r   r   r   r@  r@  p  r   r   r@  c                       e Zd ZdgZd Zy)OlmoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoPreTrainedModel.__init__z  r   r   Nr   r   r   r   rC  rC  w  r   r   rC  c                       e Zd ZdgZd Zy)Olmo2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmo2ForCausalLM.__init__  r   r   Nr   r   r   r   rF  rF  ~  r   r   rF  c                       e Zd ZdgZd Zy)
Olmo2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmo2Model.__init__  r   r   Nr   r   r   r   rI  rI    r   r   rI  c                       e Zd ZdgZd Zy)Olmo2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmo2PreTrainedModel.__init__  r   r   Nr   r   r   r   rL  rL    r   r   rL  c                       e Zd ZdgZd Zy)OlmoeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoeForCausalLM.__init__  r   r   Nr   r   r   r   rO  rO    r   r   rO  c                       e Zd ZdgZd Zy)
OlmoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoeModel.__init__  r   r   Nr   r   r   r   rR  rR    r   r   rR  c                       e Zd ZdgZd Zy)OlmoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOlmoePreTrainedModel.__init__  r   r   Nr   r   r   r   rU  rU    r   r   rU  c                       e Zd ZdgZd Zy)OmDetTurboForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z%OmDetTurboForObjectDetection.__init__  r   r   Nr   r   r   r   rX  rX    r   r   rX  c                       e Zd ZdgZd Zy)OmDetTurboPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"OmDetTurboPreTrainedModel.__init__  r   r   Nr   r   r   r   r[  r[    r   r   r[  c                       e Zd ZdgZd Zy)!OneFormerForUniversalSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z*OneFormerForUniversalSegmentation.__init__  r   r   Nr   r   r   r   r^  r^    r   r   r^  c                       e Zd ZdgZd Zy)OneFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOneFormerModel.__init__  r   r   Nr   r   r   r   ra  ra    r   r   ra  c                       e Zd ZdgZd Zy)OneFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!OneFormerPreTrainedModel.__init__  r   r   Nr   r   r   r   rd  rd    r   r   rd  c                       e Zd ZdgZd Zy)OpenAIGPTDoubleHeadsModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"OpenAIGPTDoubleHeadsModel.__init__  r   r   Nr   r   r   r   rg  rg    r   r   rg  c                       e Zd ZdgZd Zy)"OpenAIGPTForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+OpenAIGPTForSequenceClassification.__init__  r   r   Nr   r   r   r   rj  rj    r   r   rj  c                       e Zd ZdgZd Zy)OpenAIGPTLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOpenAIGPTLMHeadModel.__init__  r   r   Nr   r   r   r   rm  rm    r   r   rm  c                       e Zd ZdgZd Zy)OpenAIGPTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOpenAIGPTModel.__init__  r   r   Nr   r   r   r   rp  rp    r   r   rp  c                       e Zd ZdgZd Zy)OpenAIGPTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!OpenAIGPTPreTrainedModel.__init__  r   r   Nr   r   r   r   rs  rs    r   r   rs  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_openai_gptr  s     r   rv  rv    r  r   c                       e Zd ZdgZd Zy)OPTForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zOPTForCausalLM.__init__  r   r   Nr   r   r   r   rx  rx    r   r   rx  c                       e Zd ZdgZd Zy)OPTForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z OPTForQuestionAnswering.__init__  r   r   Nr   r   r   r   r{  r{    r   r   r{  c                       e Zd ZdgZd Zy)OPTForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%OPTForSequenceClassification.__init__  r   r   Nr   r   r   r   r~  r~     r   r   r~  c                       e Zd ZdgZd Zy)OPTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOPTModel.__init__
  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)OPTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOPTPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Owlv2ForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z Owlv2ForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
Owlv2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlv2Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Owlv2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlv2PreTrainedModel.__init__&  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)Owlv2TextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlv2TextModel.__init__-  r   r   Nr   r   r   r   r  r  *  r   r   r  c                       e Zd ZdgZd Zy)Owlv2VisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlv2VisionModel.__init__4  r   r   Nr   r   r   r   r  r  1  r   r   r  c                       e Zd ZdgZd Zy)OwlViTForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z!OwlViTForObjectDetection.__init__;  r   r   Nr   r   r   r   r  r  8  r   r   r  c                       e Zd ZdgZd Zy)OwlViTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlViTModel.__init__B  r   r   Nr   r   r   r   r  r  ?  r   r   r  c                       e Zd ZdgZd Zy)OwlViTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlViTPreTrainedModel.__init__I  r   r   Nr   r   r   r   r  r  F  r   r   r  c                       e Zd ZdgZd Zy)OwlViTTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlViTTextModel.__init__P  r   r   Nr   r   r   r   r  r  M  r   r   r  c                       e Zd ZdgZd Zy)OwlViTVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zOwlViTVisionModel.__init__W  r   r   Nr   r   r   r   r  r  T  r   r   r  c                       e Zd ZdgZd Zy)!PaliGemmaForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*PaliGemmaForConditionalGeneration.__init__^  r   r   Nr   r   r   r   r  r  [  r   r   r  c                       e Zd ZdgZd Zy)PaliGemmaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!PaliGemmaPreTrainedModel.__init__e  r   r   Nr   r   r   r   r  r  b  r   r   r  c                       e Zd ZdgZd Zy)PaliGemmaProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zPaliGemmaProcessor.__init__l  r   r   Nr   r   r   r   r  r  i  r   r   r  c                       e Zd ZdgZd Zy)PatchTSMixerForPredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z"PatchTSMixerForPrediction.__init__s  r   r   Nr   r   r   r   r  r  p  r   r   r  c                       e Zd ZdgZd Zy)PatchTSMixerForPretrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z#PatchTSMixerForPretraining.__init__z  r   r   Nr   r   r   r   r  r  w  r   r   r  c                       e Zd ZdgZd Zy)PatchTSMixerForRegressionr   c                     t        | dg       y r	   r
   r   s      r   r   z"PatchTSMixerForRegression.__init__  r   r   Nr   r   r   r   r  r  ~  r   r   r  c                       e Zd ZdgZd Zy)'PatchTSMixerForTimeSeriesClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0PatchTSMixerForTimeSeriesClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSMixerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPatchTSMixerModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSMixerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$PatchTSMixerPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTForClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"PatchTSTForClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTForPredictionr   c                     t        | dg       y r	   r
   r   s      r   r   zPatchTSTForPrediction.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTForPretrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zPatchTSTForPretraining.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTForRegressionr   c                     t        | dg       y r	   r
   r   s      r   r   zPatchTSTForRegression.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPatchTSTModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PatchTSTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z PatchTSTPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPegasusForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(PegasusForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPegasusModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPegasusPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) PegasusXForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)PegasusXForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusXModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPegasusXModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PegasusXPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z PegasusXPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)-PerceiverForImageClassificationConvProcessingr   c                     t        | dg       y r	   r
   r   s      r   r   z6PerceiverForImageClassificationConvProcessing.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)&PerceiverForImageClassificationFourierr   c                     t        | dg       y r	   r
   r   s      r   r   z/PerceiverForImageClassificationFourier.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)&PerceiverForImageClassificationLearnedr   c                     t        | dg       y r	   r
   r   s      r   r   z/PerceiverForImageClassificationLearned.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PerceiverForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPerceiverForMaskedLM.__init__  r   r   Nr   r   r   r   r  r  
  r   r   r  c                       e Zd ZdgZd Zy)"PerceiverForMultimodalAutoencodingr   c                     t        | dg       y r	   r
   r   s      r   r   z+PerceiverForMultimodalAutoencoding.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PerceiverForOpticalFlowr   c                     t        | dg       y r	   r
   r   s      r   r   z PerceiverForOpticalFlow.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"PerceiverForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+PerceiverForSequenceClassification.__init__"  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)PerceiverModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPerceiverModel.__init__)  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)PerceiverPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!PerceiverPreTrainedModel.__init__0  r   r   Nr   r   r   r   r  r  -  r   r   r  c                       e Zd ZdgZd Zy)PersimmonForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPersimmonForCausalLM.__init__7  r   r   Nr   r   r   r   r  r  4  r   r   r  c                       e Zd ZdgZd Zy)"PersimmonForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+PersimmonForSequenceClassification.__init__>  r   r   Nr   r   r   r   r  r  ;  r   r   r  c                       e Zd ZdgZd Zy)PersimmonForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(PersimmonForTokenClassification.__init__E  r   r   Nr   r   r   r   r  r  B  r   r   r  c                       e Zd ZdgZd Zy)PersimmonModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPersimmonModel.__init__L  r   r   Nr   r   r   r   r  r  I  r   r   r  c                       e Zd ZdgZd Zy)PersimmonPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!PersimmonPreTrainedModel.__init__S  r   r   Nr   r   r   r   r  r  P  r   r   r  c                       e Zd ZdgZd Zy)PhiForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPhiForCausalLM.__init__Z  r   r   Nr   r   r   r   r  r  W  r   r   r  c                       e Zd ZdgZd Zy)PhiForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%PhiForSequenceClassification.__init__a  r   r   Nr   r   r   r   r  r  ^  r   r   r  c                       e Zd ZdgZd Zy)PhiForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"PhiForTokenClassification.__init__h  r   r   Nr   r   r   r   r  r  e  r   r   r  c                       e Zd ZdgZd Zy)PhiModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhiModel.__init__o  r   r   Nr   r   r   r   r  r  l  r   r   r  c                       e Zd ZdgZd Zy)PhiPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhiPreTrainedModel.__init__v  r   r   Nr   r   r   r   r  r  s  r   r   r  c                       e Zd ZdgZd Zy)Phi3ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPhi3ForCausalLM.__init__}  r   r   Nr   r   r   r   r   r   z  r   r   r   c                       e Zd ZdgZd Zy)Phi3ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&Phi3ForSequenceClassification.__init__  r   r   Nr   r   r   r   r#  r#    r   r   r#  c                       e Zd ZdgZd Zy)Phi3ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#Phi3ForTokenClassification.__init__  r   r   Nr   r   r   r   r&  r&    r   r   r&  c                       e Zd ZdgZd Zy)	Phi3Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhi3Model.__init__  r   r   Nr   r   r   r   r)  r)    r   r   r)  c                       e Zd ZdgZd Zy)Phi3PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhi3PreTrainedModel.__init__  r   r   Nr   r   r   r   r,  r,    r   r   r,  c                       e Zd ZdgZd Zy)PhimoeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPhimoeForCausalLM.__init__  r   r   Nr   r   r   r   r/  r/    r   r   r/  c                       e Zd ZdgZd Zy)PhimoeForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(PhimoeForSequenceClassification.__init__  r   r   Nr   r   r   r   r2  r2    r   r   r2  c                       e Zd ZdgZd Zy)PhimoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhimoeModel.__init__  r   r   Nr   r   r   r   r5  r5    r   r   r5  c                       e Zd ZdgZd Zy)PhimoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPhimoePreTrainedModel.__init__  r   r   Nr   r   r   r   r8  r8    r   r   r8  c                       e Zd ZdgZd Zy)"Pix2StructForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+Pix2StructForConditionalGeneration.__init__  r   r   Nr   r   r   r   r;  r;    r   r   r;  c                       e Zd ZdgZd Zy)Pix2StructPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"Pix2StructPreTrainedModel.__init__  r   r   Nr   r   r   r   r>  r>    r   r   r>  c                       e Zd ZdgZd Zy)Pix2StructTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPix2StructTextModel.__init__  r   r   Nr   r   r   r   rA  rA    r   r   rA  c                       e Zd ZdgZd Zy)Pix2StructVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPix2StructVisionModel.__init__  r   r   Nr   r   r   r   rD  rD    r   r   rD  c                       e Zd ZdgZd Zy)PixtralPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPixtralPreTrainedModel.__init__  r   r   Nr   r   r   r   rG  rG    r   r   rG  c                       e Zd ZdgZd Zy)PixtralVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPixtralVisionModel.__init__  r   r   Nr   r   r   r   rJ  rJ    r   r   rJ  c                       e Zd ZdgZd Zy)PLBartForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zPLBartForCausalLM.__init__  r   r   Nr   r   r   r   rM  rM    r   r   rM  c                       e Zd ZdgZd Zy)PLBartForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z'PLBartForConditionalGeneration.__init__  r   r   Nr   r   r   r   rP  rP    r   r   rP  c                       e Zd ZdgZd Zy)PLBartForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(PLBartForSequenceClassification.__init__  r   r   Nr   r   r   r   rS  rS    r   r   rS  c                       e Zd ZdgZd Zy)PLBartModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPLBartModel.__init__  r   r   Nr   r   r   r   rV  rV    r   r   rV  c                       e Zd ZdgZd Zy)PLBartPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPLBartPreTrainedModel.__init__  r   r   Nr   r   r   r   rY  rY    r   r   rY  c                       e Zd ZdgZd Zy) PoolFormerForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)PoolFormerForImageClassification.__init__	  r   r   Nr   r   r   r   r\  r\    r   r   r\  c                       e Zd ZdgZd Zy)PoolFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPoolFormerModel.__init__  r   r   Nr   r   r   r   r_  r_    r   r   r_  c                       e Zd ZdgZd Zy)PoolFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"PoolFormerPreTrainedModel.__init__  r   r   Nr   r   r   r   rb  rb    r   r   rb  c                       e Zd ZdgZd Zy)!Pop2PianoForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z*Pop2PianoForConditionalGeneration.__init__  r   r   Nr   r   r   r   re  re    r   r   re  c                       e Zd ZdgZd Zy)Pop2PianoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!Pop2PianoPreTrainedModel.__init__%  r   r   Nr   r   r   r   rh  rh  "  r   r   rh  c                       e Zd ZdgZd Zy)ProphetNetDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   zProphetNetDecoder.__init__,  r   r   Nr   r   r   r   rk  rk  )  r   r   rk  c                       e Zd ZdgZd Zy)ProphetNetEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zProphetNetEncoder.__init__3  r   r   Nr   r   r   r   rn  rn  0  r   r   rn  c                       e Zd ZdgZd Zy)ProphetNetForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zProphetNetForCausalLM.__init__:  r   r   Nr   r   r   r   rq  rq  7  r   r   rq  c                       e Zd ZdgZd Zy)"ProphetNetForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+ProphetNetForConditionalGeneration.__init__A  r   r   Nr   r   r   r   rt  rt  >  r   r   rt  c                       e Zd ZdgZd Zy)ProphetNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zProphetNetModel.__init__H  r   r   Nr   r   r   r   rw  rw  E  r   r   rw  c                       e Zd ZdgZd Zy)ProphetNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"ProphetNetPreTrainedModel.__init__O  r   r   Nr   r   r   r   rz  rz  L  r   r   rz  c                       e Zd ZdgZd Zy)PvtForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"PvtForImageClassification.__init__V  r   r   Nr   r   r   r   r}  r}  S  r   r   r}  c                       e Zd ZdgZd Zy)PvtModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPvtModel.__init__]  r   r   Nr   r   r   r   r  r  Z  r   r   r  c                       e Zd ZdgZd Zy)PvtPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPvtPreTrainedModel.__init__d  r   r   Nr   r   r   r   r  r  a  r   r   r  c                       e Zd ZdgZd Zy)PvtV2Backboner   c                     t        | dg       y r	   r
   r   s      r   r   zPvtV2Backbone.__init__k  r   r   Nr   r   r   r   r  r  h  r   r   r  c                       e Zd ZdgZd Zy)PvtV2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$PvtV2ForImageClassification.__init__r  r   r   Nr   r   r   r   r  r  o  r   r   r  c                       e Zd ZdgZd Zy)
PvtV2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zPvtV2Model.__init__y  r   r   Nr   r   r   r   r  r  v  r   r   r  c                       e Zd ZdgZd Zy)PvtV2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zPvtV2PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r  }  r   r   r  c                       e Zd ZdgZd Zy)Qwen2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2ForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"Qwen2ForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'Qwen2ForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$Qwen2ForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)
Qwen2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2Model.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2PreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2AudioEncoderr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2AudioEncoder.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)"Qwen2AudioForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+Qwen2AudioForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2AudioPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"Qwen2AudioPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2MoeForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2MoeForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2MoeForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%Qwen2MoeForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)!Qwen2MoeForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*Qwen2MoeForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2MoeForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'Qwen2MoeForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2MoeModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2MoeModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2MoePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z Qwen2MoePreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2VLForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(Qwen2VLForConditionalGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2VLModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2VLModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)Qwen2VLPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zQwen2VLPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RagModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRagModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RagPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRagPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r  	  r   r   r  c                       e Zd ZdgZd Zy)RagSequenceForGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z!RagSequenceForGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RagTokenForGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   zRagTokenForGeneration.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RecurrentGemmaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z"RecurrentGemmaForCausalLM.__init__!  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RecurrentGemmaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRecurrentGemmaModel.__init__(  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)RecurrentGemmaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&RecurrentGemmaPreTrainedModel.__init__/  r   r   Nr   r   r   r   r  r  ,  r   r   r  c                       e Zd ZdgZd Zy)ReformerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zReformerForMaskedLM.__init__6  r   r   Nr   r   r   r   r  r  3  r   r   r  c                       e Zd ZdgZd Zy)ReformerForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%ReformerForQuestionAnswering.__init__=  r   r   Nr   r   r   r   r  r  :  r   r   r  c                       e Zd ZdgZd Zy)!ReformerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*ReformerForSequenceClassification.__init__D  r   r   Nr   r   r   r   r  r  A  r   r   r  c                       e Zd ZdgZd Zy)ReformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zReformerModel.__init__K  r   r   Nr   r   r   r   r  r  H  r   r   r  c                       e Zd ZdgZd Zy)ReformerModelWithLMHeadr   c                     t        | dg       y r	   r
   r   s      r   r   z ReformerModelWithLMHead.__init__R  r   r   Nr   r   r   r   r  r  O  r   r   r  c                       e Zd ZdgZd Zy)ReformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z ReformerPreTrainedModel.__init__Y  r   r   Nr   r   r   r   r  r  V  r   r   r  c                       e Zd ZdgZd Zy)RegNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%RegNetForImageClassification.__init__`  r   r   Nr   r   r   r   r  r  ]  r   r   r  c                       e Zd ZdgZd Zy)RegNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRegNetModel.__init__g  r   r   Nr   r   r   r   r  r  d  r   r   r  c                       e Zd ZdgZd Zy)RegNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRegNetPreTrainedModel.__init__n  r   r   Nr   r   r   r   r  r  k  r   r   r  c                       e Zd ZdgZd Zy)RemBertForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRemBertForCausalLM.__init__u  r   r   Nr   r   r   r   r  r  r  r   r   r  c                       e Zd ZdgZd Zy)RemBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRemBertForMaskedLM.__init__|  r   r   Nr   r   r   r   r  r  y  r   r   r  c                       e Zd ZdgZd Zy)RemBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!RemBertForMultipleChoice.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RemBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$RemBertForQuestionAnswering.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy) RemBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)RemBertForSequenceClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RemBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&RemBertForTokenClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RemBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRemBertModel.__init__  r   r   Nr   r   r   r   r
  r
    r   r   r
  c                       e Zd ZdgZd Zy)RemBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRemBertPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_rembertr  s     r   r  r    r@  r   c                       e Zd ZdgZd Zy)ResNetBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zResNetBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ResNetForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%ResNetForImageClassification.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ResNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zResNetModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)ResNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zResNetPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RobertaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRobertaForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RobertaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRobertaForMaskedLM.__init__  r   r   Nr   r   r   r   r!  r!    r   r   r!  c                       e Zd ZdgZd Zy)RobertaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!RobertaForMultipleChoice.__init__  r   r   Nr   r   r   r   r$  r$    r   r   r$  c                       e Zd ZdgZd Zy)RobertaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$RobertaForQuestionAnswering.__init__  r   r   Nr   r   r   r   r'  r'    r   r   r'  c                       e Zd ZdgZd Zy) RobertaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)RobertaForSequenceClassification.__init__  r   r   Nr   r   r   r   r*  r*    r   r   r*  c                       e Zd ZdgZd Zy)RobertaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&RobertaForTokenClassification.__init__  r   r   Nr   r   r   r   r-  r-    r   r   r-  c                       e Zd ZdgZd Zy)RobertaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRobertaModel.__init__  r   r   Nr   r   r   r   r0  r0    r   r   r0  c                       e Zd ZdgZd Zy)RobertaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRobertaPreTrainedModel.__init__  r   r   Nr   r   r   r   r3  r3    r   r   r3  c                       e Zd ZdgZd Zy)RobertaPreLayerNormForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z'RobertaPreLayerNormForCausalLM.__init__  r   r   Nr   r   r   r   r6  r6    r   r   r6  c                       e Zd ZdgZd Zy)RobertaPreLayerNormForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z'RobertaPreLayerNormForMaskedLM.__init__  r   r   Nr   r   r   r   r9  r9  	  r   r   r9  c                       e Zd ZdgZd Zy)$RobertaPreLayerNormForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z-RobertaPreLayerNormForMultipleChoice.__init__  r   r   Nr   r   r   r   r<  r<    r   r   r<  c                       e Zd ZdgZd Zy)'RobertaPreLayerNormForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z0RobertaPreLayerNormForQuestionAnswering.__init__  r   r   Nr   r   r   r   r?  r?    r   r   r?  c                       e Zd ZdgZd Zy),RobertaPreLayerNormForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z5RobertaPreLayerNormForSequenceClassification.__init__!  r   r   Nr   r   r   r   rB  rB    r   r   rB  c                       e Zd ZdgZd Zy))RobertaPreLayerNormForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z2RobertaPreLayerNormForTokenClassification.__init__(  r   r   Nr   r   r   r   rE  rE  %  r   r   rE  c                       e Zd ZdgZd Zy)RobertaPreLayerNormModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!RobertaPreLayerNormModel.__init__/  r   r   Nr   r   r   r   rH  rH  ,  r   r   rH  c                       e Zd ZdgZd Zy)"RobertaPreLayerNormPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z+RobertaPreLayerNormPreTrainedModel.__init__6  r   r   Nr   r   r   r   rK  rK  3  r   r   rK  c                       e Zd ZdgZd Zy)RoCBertForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRoCBertForCausalLM.__init__=  r   r   Nr   r   r   r   rN  rN  :  r   r   rN  c                       e Zd ZdgZd Zy)RoCBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRoCBertForMaskedLM.__init__D  r   r   Nr   r   r   r   rQ  rQ  A  r   r   rQ  c                       e Zd ZdgZd Zy)RoCBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z!RoCBertForMultipleChoice.__init__K  r   r   Nr   r   r   r   rT  rT  H  r   r   rT  c                       e Zd ZdgZd Zy)RoCBertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zRoCBertForPreTraining.__init__R  r   r   Nr   r   r   r   rW  rW  O  r   r   rW  c                       e Zd ZdgZd Zy)RoCBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z$RoCBertForQuestionAnswering.__init__Y  r   r   Nr   r   r   r   rZ  rZ  V  r   r   rZ  c                       e Zd ZdgZd Zy) RoCBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)RoCBertForSequenceClassification.__init__`  r   r   Nr   r   r   r   r]  r]  ]  r   r   r]  c                       e Zd ZdgZd Zy)RoCBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&RoCBertForTokenClassification.__init__g  r   r   Nr   r   r   r   r`  r`  d  r   r   r`  c                       e Zd ZdgZd Zy)RoCBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRoCBertModel.__init__n  r   r   Nr   r   r   r   rc  rc  k  r   r   rc  c                       e Zd ZdgZd Zy)RoCBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRoCBertPreTrainedModel.__init__u  r   r   Nr   r   r   r   rf  rf  r  r   r   rf  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_roc_bertr  s     r   ri  ri  y  rK  r   c                       e Zd ZdgZd Zy)RoFormerForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRoFormerForCausalLM.__init__  r   r   Nr   r   r   r   rk  rk  }  r   r   rk  c                       e Zd ZdgZd Zy)RoFormerForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRoFormerForMaskedLM.__init__  r   r   Nr   r   r   r   rn  rn    r   r   rn  c                       e Zd ZdgZd Zy)RoFormerForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z"RoFormerForMultipleChoice.__init__  r   r   Nr   r   r   r   rq  rq    r   r   rq  c                       e Zd ZdgZd Zy)RoFormerForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%RoFormerForQuestionAnswering.__init__  r   r   Nr   r   r   r   rt  rt    r   r   rt  c                       e Zd ZdgZd Zy)!RoFormerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*RoFormerForSequenceClassification.__init__  r   r   Nr   r   r   r   rw  rw    r   r   rw  c                       e Zd ZdgZd Zy)RoFormerForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'RoFormerForTokenClassification.__init__  r   r   Nr   r   r   r   rz  rz    r   r   rz  c                       e Zd ZdgZd Zy)RoFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRoFormerModel.__init__  r   r   Nr   r   r   r   r}  r}    r   r   r}  c                       e Zd ZdgZd Zy)RoFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z RoFormerPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_roformerr  s     r   r  r    rK  r   c                       e Zd ZdgZd Zy)RTDetrForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z!RTDetrForObjectDetection.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RTDetrModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRTDetrModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RTDetrPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRTDetrPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RTDetrResNetBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zRTDetrResNetBackbone.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RTDetrResNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$RTDetrResNetPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RwkvForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zRwkvForCausalLM.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)	RwkvModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRwkvModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)RwkvPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zRwkvPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)SamModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSamModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)SamPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSamPreTrainedModel.__init__  r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TCodeHifiGanr   c                     t        | dg       y r	   r
   r   s      r   r   zSeamlessM4TCodeHifiGan.__init__   r   r   Nr   r   r   r   r  r    r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TForSpeechToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z%SeamlessM4TForSpeechToSpeech.__init__	   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TForSpeechToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z#SeamlessM4TForSpeechToText.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TForTextToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z#SeamlessM4TForTextToSpeech.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TForTextToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z!SeamlessM4TForTextToText.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4THifiGanr   c                     t        | dg       y r	   r
   r   s      r   r   zSeamlessM4THifiGan.__init__%   r   r   Nr   r   r   r   r  r  "   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSeamlessM4TModel.__init__,   r   r   Nr   r   r   r   r  r  )   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#SeamlessM4TPreTrainedModel.__init__3   r   r   Nr   r   r   r   r  r  0   r   r   r  c                       e Zd ZdgZd Zy)-SeamlessM4TTextToUnitForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z6SeamlessM4TTextToUnitForConditionalGeneration.__init__:   r   r   Nr   r   r   r   r  r  7   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4TTextToUnitModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#SeamlessM4TTextToUnitModel.__init__A   r   r   Nr   r   r   r   r  r  >   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2ForSpeechToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z'SeamlessM4Tv2ForSpeechToSpeech.__init__H   r   r   Nr   r   r   r   r  r  E   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2ForSpeechToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z%SeamlessM4Tv2ForSpeechToText.__init__O   r   r   Nr   r   r   r   r  r  L   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2ForTextToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z%SeamlessM4Tv2ForTextToSpeech.__init__V   r   r   Nr   r   r   r   r  r  S   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2ForTextToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z#SeamlessM4Tv2ForTextToText.__init__]   r   r   Nr   r   r   r   r  r  Z   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zSeamlessM4Tv2Model.__init__d   r   r   Nr   r   r   r   r  r  a   r   r   r  c                       e Zd ZdgZd Zy)SeamlessM4Tv2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%SeamlessM4Tv2PreTrainedModel.__init__k   r   r   Nr   r   r   r   r  r  h   r   r   r  c                       e Zd ZdgZd Zy)SegformerDecodeHeadr   c                     t        | dg       y r	   r
   r   s      r   r   zSegformerDecodeHead.__init__r   r   r   Nr   r   r   r   r  r  o   r   r   r  c                       e Zd ZdgZd Zy)SegformerForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z(SegformerForImageClassification.__init__y   r   r   Nr   r   r   r   r  r  v   r   r   r  c                       e Zd ZdgZd Zy) SegformerForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z)SegformerForSemanticSegmentation.__init__   r   r   Nr   r   r   r   r  r  }   r   r   r  c                       e Zd ZdgZd Zy)SegformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSegformerModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SegformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!SegformerPreTrainedModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SegGptForImageSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z#SegGptForImageSegmentation.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SegGptModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSegGptModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SegGptPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSegGptPreTrainedModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)	SEWForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWForCTC.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SEWForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%SEWForSequenceClassification.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SEWModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SEWPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWPreTrainedModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)
SEWDForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWDForCTC.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SEWDForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&SEWDForSequenceClassification.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)	SEWDModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWDModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SEWDPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSEWDPreTrainedModel.__init__   r   r   Nr   r   r   r   r   r      r   r   r   c                       e Zd ZdgZd Zy)SiglipForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%SiglipForImageClassification.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SiglipModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSiglipModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SiglipPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSiglipPreTrainedModel.__init__   r   r   Nr   r   r   r   r	  r	     r   r   r	  c                       e Zd ZdgZd Zy)SiglipTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSiglipTextModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SiglipVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSiglipVisionModel.__init__   r   r   Nr   r   r   r   r  r     r   r   r  c                       e Zd ZdgZd Zy)SpeechEncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"SpeechEncoderDecoderModel.__init__!  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)#Speech2TextForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z,Speech2TextForConditionalGeneration.__init__!  r   r   Nr   r   r   r   r  r  	!  r   r   r  c                       e Zd ZdgZd Zy)Speech2TextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSpeech2TextModel.__init__!  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)Speech2TextPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#Speech2TextPreTrainedModel.__init__!  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)SpeechT5ForSpeechToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z"SpeechT5ForSpeechToSpeech.__init__!!  r   r   Nr   r   r   r   r  r  !  r   r   r  c                       e Zd ZdgZd Zy)SpeechT5ForSpeechToTextr   c                     t        | dg       y r	   r
   r   s      r   r   z SpeechT5ForSpeechToText.__init__(!  r   r   Nr   r   r   r   r!  r!  %!  r   r   r!  c                       e Zd ZdgZd Zy)SpeechT5ForTextToSpeechr   c                     t        | dg       y r	   r
   r   s      r   r   z SpeechT5ForTextToSpeech.__init__/!  r   r   Nr   r   r   r   r$  r$  ,!  r   r   r$  c                       e Zd ZdgZd Zy)SpeechT5HifiGanr   c                     t        | dg       y r	   r
   r   s      r   r   zSpeechT5HifiGan.__init__6!  r   r   Nr   r   r   r   r'  r'  3!  r   r   r'  c                       e Zd ZdgZd Zy)SpeechT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zSpeechT5Model.__init__=!  r   r   Nr   r   r   r   r*  r*  :!  r   r   r*  c                       e Zd ZdgZd Zy)SpeechT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z SpeechT5PreTrainedModel.__init__D!  r   r   Nr   r   r   r   r-  r-  A!  r   r   r-  c                       e Zd ZdgZd Zy)SplinterForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zSplinterForPreTraining.__init__K!  r   r   Nr   r   r   r   r0  r0  H!  r   r   r0  c                       e Zd ZdgZd Zy)SplinterForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z%SplinterForQuestionAnswering.__init__R!  r   r   Nr   r   r   r   r3  r3  O!  r   r   r3  c                       e Zd ZdgZd Zy)SplinterModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSplinterModel.__init__Y!  r   r   Nr   r   r   r   r6  r6  V!  r   r   r6  c                       e Zd ZdgZd Zy)SplinterPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z SplinterPreTrainedModel.__init__`!  r   r   Nr   r   r   r   r9  r9  ]!  r   r   r9  c                       e Zd ZdgZd Zy)SqueezeBertForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zSqueezeBertForMaskedLM.__init__g!  r   r   Nr   r   r   r   r<  r<  d!  r   r   r<  c                       e Zd ZdgZd Zy)SqueezeBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z%SqueezeBertForMultipleChoice.__init__n!  r   r   Nr   r   r   r   r?  r?  k!  r   r   r?  c                       e Zd ZdgZd Zy)SqueezeBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z(SqueezeBertForQuestionAnswering.__init__u!  r   r   Nr   r   r   r   rB  rB  r!  r   r   rB  c                       e Zd ZdgZd Zy)$SqueezeBertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z-SqueezeBertForSequenceClassification.__init__|!  r   r   Nr   r   r   r   rE  rE  y!  r   r   rE  c                       e Zd ZdgZd Zy)!SqueezeBertForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*SqueezeBertForTokenClassification.__init__!  r   r   Nr   r   r   r   rH  rH  !  r   r   rH  c                       e Zd ZdgZd Zy)SqueezeBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSqueezeBertModel.__init__!  r   r   Nr   r   r   r   rK  rK  !  r   r   rK  c                       e Zd ZdgZd Zy)SqueezeBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#SqueezeBertPreTrainedModel.__init__!  r   r   Nr   r   r   r   rN  rN  !  r   r   rN  c                       e Zd ZdgZd Zy)StableLmForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zStableLmForCausalLM.__init__!  r   r   Nr   r   r   r   rQ  rQ  !  r   r   rQ  c                       e Zd ZdgZd Zy)!StableLmForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*StableLmForSequenceClassification.__init__!  r   r   Nr   r   r   r   rT  rT  !  r   r   rT  c                       e Zd ZdgZd Zy)StableLmForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'StableLmForTokenClassification.__init__!  r   r   Nr   r   r   r   rW  rW  !  r   r   rW  c                       e Zd ZdgZd Zy)StableLmModelr   c                     t        | dg       y r	   r
   r   s      r   r   zStableLmModel.__init__!  r   r   Nr   r   r   r   rZ  rZ  !  r   r   rZ  c                       e Zd ZdgZd Zy)StableLmPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z StableLmPreTrainedModel.__init__!  r   r   Nr   r   r   r   r]  r]  !  r   r   r]  c                       e Zd ZdgZd Zy)Starcoder2ForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zStarcoder2ForCausalLM.__init__!  r   r   Nr   r   r   r   r`  r`  !  r   r   r`  c                       e Zd ZdgZd Zy)#Starcoder2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,Starcoder2ForSequenceClassification.__init__!  r   r   Nr   r   r   r   rc  rc  !  r   r   rc  c                       e Zd ZdgZd Zy) Starcoder2ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)Starcoder2ForTokenClassification.__init__!  r   r   Nr   r   r   r   rf  rf  !  r   r   rf  c                       e Zd ZdgZd Zy)Starcoder2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zStarcoder2Model.__init__!  r   r   Nr   r   r   r   ri  ri  !  r   r   ri  c                       e Zd ZdgZd Zy)Starcoder2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"Starcoder2PreTrainedModel.__init__!  r   r   Nr   r   r   r   rl  rl  !  r   r   rl  c                       e Zd ZdgZd Zy)SuperPointForKeypointDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z'SuperPointForKeypointDetection.__init__!  r   r   Nr   r   r   r   ro  ro  !  r   r   ro  c                       e Zd ZdgZd Zy)SuperPointPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"SuperPointPreTrainedModel.__init__!  r   r   Nr   r   r   r   rr  rr  !  r   r   rr  c                       e Zd ZdgZd Zy)!SwiftFormerForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*SwiftFormerForImageClassification.__init__!  r   r   Nr   r   r   r   ru  ru  !  r   r   ru  c                       e Zd ZdgZd Zy)SwiftFormerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwiftFormerModel.__init__!  r   r   Nr   r   r   r   rx  rx  !  r   r   rx  c                       e Zd ZdgZd Zy)SwiftFormerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#SwiftFormerPreTrainedModel.__init__!  r   r   Nr   r   r   r   r{  r{  !  r   r   r{  c                       e Zd ZdgZd Zy)SwinBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zSwinBackbone.__init__"  r   r   Nr   r   r   r   r~  r~  !  r   r   r~  c                       e Zd ZdgZd Zy)SwinForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#SwinForImageClassification.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)SwinForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z#SwinForMaskedImageModeling.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)	SwinModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwinModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)SwinPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwinPreTrainedModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)Swin2SRForImageSuperResolutionr   c                     t        | dg       y r	   r
   r   s      r   r   z'Swin2SRForImageSuperResolution.__init__$"  r   r   Nr   r   r   r   r  r  !"  r   r   r  c                       e Zd ZdgZd Zy)Swin2SRModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwin2SRModel.__init__+"  r   r   Nr   r   r   r   r  r  ("  r   r   r  c                       e Zd ZdgZd Zy)Swin2SRPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwin2SRPreTrainedModel.__init__2"  r   r   Nr   r   r   r   r  r  /"  r   r   r  c                       e Zd ZdgZd Zy)Swinv2Backboner   c                     t        | dg       y r	   r
   r   s      r   r   zSwinv2Backbone.__init__9"  r   r   Nr   r   r   r   r  r  6"  r   r   r  c                       e Zd ZdgZd Zy)Swinv2ForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%Swinv2ForImageClassification.__init__@"  r   r   Nr   r   r   r   r  r  ="  r   r   r  c                       e Zd ZdgZd Zy)Swinv2ForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z%Swinv2ForMaskedImageModeling.__init__G"  r   r   Nr   r   r   r   r  r  D"  r   r   r  c                       e Zd ZdgZd Zy)Swinv2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwinv2Model.__init__N"  r   r   Nr   r   r   r   r  r  K"  r   r   r  c                       e Zd ZdgZd Zy)Swinv2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSwinv2PreTrainedModel.__init__U"  r   r   Nr   r   r   r   r  r  R"  r   r   r  c                       e Zd ZdgZd Zy)SwitchTransformersEncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'SwitchTransformersEncoderModel.__init__\"  r   r   Nr   r   r   r   r  r  Y"  r   r   r  c                       e Zd ZdgZd Zy)*SwitchTransformersForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z3SwitchTransformersForConditionalGeneration.__init__c"  r   r   Nr   r   r   r   r  r  `"  r   r   r  c                       e Zd ZdgZd Zy)SwitchTransformersModelr   c                     t        | dg       y r	   r
   r   s      r   r   z SwitchTransformersModel.__init__j"  r   r   Nr   r   r   r   r  r  g"  r   r   r  c                       e Zd ZdgZd Zy)!SwitchTransformersPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z*SwitchTransformersPreTrainedModel.__init__q"  r   r   Nr   r   r   r   r  r  n"  r   r   r  c                       e Zd ZdgZd Zy)SwitchTransformersSparseMLPr   c                     t        | dg       y r	   r
   r   s      r   r   z$SwitchTransformersSparseMLP.__init__x"  r   r   Nr   r   r   r   r  r  u"  r   r   r  c                       e Zd ZdgZd Zy)SwitchTransformersTop1Routerr   c                     t        | dg       y r	   r
   r   s      r   r   z%SwitchTransformersTop1Router.__init__"  r   r   Nr   r   r   r   r  r  |"  r   r   r  c                       e Zd ZdgZd Zy)T5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zT5EncoderModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z#T5ForConditionalGeneration.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   zT5ForQuestionAnswering.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$T5ForSequenceClassification.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z!T5ForTokenClassification.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zT5Model.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)T5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zT5PreTrainedModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_t5r  s     r   r  r  "      +gY7r   c                       e Zd ZdgZd Zy)"TableTransformerForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z+TableTransformerForObjectDetection.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TableTransformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTableTransformerModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TableTransformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z(TableTransformerPreTrainedModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TapasForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zTapasForMaskedLM.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TapasForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"TapasForQuestionAnswering.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TapasForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'TapasForSequenceClassification.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)
TapasModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTapasModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TapasPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTapasPreTrainedModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_tapasr  s     r   r  r  "  rX  r   c                       e Zd ZdgZd Zy)"TimeSeriesTransformerForPredictionr   c                     t        | dg       y r	   r
   r   s      r   r   z+TimeSeriesTransformerForPrediction.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)TimeSeriesTransformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#TimeSeriesTransformerModel.__init__"  r   r   Nr   r   r   r   r  r  "  r   r   r  c                       e Zd ZdgZd Zy)$TimeSeriesTransformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z-TimeSeriesTransformerPreTrainedModel.__init__#  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)!TimesformerForVideoClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*TimesformerForVideoClassification.__init__#  r   r   Nr   r   r   r   r  r  	#  r   r   r  c                       e Zd ZdgZd Zy)TimesformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTimesformerModel.__init__#  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)TimesformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#TimesformerPreTrainedModel.__init__#  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)TimmBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zTimmBackbone.__init__!#  r   r   Nr   r   r   r   r  r  #  r   r   r  c                       e Zd ZdgZd Zy)TrOCRForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zTrOCRForCausalLM.__init__(#  r   r   Nr   r   r   r   r  r  %#  r   r   r  c                       e Zd ZdgZd Zy)TrOCRPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTrOCRPreTrainedModel.__init__/#  r   r   Nr   r   r   r   r  r  ,#  r   r   r  c                       e Zd ZdgZd Zy)TvpForVideoGroundingr   c                     t        | dg       y r	   r
   r   s      r   r   zTvpForVideoGrounding.__init__6#  r   r   Nr   r   r   r   r  r  3#  r   r   r  c                       e Zd ZdgZd Zy)TvpModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTvpModel.__init__=#  r   r   Nr   r   r   r   r  r  :#  r   r   r  c                       e Zd ZdgZd Zy)TvpPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTvpPreTrainedModel.__init__D#  r   r   Nr   r   r   r   r
  r
  A#  r   r   r
  c                       e Zd ZdgZd Zy)UdopEncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUdopEncoderModel.__init__K#  r   r   Nr   r   r   r   r  r  H#  r   r   r  c                       e Zd ZdgZd Zy)UdopForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z%UdopForConditionalGeneration.__init__R#  r   r   Nr   r   r   r   r  r  O#  r   r   r  c                       e Zd ZdgZd Zy)	UdopModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUdopModel.__init__Y#  r   r   Nr   r   r   r   r  r  V#  r   r   r  c                       e Zd ZdgZd Zy)UdopPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUdopPreTrainedModel.__init__`#  r   r   Nr   r   r   r   r  r  ]#  r   r   r  c                       e Zd ZdgZd Zy)UMT5EncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUMT5EncoderModel.__init__g#  r   r   Nr   r   r   r   r  r  d#  r   r   r  c                       e Zd ZdgZd Zy)UMT5ForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z%UMT5ForConditionalGeneration.__init__n#  r   r   Nr   r   r   r   r  r  k#  r   r   r  c                       e Zd ZdgZd Zy)UMT5ForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!UMT5ForQuestionAnswering.__init__u#  r   r   Nr   r   r   r   r  r  r#  r   r   r  c                       e Zd ZdgZd Zy)UMT5ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&UMT5ForSequenceClassification.__init__|#  r   r   Nr   r   r   r   r"  r"  y#  r   r   r"  c                       e Zd ZdgZd Zy)UMT5ForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#UMT5ForTokenClassification.__init__#  r   r   Nr   r   r   r   r%  r%  #  r   r   r%  c                       e Zd ZdgZd Zy)	UMT5Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zUMT5Model.__init__#  r   r   Nr   r   r   r   r(  r(  #  r   r   r(  c                       e Zd ZdgZd Zy)UMT5PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUMT5PreTrainedModel.__init__#  r   r   Nr   r   r   r   r+  r+  #  r   r   r+  c                       e Zd ZdgZd Zy)UniSpeechForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zUniSpeechForCTC.__init__#  r   r   Nr   r   r   r   r.  r.  #  r   r   r.  c                       e Zd ZdgZd Zy)UniSpeechForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z UniSpeechForPreTraining.__init__#  r   r   Nr   r   r   r   r1  r1  #  r   r   r1  c                       e Zd ZdgZd Zy)"UniSpeechForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+UniSpeechForSequenceClassification.__init__#  r   r   Nr   r   r   r   r4  r4  #  r   r   r4  c                       e Zd ZdgZd Zy)UniSpeechModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUniSpeechModel.__init__#  r   r   Nr   r   r   r   r7  r7  #  r   r   r7  c                       e Zd ZdgZd Zy)UniSpeechPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!UniSpeechPreTrainedModel.__init__#  r   r   Nr   r   r   r   r:  r:  #  r   r   r:  c                       e Zd ZdgZd Zy)'UniSpeechSatForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0UniSpeechSatForAudioFrameClassification.__init__#  r   r   Nr   r   r   r   r=  r=  #  r   r   r=  c                       e Zd ZdgZd Zy)UniSpeechSatForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zUniSpeechSatForCTC.__init__#  r   r   Nr   r   r   r   r@  r@  #  r   r   r@  c                       e Zd ZdgZd Zy)UniSpeechSatForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z#UniSpeechSatForPreTraining.__init__#  r   r   Nr   r   r   r   rC  rC  #  r   r   rC  c                       e Zd ZdgZd Zy)%UniSpeechSatForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.UniSpeechSatForSequenceClassification.__init__#  r   r   Nr   r   r   r   rF  rF  #  r   r   rF  c                       e Zd ZdgZd Zy)UniSpeechSatForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   zUniSpeechSatForXVector.__init__#  r   r   Nr   r   r   r   rI  rI  #  r   r   rI  c                       e Zd ZdgZd Zy)UniSpeechSatModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUniSpeechSatModel.__init__#  r   r   Nr   r   r   r   rL  rL  #  r   r   rL  c                       e Zd ZdgZd Zy)UniSpeechSatPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$UniSpeechSatPreTrainedModel.__init__#  r   r   Nr   r   r   r   rO  rO  #  r   r   rO  c                       e Zd ZdgZd Zy)UnivNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUnivNetModel.__init__#  r   r   Nr   r   r   r   rR  rR  #  r   r   rR  c                       e Zd ZdgZd Zy)UperNetForSemanticSegmentationr   c                     t        | dg       y r	   r
   r   s      r   r   z'UperNetForSemanticSegmentation.__init__#  r   r   Nr   r   r   r   rU  rU  #  r   r   rU  c                       e Zd ZdgZd Zy)UperNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUperNetPreTrainedModel.__init__#  r   r   Nr   r   r   r   rX  rX  #  r   r   rX  c                       e Zd ZdgZd Zy)"VideoLlavaForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z+VideoLlavaForConditionalGeneration.__init__$  r   r   Nr   r   r   r   r[  r[  #  r   r   r[  c                       e Zd ZdgZd Zy)VideoLlavaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"VideoLlavaPreTrainedModel.__init__$  r   r   Nr   r   r   r   r^  r^  $  r   r   r^  c                       e Zd ZdgZd Zy)VideoLlavaProcessorr   c                     t        | dg       y r	   r
   r   s      r   r   zVideoLlavaProcessor.__init__$  r   r   Nr   r   r   r   ra  ra  $  r   r   ra  c                       e Zd ZdgZd Zy)VideoMAEForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zVideoMAEForPreTraining.__init__$  r   r   Nr   r   r   r   rd  rd  $  r   r   rd  c                       e Zd ZdgZd Zy)VideoMAEForVideoClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'VideoMAEForVideoClassification.__init__$  r   r   Nr   r   r   r   rg  rg  $  r   r   rg  c                       e Zd ZdgZd Zy)VideoMAEModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVideoMAEModel.__init__$$  r   r   Nr   r   r   r   rj  rj  !$  r   r   rj  c                       e Zd ZdgZd Zy)VideoMAEPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z VideoMAEPreTrainedModel.__init__+$  r   r   Nr   r   r   r   rm  rm  ($  r   r   rm  c                       e Zd ZdgZd Zy)ViltForImageAndTextRetrievalr   c                     t        | dg       y r	   r
   r   s      r   r   z%ViltForImageAndTextRetrieval.__init__2$  r   r   Nr   r   r   r   rp  rp  /$  r   r   rp  c                       e Zd ZdgZd Zy)"ViltForImagesAndTextClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+ViltForImagesAndTextClassification.__init__9$  r   r   Nr   r   r   r   rs  rs  6$  r   r   rs  c                       e Zd ZdgZd Zy)ViltForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zViltForMaskedLM.__init__@$  r   r   Nr   r   r   r   rv  rv  =$  r   r   rv  c                       e Zd ZdgZd Zy)ViltForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!ViltForQuestionAnswering.__init__G$  r   r   Nr   r   r   r   ry  ry  D$  r   r   ry  c                       e Zd ZdgZd Zy)ViltForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#ViltForTokenClassification.__init__N$  r   r   Nr   r   r   r   r|  r|  K$  r   r   r|  c                       e Zd ZdgZd Zy)	ViltModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViltModel.__init__U$  r   r   Nr   r   r   r   r  r  R$  r   r   r  c                       e Zd ZdgZd Zy)ViltPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViltPreTrainedModel.__init__\$  r   r   Nr   r   r   r   r  r  Y$  r   r   r  c                       e Zd ZdgZd Zy) VipLlavaForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z)VipLlavaForConditionalGeneration.__init__c$  r   r   Nr   r   r   r   r  r  `$  r   r   r  c                       e Zd ZdgZd Zy)VipLlavaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z VipLlavaPreTrainedModel.__init__j$  r   r   Nr   r   r   r   r  r  g$  r   r   r  c                       e Zd ZdgZd Zy)VisionEncoderDecoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"VisionEncoderDecoderModel.__init__q$  r   r   Nr   r   r   r   r  r  n$  r   r   r  c                       e Zd ZdgZd Zy)VisionTextDualEncoderModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#VisionTextDualEncoderModel.__init__x$  r   r   Nr   r   r   r   r  r  u$  r   r   r  c                       e Zd ZdgZd Zy)VisualBertForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$VisualBertForMultipleChoice.__init__$  r   r   Nr   r   r   r   r  r  |$  r   r   r  c                       e Zd ZdgZd Zy)VisualBertForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z!VisualBertForPreTraining.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VisualBertForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'VisualBertForQuestionAnswering.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)$VisualBertForRegionToPhraseAlignmentr   c                     t        | dg       y r	   r
   r   s      r   r   z-VisualBertForRegionToPhraseAlignment.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VisualBertForVisualReasoningr   c                     t        | dg       y r	   r
   r   s      r   r   z%VisualBertForVisualReasoning.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VisualBertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVisualBertModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VisualBertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"VisualBertPreTrainedModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"ViTForImageClassification.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTForMaskedImageModelingr   c                     t        | dg       y r	   r
   r   s      r   r   z"ViTForMaskedImageModeling.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTPreTrainedModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMAEForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zViTMAEForPreTraining.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMAEModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTMAEModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMAEPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTMAEPreTrainedModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMSNForImageClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%ViTMSNForImageClassification.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMSNModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTMSNModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)ViTMSNPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zViTMSNPreTrainedModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VitDetBackboner   c                     t        | dg       y r	   r
   r   s      r   r   zVitDetBackbone.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VitDetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVitDetModel.__init__$  r   r   Nr   r   r   r   r  r  $  r   r   r  c                       e Zd ZdgZd Zy)VitDetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVitDetPreTrainedModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)VitMatteForImageMattingr   c                     t        | dg       y r	   r
   r   s      r   r   z VitMatteForImageMatting.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)VitMattePreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z VitMattePreTrainedModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)	VitsModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVitsModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)VitsPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVitsPreTrainedModel.__init__ %  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)VivitForVideoClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$VivitForVideoClassification.__init__'%  r   r   Nr   r   r   r   r  r  $%  r   r   r  c                       e Zd ZdgZd Zy)
VivitModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVivitModel.__init__.%  r   r   Nr   r   r   r   r  r  +%  r   r   r  c                       e Zd ZdgZd Zy)VivitPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVivitPreTrainedModel.__init__5%  r   r   Nr   r   r   r   r  r  2%  r   r   r  c                       e Zd ZdgZd Zy)#Wav2Vec2ForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,Wav2Vec2ForAudioFrameClassification.__init__<%  r   r   Nr   r   r   r   r  r  9%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2ForCTC.__init__C%  r   r   Nr   r   r   r   r  r  @%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2ForMaskedLM.__init__J%  r   r   Nr   r   r   r   r  r  G%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2ForPreTraining.__init__Q%  r   r   Nr   r   r   r   r  r  N%  r   r   r  c                       e Zd ZdgZd Zy)!Wav2Vec2ForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z*Wav2Vec2ForSequenceClassification.__init__X%  r   r   Nr   r   r   r   r  r  U%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2ForXVector.__init___%  r   r   Nr   r   r   r   r  r  \%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2Modelr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2Model.__init__f%  r   r   Nr   r   r   r   r  r  c%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2PreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z Wav2Vec2PreTrainedModel.__init__m%  r   r   Nr   r   r   r   r  r  j%  r   r   r  c                       e Zd ZdgZd Zy)'Wav2Vec2BertForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z0Wav2Vec2BertForAudioFrameClassification.__init__t%  r   r   Nr   r   r   r   r  r  q%  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2BertForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2BertForCTC.__init__{%  r   r   Nr   r   r   r   r  r  x%  r   r   r  c                       e Zd ZdgZd Zy)%Wav2Vec2BertForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.Wav2Vec2BertForSequenceClassification.__init__%  r   r   Nr   r   r   r   r   r   %  r   r   r   c                       e Zd ZdgZd Zy)Wav2Vec2BertForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2BertForXVector.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2BertModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2BertModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2BertPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$Wav2Vec2BertPreTrainedModel.__init__%  r   r   Nr   r   r   r   r	  r	  %  r   r   r	  c                       e Zd ZdgZd Zy),Wav2Vec2ConformerForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z5Wav2Vec2ConformerForAudioFrameClassification.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ConformerForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   z Wav2Vec2ConformerForCTC.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ConformerForPreTrainingr   c                     t        | dg       y r	   r
   r   s      r   r   z(Wav2Vec2ConformerForPreTraining.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)*Wav2Vec2ConformerForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z3Wav2Vec2ConformerForSequenceClassification.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ConformerForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   z$Wav2Vec2ConformerForXVector.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy)Wav2Vec2ConformerModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWav2Vec2ConformerModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy) Wav2Vec2ConformerPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z)Wav2Vec2ConformerPreTrainedModel.__init__%  r   r   Nr   r   r   r   r  r  %  r   r   r  c                       e Zd ZdgZd Zy) WavLMForAudioFrameClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)WavLMForAudioFrameClassification.__init__%  r   r   Nr   r   r   r   r!  r!  %  r   r   r!  c                       e Zd ZdgZd Zy)WavLMForCTCr   c                     t        | dg       y r	   r
   r   s      r   r   zWavLMForCTC.__init__%  r   r   Nr   r   r   r   r$  r$  %  r   r   r$  c                       e Zd ZdgZd Zy)WavLMForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'WavLMForSequenceClassification.__init__%  r   r   Nr   r   r   r   r'  r'  %  r   r   r'  c                       e Zd ZdgZd Zy)WavLMForXVectorr   c                     t        | dg       y r	   r
   r   s      r   r   zWavLMForXVector.__init__%  r   r   Nr   r   r   r   r*  r*  %  r   r   r*  c                       e Zd ZdgZd Zy)
WavLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWavLMModel.__init__%  r   r   Nr   r   r   r   r-  r-  %  r   r   r-  c                       e Zd ZdgZd Zy)WavLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWavLMPreTrainedModel.__init__%  r   r   Nr   r   r   r   r0  r0  %  r   r   r0  c                       e Zd ZdgZd Zy)WhisperForAudioClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&WhisperForAudioClassification.__init__%  r   r   Nr   r   r   r   r3  r3  %  r   r   r3  c                       e Zd ZdgZd Zy)WhisperForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zWhisperForCausalLM.__init__ &  r   r   Nr   r   r   r   r6  r6  %  r   r   r6  c                       e Zd ZdgZd Zy)WhisperForConditionalGenerationr   c                     t        | dg       y r	   r
   r   s      r   r   z(WhisperForConditionalGeneration.__init__&  r   r   Nr   r   r   r   r9  r9  &  r   r   r9  c                       e Zd ZdgZd Zy)WhisperModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWhisperModel.__init__&  r   r   Nr   r   r   r   r<  r<  &  r   r   r<  c                       e Zd ZdgZd Zy)WhisperPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWhisperPreTrainedModel.__init__&  r   r   Nr   r   r   r   r?  r?  &  r   r   r?  c                       e Zd ZdgZd Zy)
XCLIPModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXCLIPModel.__init__&  r   r   Nr   r   r   r   rB  rB  &  r   r   rB  c                       e Zd ZdgZd Zy)XCLIPPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXCLIPPreTrainedModel.__init__#&  r   r   Nr   r   r   r   rE  rE   &  r   r   rE  c                       e Zd ZdgZd Zy)XCLIPTextModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXCLIPTextModel.__init__*&  r   r   Nr   r   r   r   rH  rH  '&  r   r   rH  c                       e Zd ZdgZd Zy)XCLIPVisionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXCLIPVisionModel.__init__1&  r   r   Nr   r   r   r   rK  rK  .&  r   r   rK  c                       e Zd ZdgZd Zy)XGLMForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zXGLMForCausalLM.__init__8&  r   r   Nr   r   r   r   rN  rN  5&  r   r   rN  c                       e Zd ZdgZd Zy)	XGLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXGLMModel.__init__?&  r   r   Nr   r   r   r   rQ  rQ  <&  r   r   rQ  c                       e Zd ZdgZd Zy)XGLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXGLMPreTrainedModel.__init__F&  r   r   Nr   r   r   r   rT  rT  C&  r   r   rT  c                       e Zd ZdgZd Zy)XLMForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zXLMForMultipleChoice.__init__M&  r   r   Nr   r   r   r   rW  rW  J&  r   r   rW  c                       e Zd ZdgZd Zy)XLMForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z XLMForQuestionAnswering.__init__T&  r   r   Nr   r   r   r   rZ  rZ  Q&  r   r   rZ  c                       e Zd ZdgZd Zy)XLMForQuestionAnsweringSimpler   c                     t        | dg       y r	   r
   r   s      r   r   z&XLMForQuestionAnsweringSimple.__init__[&  r   r   Nr   r   r   r   r]  r]  X&  r   r   r]  c                       e Zd ZdgZd Zy)XLMForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z%XLMForSequenceClassification.__init__b&  r   r   Nr   r   r   r   r`  r`  _&  r   r   r`  c                       e Zd ZdgZd Zy)XLMForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z"XLMForTokenClassification.__init__i&  r   r   Nr   r   r   r   rc  rc  f&  r   r   rc  c                       e Zd ZdgZd Zy)XLMModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMModel.__init__p&  r   r   Nr   r   r   r   rf  rf  m&  r   r   rf  c                       e Zd ZdgZd Zy)XLMPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMPreTrainedModel.__init__w&  r   r   Nr   r   r   r   ri  ri  t&  r   r   ri  c                       e Zd ZdgZd Zy)XLMWithLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMWithLMHeadModel.__init__~&  r   r   Nr   r   r   r   rl  rl  {&  r   r   rl  c                       e Zd ZdgZd Zy)XLMRobertaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMRobertaForCausalLM.__init__&  r   r   Nr   r   r   r   ro  ro  &  r   r   ro  c                       e Zd ZdgZd Zy)XLMRobertaForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMRobertaForMaskedLM.__init__&  r   r   Nr   r   r   r   rr  rr  &  r   r   rr  c                       e Zd ZdgZd Zy)XLMRobertaForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z$XLMRobertaForMultipleChoice.__init__&  r   r   Nr   r   r   r   ru  ru  &  r   r   ru  c                       e Zd ZdgZd Zy)XLMRobertaForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z'XLMRobertaForQuestionAnswering.__init__&  r   r   Nr   r   r   r   rx  rx  &  r   r   rx  c                       e Zd ZdgZd Zy)#XLMRobertaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z,XLMRobertaForSequenceClassification.__init__&  r   r   Nr   r   r   r   r{  r{  &  r   r   r{  c                       e Zd ZdgZd Zy) XLMRobertaForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z)XLMRobertaForTokenClassification.__init__&  r   r   Nr   r   r   r   r~  r~  &  r   r   r~  c                       e Zd ZdgZd Zy)XLMRobertaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMRobertaModel.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"XLMRobertaPreTrainedModel.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaXLForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   z XLMRobertaXLForCausalLM.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaXLForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   z XLMRobertaXLForMaskedLM.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaXLForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   z&XLMRobertaXLForMultipleChoice.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy) XLMRobertaXLForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z)XLMRobertaXLForQuestionAnswering.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)%XLMRobertaXLForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z.XLMRobertaXLForSequenceClassification.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)"XLMRobertaXLForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z+XLMRobertaXLForTokenClassification.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaXLModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLMRobertaXLModel.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLMRobertaXLPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$XLMRobertaXLPreTrainedModel.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLNetForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zXLNetForMultipleChoice.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLNetForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z"XLNetForQuestionAnswering.__init__&  r   r   Nr   r   r   r   r  r  &  r   r   r  c                       e Zd ZdgZd Zy)XLNetForQuestionAnsweringSimpler   c                     t        | dg       y r	   r
   r   s      r   r   z(XLNetForQuestionAnsweringSimple.__init__'  r   r   Nr   r   r   r   r  r   '  r   r   r  c                       e Zd ZdgZd Zy)XLNetForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'XLNetForSequenceClassification.__init__
'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)XLNetForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z$XLNetForTokenClassification.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)XLNetLMHeadModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLNetLMHeadModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)
XLNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLNetModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)XLNetPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXLNetPreTrainedModel.__init__&'  r   r   Nr   r   r   r   r  r  #'  r   r   r  c                  &    t        t        dg       y r	   )r   load_tf_weights_in_xlnetr  s     r   r  r  *'  rX  r   c                       e Zd ZdgZd Zy)XmodForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zXmodForCausalLM.__init__1'  r   r   Nr   r   r   r   r  r  .'  r   r   r  c                       e Zd ZdgZd Zy)XmodForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zXmodForMaskedLM.__init__8'  r   r   Nr   r   r   r   r  r  5'  r   r   r  c                       e Zd ZdgZd Zy)XmodForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zXmodForMultipleChoice.__init__?'  r   r   Nr   r   r   r   r  r  <'  r   r   r  c                       e Zd ZdgZd Zy)XmodForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!XmodForQuestionAnswering.__init__F'  r   r   Nr   r   r   r   r  r  C'  r   r   r  c                       e Zd ZdgZd Zy)XmodForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&XmodForSequenceClassification.__init__M'  r   r   Nr   r   r   r   r  r  J'  r   r   r  c                       e Zd ZdgZd Zy)XmodForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#XmodForTokenClassification.__init__T'  r   r   Nr   r   r   r   r  r  Q'  r   r   r  c                       e Zd ZdgZd Zy)	XmodModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXmodModel.__init__['  r   r   Nr   r   r   r   r  r  X'  r   r   r  c                       e Zd ZdgZd Zy)XmodPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zXmodPreTrainedModel.__init__b'  r   r   Nr   r   r   r   r  r  _'  r   r   r  c                       e Zd ZdgZd Zy)YolosForObjectDetectionr   c                     t        | dg       y r	   r
   r   s      r   r   z YolosForObjectDetection.__init__i'  r   r   Nr   r   r   r   r  r  f'  r   r   r  c                       e Zd ZdgZd Zy)
YolosModelr   c                     t        | dg       y r	   r
   r   s      r   r   zYolosModel.__init__p'  r   r   Nr   r   r   r   r  r  m'  r   r   r  c                       e Zd ZdgZd Zy)YolosPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zYolosPreTrainedModel.__init__w'  r   r   Nr   r   r   r   r  r  t'  r   r   r  c                       e Zd ZdgZd Zy)YosoForMaskedLMr   c                     t        | dg       y r	   r
   r   s      r   r   zYosoForMaskedLM.__init__~'  r   r   Nr   r   r   r   r  r  {'  r   r   r  c                       e Zd ZdgZd Zy)YosoForMultipleChoicer   c                     t        | dg       y r	   r
   r   s      r   r   zYosoForMultipleChoice.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)YosoForQuestionAnsweringr   c                     t        | dg       y r	   r
   r   s      r   r   z!YosoForQuestionAnswering.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)YosoForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z&YosoForSequenceClassification.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)YosoForTokenClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z#YosoForTokenClassification.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)	YosoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zYosoModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)YosoPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zYosoPreTrainedModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)ZambaForCausalLMr   c                     t        | dg       y r	   r
   r   s      r   r   zZambaForCausalLM.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)ZambaForSequenceClassificationr   c                     t        | dg       y r	   r
   r   s      r   r   z'ZambaForSequenceClassification.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)
ZambaModelr   c                     t        | dg       y r	   r
   r   s      r   r   zZambaModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)ZambaPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   zZambaPreTrainedModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)ZoeDepthForDepthEstimationr   c                     t        | dg       y r	   r
   r   s      r   r   z#ZoeDepthForDepthEstimation.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)ZoeDepthPreTrainedModelr   c                     t        | dg       y r	   r
   r   s      r   r   z ZoeDepthPreTrainedModel.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)	Adafactorr   c                     t        | dg       y r	   r
   r   s      r   r   zAdafactor.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                       e Zd ZdgZd Zy)AdamWr   c                     t        | dg       y r	   r
   r   s      r   r   zAdamW.__init__'  r   r   Nr   r   r   r   r  r  '  r   r   r  c                  &    t        t        dg       y r	   )r   get_constant_scheduler  s     r   r  r  '  r  r   c                  &    t        t        dg       y r	   )r   !get_constant_schedule_with_warmupr  s     r   r	  r	  '  s    7'Cr   c                  &    t        t        dg       y r	   )r   get_cosine_schedule_with_warmupr  s     r   r  r  '  rU
  r   c                  &    t        t        dg       y r	   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr  s     r   r  r  '  s    H7)Tr   c                  &    t        t        dg       y r	   )r   get_inverse_sqrt_scheduler  s     r   r  r  '  r&  r   c                  &    t        t        dg       y r	   )r   get_linear_schedule_with_warmupr  s     r   r  r  '  rU
  r   c                  &    t        t        dg       y r	   )r   )get_polynomial_decay_schedule_with_warmupr  s     r   r  r  '  s    ?'Kr   c                  &    t        t        dg       y r	   )r   get_schedulerr  s     r   r  r   (  s    mgY/r   c                  &    t        t        dg       y r	   )r   get_wsd_scheduler  s     r   r  r  (  s    &	2r   c                       e Zd ZdgZd Zy)Conv1Dr   c                     t        | dg       y r	   r
   r   s      r   r   zConv1D.__init__(  r   r   Nr   r   r   r   r  r  (  r   r   r  c                  &    t        t        dg       y r	   )r   apply_chunking_to_forwardr  s     r   r  r  (  r&  r   c                  &    t        t        dg       y r	   )r   prune_layerr  s     r   r  r  (  s    kG9-r   c                       e Zd ZdgZd Zy)Trainerr   c                     t        | dg       y r	   r
   r   s      r   r   zTrainer.__init__(  r   r   Nr   r   r   r   r   r   (  r   r   r   c                  &    t        t        dg       y r	   )r   torch_distributed_zero_firstr  s     r   r#  r#  (  s    2WI>r   c                       e Zd ZdgZd Zy)Seq2SeqTrainerr   c                     t        | dg       y r	   r
   r   s      r   r   zSeq2SeqTrainer.__init__%(  r   r   Nr   r   r   r   r%  r%  "(  r   r   r%  (  utilsr   r   r   r   r   r"   r%   r(   r+   r.   r1   r4   r7   r:   r=   r@   rC   rF   rI   rL   rO   rR   rU   rX   r[   r^   ra   rd   rg   rj   rm   rp   rs   rv   ry   r|   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  ROPE_INIT_FUNCTIONSr
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  &MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING,MODEL_FOR_AUDIO_FRAME_CLASSIFICATION_MAPPINGMODEL_FOR_AUDIO_XVECTOR_MAPPINGMODEL_FOR_BACKBONE_MAPPING'MODEL_FOR_CAUSAL_IMAGE_MODELING_MAPPINGMODEL_FOR_CAUSAL_LM_MAPPINGMODEL_FOR_CTC_MAPPING"MODEL_FOR_DEPTH_ESTIMATION_MAPPING-MODEL_FOR_DOCUMENT_QUESTION_ANSWERING_MAPPING&MODEL_FOR_IMAGE_CLASSIFICATION_MAPPINGMODEL_FOR_IMAGE_MAPPING$MODEL_FOR_IMAGE_SEGMENTATION_MAPPING$MODEL_FOR_IMAGE_TEXT_TO_TEXT_MAPPING MODEL_FOR_IMAGE_TO_IMAGE_MAPPING'MODEL_FOR_INSTANCE_SEGMENTATION_MAPPING$MODEL_FOR_KEYPOINT_DETECTION_MAPPING!MODEL_FOR_MASK_GENERATION_MAPPING'MODEL_FOR_MASKED_IMAGE_MODELING_MAPPINGMODEL_FOR_MASKED_LM_MAPPING!MODEL_FOR_MULTIPLE_CHOICE_MAPPING*MODEL_FOR_NEXT_SENTENCE_PREDICTION_MAPPING"MODEL_FOR_OBJECT_DETECTION_MAPPINGMODEL_FOR_PRETRAINING_MAPPING$MODEL_FOR_QUESTION_ANSWERING_MAPPING'MODEL_FOR_SEMANTIC_SEGMENTATION_MAPPING&MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING)MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING"MODEL_FOR_SPEECH_SEQ_2_SEQ_MAPPING*MODEL_FOR_TABLE_QUESTION_ANSWERING_MAPPINGMODEL_FOR_TEXT_ENCODING_MAPPING%MODEL_FOR_TEXT_TO_SPECTROGRAM_MAPPING"MODEL_FOR_TEXT_TO_WAVEFORM_MAPPING,MODEL_FOR_TIME_SERIES_CLASSIFICATION_MAPPING(MODEL_FOR_TIME_SERIES_REGRESSION_MAPPING&MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING(MODEL_FOR_UNIVERSAL_SEGMENTATION_MAPPING&MODEL_FOR_VIDEO_CLASSIFICATION_MAPPINGMODEL_FOR_VISION_2_SEQ_MAPPING+MODEL_FOR_VISUAL_QUESTION_ANSWERING_MAPPING0MODEL_FOR_ZERO_SHOT_IMAGE_CLASSIFICATION_MAPPING,MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPINGMODEL_MAPPINGMODEL_WITH_LM_HEAD_MAPPINGrI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  rj  rm  rp  rs  rv  ry  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r	  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r/  r2  r5  r8  r;  r>  rA  rD  rG  rJ  rM  rP  rS  rV  rY  r\  r_  rb  re  rh  rk  rn  rq  rt  rw  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  rI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  rj  rm  rp  rs  rv  ry  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r	  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  rI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  rj  rm  rp  rs  rv  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  rI  rL  rO  rR  rU  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r 	  r#	  r&	  r)	  r,	  r/	  r2	  r5	  r8	  r;	  r>	  rA	  rD	  rG	  rJ	  rM	  rP	  rS	  rV	  rY	  r\	  r_	  rb	  re	  rh	  rk	  rn	  rq	  rt	  rw	  rz	  r}	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r	  r
  r
  r
  r

  r
  r
  r
  r
  r
  r
  r
  r"
  r%
  r(
  r+
  r.
  r1
  r4
  r7
  r:
  r=
  r@
  rC
  rF
  rI
  rK
  rN
  rQ
  rT
  rW
  rZ
  r]
  r`
  rc
  re
  rh
  rk
  rn
  rq
  rt
  rw
  rz
  r}
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r
  r  r  r  r
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  rI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  rj  rm  rp  rs  rv  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r#  r&  r)  r,  r/  r2  r5  r8  r;  r>  rA  rD  rG  rJ  rM  rP  rS  rV  rY  r\  r_  rb  re  rh  rk  rn  rq  rt  rw  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rk  rn  rq  rt  rw  rz  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r	  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r  r  r  r  r  r"  r%  r(  r+  r.  r1  r4  r7  r:  r=  r@  rC  rF  rI  rL  rO  rR  rU  rX  r[  r^  ra  rd  rg  rj  rm  rp  rs  rv  ry  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r  r  r	  r  r  r  r  r  r  r  r!  r$  r'  r*  r-  r0  r3  r6  r9  r<  r?  rB  rE  rH  rK  rN  rQ  rT  rW  rZ  r]  r`  rc  rf  ri  rl  ro  rr  ru  rx  r{  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r  r  r  r  r  r  r  r  r  r  r   r#  r%  r   r   r   <module>rT     sE   2+ +++ ++k ++K ++[ ++K +++ ++K ++; ++{ ++[ ++{ ++[ ++[ +++ ++; ++K ++K +++ ++k ++ ++[ ++[ ++; ++K ++[ ++K ++{ ++k ++; ++ ++k ++K ++; ++K ++k ++K ++ ++ ++K ++ ++k ++k ++k ++ ++ ++K ++; ++ ++K ++[ +++ ++ ++ +++ ++ +++ ++[ +++ ++ ++ ++K ++ ++[ ++; ++[ ++k ++[ ++k ++K ++ ++ ++ ++K ++{ +++ ++ ++ ++[ +@  + +++ ++ ++[ ++; ++ ++[ ++K ++k +<+; ++[ ++{ ++ ++[ ++{ ++ ++; +++ ++ ++; + *. & 04 , #'  "  +/ ' #    &* " 15 - *. &   (, $ (, $ $(   +/ ' (, $ %) ! +/ ' #  %) ! .2 * &* " !%  (, $ +/ ' *. & -1 ) &* " .2 * #'  )- % &* " 04 , ,0 ( *. & ,0 ( *. & "&  /3 + 48 0 04 ,  " +[ +++ ++ ++[ ++ ++[ ++ ++K ++k ++ ++k ++K ++ ++ ++k ++ ++[ ++; ++; ++; ++K ++ ++k ++ ++k ++; +++ ++; ++ ++k ++; ++ +++ ++ ++{ ++K ++ ++K ++K ++ ++ +++ ++ ++ ++k +++ ++K +++ ++ ++[ ++ ++k +++ ++K ++K ++K ++[ ++; ++; ++K +++ ++K ++ ++k ++k ++; ++ ++k ++; ++ +++ ++K +:+k ++k ++k +E+; ++; ++ ++k ++K ++ ++k ++[ ++{ +>++ ++{ ++; ++ ++K ++k +++ ++ ++[ ++K ++k ++K +++ ++ ++; ++k ++; ++ +++ ++; ++ ++[ ++{ ++[ +++ ++ +++ ++K ++k ++ ++k ++; ++; ++[ +++ ++[ ++ ++{ ++ +++ ++{ ++K ++; ++[ +++ ++K ++{ ++ ++; ++; +++ ++K ++k +++ +++ ++[ ++[ ++; ++k ++; ++ ++{ ++ ++ ++; ++ ++[ ++K ++k +<++ ++{ ++ ++; ++{ ++ ++; ++[ ++{ ++{ ++[ ++[ +++ ++K ++k ++K ++; +++ ++K ++k ++K ++ ++k ++K ++[ ++{ ++ ++; ++K ++K ++ +++ +++ ++K ++; ++[ ++{ +++ ++K ++k +++ ++{ ++[ ++{ ++K +++ ++[ +++ ++{ ++k ++ +>+ ++{ ++k ++ ++; ++ ++ +++ +++ ++K ++k ++k ++ +++ ++K +++ ++ ++; ++ ++; ++ ++K ++{ ++ ++; ++[ ++ ++ ++k ++ ++k ++; +++ ++K ++[ ++k ++K ++k ++ +++ ++K ++; ++K ++ ++k ++[ ++{ ++[ ++; ++k ++; ++ ++{ ++ ++[ ++{ ++ ++; ++ ++K ++k ++; ++k ++; +++ ++K ++{ +++ ++K ++k ++ ++[ ++{ ++{ ++ ++; ++ ++[ ++K ++k ++{ ++K ++k ++ ++ +++ ++[ ++{ ++[ ++[ ++K ++; ++[ ++ ++ ++k ++ ++k ++; +++ ++K ++k +++ ++ ++K +++ ++ ++; ++ ++{ ++{ ++K +++ ++{ ++K ++; ++[ ++[ ++; ++{ ++ ++; ++ ++ ++K ++ ++k ++; ++[ ++{ +=+k ++{ ++ ++[ ++K ++{ ++K +;+{ ++ ++ ++K ++; ++[ +++ ++; ++[ ++{ ++ +@+ ++; +++ ++K +++ ++ ++; ++ ++{ ++ ++[ ++[ ++ ++k ++; ++[ ++ ++[ ++{ ++K +++ ++K ++k ++K ++; ++[ ++{ ++[ ++K ++k ++k ++K ++{ ++K ++ ++ +++ ++{ ++ +++ ++K ++; ++[ ++K ++; +++ ++k ++; ++ ++; ++; +++ ++K ++; ++; ++ ++k ++K ++ ++k ++[ ++{ +=+[ ++{ ++K ++ ++ ++{ ++{ ++K +++ ++{ ++K ++; ++[ ++{ ++{ ++[ ++[ +++ ++ ++; +++ ++; ++ ++[ ++K ++k ++{ ++ ++; ++K +++ ++K ++ +++ ++[ ++; +++ ++{ ++k ++ ++ ++K ++; ++ ++; ++[ ++[ ++{ ++ ++k ++k ++; ++ ++k ++; +++ ++K ++ ++{ ++{ ++k ++ ++[ +++ ++K ++ +++ ++ ++[ ++; ++ ++[ ++K ++k +<+ ++K ++ ++{ ++K ++; ++[ +++ ++ ++[ ++K ++k ++{ ++ ++; ++{ ++{ ++[ +++ ++ ++; ++{ +++ ++K ++[ ++ ++k ++; ++ +++ ++K +:+k ++K ++ ++ +++ +++ ++; ++ ++[ ++K ++k +=+; ++K ++ ++k ++[ ++{ ++; ++[ ++{ ++ ++ ++k +++ ++K ++; ++[ ++{ ++k ++ +++ ++ ++; ++[ ++k ++ +++ ++K ++k ++K ++K ++; ++[ ++[ ++ ++K ++k ++ ++{ +++ ++{ ++K ++; ++[ +++ ++[ ++{ ++ ++ ++k ++ +++ ++ ++k ++ +++ ++K ++; ++[ ++{ ++{ ++k ++ +>+k ++k ++ ++[ ++K ++ ++ +++ ++ ++k ++[ ++ ++{ ++; ++[ +++ ++ ++K ++k ++ ++[ ++{ ++K ++[ +++ ++{ ++k ++ ++{ ++K ++ ++ +++ ++{ ++K ++ ++ +++ ++K ++ ++[ ++ ++; ++K ++{ ++; ++[ ++ ++k ++; +++ ++K ++ +++ ++{ ++K ++; ++[ ++k ++[ +++ ++ ++{ ++k ++{ ++k ++k ++K ++{ ++K ++ ++ +++ ++; ++{ ++K ++k ++K ++ ++ ++ ++k ++ ++k ++; +++ ++K ++k ++[ ++; ++K ++k ++; ++{ ++K ++k ++ ++; ++[ +++ ++K ++k +++ ++K ++k ++k ++[ +++ ++{ ++k ++ ++K ++ ++; +++ ++ +++ ++{ ++ ++k +++ ++{ ++; ++[ ++ ++ ++k ++k ++; ++ ++k ++; +++ ++K ++k ++K ++k +++ ++K ++; ++K ++ ++k ++[ ++{ ++; ++K ++ ++k ++[ ++{ +++ ++{ ++k ++ ++ +++ ++k ++K ++K ++ ++{ ++K ++ ++ +++ +@++ ++ ++; +B++ ++; ++ ++; +B+ ++ ++{ ++ +++ ++; ++ ++; ++ ++k ++; ++[ ++ ++{ +++ ++{ ++K ++; ++[ ++{ ++ ++[ +++ ++ ++; ++{ ++[ ++ ++[ +++ ++ ++; ++ ++K ++ ++[ +++ ++ ++; ++K ++ ++k ++ +++ +++ ++{ ++K ++k ++{ ++K ++ ++[ ++ ++; ++K ++[ +++ ++{ ++k ++ ++ ++[ ++{ ++ +++ ++ ++{ +++ ++{ ++K ++; ++[ ++ +++ ++K ++ ++; ++[ ++ ++; ++[ ++[ +++ +++ ++{ ++ +++ ++; ++[ ++{ ++ +@+{ ++ ++[ ++ ++; ++ ++; ++[ ++{ ++ ++ ++K ++k ++ +++ +++ ++ ++; +++ ++; +++ ++ +++ ++K +++ ++k ++{ ++k ++k ++ ++; ++ ++[ ++{ ++ ++k ++ ++k ++{ ++{ ++[ ++; ++ ++; ++{ ++ ++[ ++; ++ ++{ ++ ++{ ++[ +++ ++ ++; ++ ++k ++; +++ ++K +++ ++ ++K ++k ++; +++ ++K ++k ++{ ++; +++ ++{ ++ ++K ++k ++ ++ +++ +++ ++ +++ +++ ++k ++; ++ +++ +++ ++ ++; ++k ++K ++; ++[ ++ +++ ++{ ++K ++; ++[ +++ ++; +++ ++K ++[ +++ ++{ ++k ++ ++ ++[ ++{ ++ ++; ++ ++k +++ ++K ++k ++K ++[ +++ ++k ++ ++ ++[ ++K ++k ++; ++; ++ ++K ++ ++k ++[ ++{ +=+{ ++[ ++K ++k ++; ++; ++ ++K ++ ++k ++[ ++{ ++{ ++{ ++[ ++ ++[ +++ ++ ++; ++; ++; ++ ++k ++K ++ ++k ++[ ++{ +>+K ++K +++ ++[ +++ ++{ ++k ++ +>+ ++K ++k ++[ ++K ++ +++ ++K ++ ++; ++{ ++[ ++; ++; ++ ++; ++ ++; ++k ++; ++{ ++[ ++[ ++; ++; ++[ ++K ++ ++ ++{ ++ ++; ++K ++k +++ ++[ ++ ++; ++; ++k +++ ++K ++[ ++K ++k ++ +++ +++ ++K ++ ++; +++ ++ ++ ++ ++k ++ ++{ ++[ ++k ++ ++{ ++[ ++ ++[ +++ ++ ++; ++K +++ ++{ ++k ++ ++k ++K ++ ++ +++ ++{ +++ +++ ++ ++; ++[ ++; ++; +++ ++K ++{ ++[ ++{ ++{ ++[ ++[ ++K ++k ++{ ++; ++ +++ ++K ++[ ++{ ++; ++{ ++K ++ ++ +++ +8+; ++k ++ ++ +++ ++{ ++; ++[ +;+; ++; ++[ +++ ++ ++; ++[ ++ ++[ ++[ ++ ++; ++ ++[ +++ ++K ++ ++[ ++ ++k ++; +++ ++K ++ ++ ++; ++{ ++ ++ ++; ++; ++k ++{ +++ ++K ++[ ++{ ++{ ++; +++ ++K ++{ ++{ ++k ++ ++[ ++; ++ ++ ++; +++ ++K ++ ++ +++ ++; ++K ++ ++{ ++[ ++[ ++ +++ +++ +++ ++ ++; ++[ ++K ++k ++[ ++K ++k ++{ ++K ++k ++ ++ +++ ++K ++K ++; ++[ ++K ++{ ++K ++{ +++ ++; ++k ++ ++ ++; ++k ++{ +++ ++K ++[ ++ ++ ++; ++K ++{ ++ ++ ++K ++{ ++ ++; ++[ ++k ++; ++ ++[ ++{ ++; ++[ ++{ ++ ++ +++ ++K ++[ ++ ++k ++[ +++ ++ ++; ++; ++k ++k ++K ++{ ++K ++ ++ +++ ++ ++ ++k ++ ++k ++; +++ ++K ++{ +++ ++ ++{ ++K ++ ++; ++[ +;+ ++ ++k ++ ++k ++; +++ ++K ++ ++; ++[ ++ ++k ++ ++k ++; +++ ++K ++ ++{ ++; ++[ ++; ++ +++ ++k +8DBU<BL03+{ +<.+ +?+{ +r   