o
    Gi,                    @   s  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZG d%d& d&edZG d'd( d(edZd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 ZG d5d6 d6edZG d7d8 d8edZG d9d: d:edZG d;d< d<edZG d=d> d>edZ G d?d@ d@edZ!G dAdB dBedZ"G dCdD dDedZ#G dEdF dFedZ$G dGdH dHedZ%G dIdJ dJedZ&G dKdL dLedZ'G dMdN dNedZ(G dOdP dPedZ)G dQdR dRedZ*G dSdT dTedZ+G dUdV dVedZ,G dWdX dXedZ-G dYdZ dZedZ.G d[d\ d\edZ/G d]d^ d^edZ0G d_d` d`edZ1G dadb dbedZ2G dcdd ddedZ3G dedf dfedZ4G dgdh dhedZ5G didj djedZ6G dkdl dledZ7G dmdn dnedZ8G dodp dpedZ9G dqdr dredZ:G dsdt dtedZ;G dudv dvedZ<G dwdx dxedZ=G dydz dzedZ>G d{d| d|edZ?G d}d~ d~edZ@G dd dedZAG dd dedZBG dd dedZCG dd dedZDG dd dedZEG dd dedZFG dd dedZGG dd dedZHG dd dedZIG dd dedZJG dd dedZKG dd dedZLG dd dedZMG dd dedZNG dd dedZOG dd dedZPG dd dedZQG dd dedZRG dd dedZSG dd dedZTG dd dedZUG dd dedZVG dd dedZWG dd dedZXG dd dedZYG dd dedZZG dd dedZ[G dd dedZ\G dd dedZ]G dd dedZ^G dd dedZ_G dd dedZ`G dd dedZaG dd dedZbG ddĄ dedZcG ddƄ dedZdG ddȄ dedZeG ddʄ dedZfG dd̄ dedZgG dd΄ dedZhG ddЄ dedZiG dd҄ dedZjG ddԄ dedZkG ddք dedZlG dd؄ dedZmG ddڄ dedZnG dd܄ dedZoG ddބ dedZpG dd dedZqG dd dedZrG dd dedZsG dd dedZtG dd dedZuG dd dedZvG dd dedZwG dd dedZxG dd dedZyG dd dedZzG dd dedZ{G dd dedZ|G dd dedZ}G dd dedZ~G dd dedZG dd dedZG dd  d edZG dd dedZG dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZdd ZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 ZG d3d4 d4edZG d5d6 d6edZG d7d8 d8edZG d9d: d:edZG d;d< d<edZG d=d> d>edZG d?d@ d@edZG dAdB dBedZG dCdD dDedZG dEdF dFedZG dGdH dHedZG dIdJ dJedZG dKdL dLedZG dMdN dNedZG dOdP dPedZG dQdR dRedZG dSdT dTedZG dUdV dVedZG dWdX dXedZG dYdZ dZedZG d[d\ d\edZG d]d^ d^edZG d_d` d`edZG dadb dbedZG dcdd ddedZG dedf dfedZG dgdh dhedZG didj djedZG dkdl dledZG dmdn dnedZG dodp dpedZG dqdr dredZG dsdt dtedZG dudv dvedZG dwdx dxedZG dydz dzedZG d{d| d|edZG d}d~ d~edZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZِdS (     )DummyObjectrequires_backendsc                   @   2   e Zd ZdgZdd Zedd Zedd ZdS )	AdaptiveProjectedGuidancetorchc                 O      t | dg d S Nr   r   selfargskwargs r   T/home/ubuntu/.local/lib/python3.10/site-packages/diffusers/utils/dummy_pt_objects.py__init__      z"AdaptiveProjectedGuidance.__init__c                 O   r   r   r	   clsr   r   r   r   r   from_config      z%AdaptiveProjectedGuidance.from_configc                 O   r   r   r	   r   r   r   r   from_pretrained   r   z)AdaptiveProjectedGuidance.from_pretrainedN__name__
__module____qualname__	_backendsr   classmethodr   r   r   r   r   r   r          
r   )	metaclassc                   @   r   )	AdaptiveProjectedMixGuidancer   c                 O   r   r   r	   r
   r   r   r   r      r   z%AdaptiveProjectedMixGuidance.__init__c                 O   r   r   r	   r   r   r   r   r      r   z(AdaptiveProjectedMixGuidance.from_configc                 O   r   r   r	   r   r   r   r   r      r   z,AdaptiveProjectedMixGuidance.from_pretrainedNr   r   r   r   r   r      r   r   c                   @   r   )	AutoGuidancer   c                 O   r   r   r	   r
   r   r   r   r   &   r   zAutoGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   )   r   zAutoGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   -   r   zAutoGuidance.from_pretrainedNr   r   r   r   r   r    #   r   r    c                   @   r   )	BaseGuidancer   c                 O   r   r   r	   r
   r   r   r   r   5   r   zBaseGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   8   r   zBaseGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   <   r   zBaseGuidance.from_pretrainedNr   r   r   r   r   r!   2   r   r!   c                   @   r   )	ClassifierFreeGuidancer   c                 O   r   r   r	   r
   r   r   r   r   D   r   zClassifierFreeGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   G   r   z"ClassifierFreeGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   K   r   z&ClassifierFreeGuidance.from_pretrainedNr   r   r   r   r   r"   A   r   r"   c                   @   r   )	ClassifierFreeZeroStarGuidancer   c                 O   r   r   r	   r
   r   r   r   r   S   r   z'ClassifierFreeZeroStarGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   V   r   z*ClassifierFreeZeroStarGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   Z   r   z.ClassifierFreeZeroStarGuidance.from_pretrainedNr   r   r   r   r   r#   P   r   r#   c                   @   r   )	FrequencyDecoupledGuidancer   c                 O   r   r   r	   r
   r   r   r   r   b   r   z#FrequencyDecoupledGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   e   r   z&FrequencyDecoupledGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   i   r   z*FrequencyDecoupledGuidance.from_pretrainedNr   r   r   r   r   r$   _   r   r$   c                   @   r   )	PerturbedAttentionGuidancer   c                 O   r   r   r	   r
   r   r   r   r   q   r   z#PerturbedAttentionGuidance.__init__c                 O   r   r   r	   r   r   r   r   r   t   r   z&PerturbedAttentionGuidance.from_configc                 O   r   r   r	   r   r   r   r   r   x   r   z*PerturbedAttentionGuidance.from_pretrainedNr   r   r   r   r   r%   n   r   r%   c                   @   r   )	SkipLayerGuidancer   c                 O   r   r   r	   r
   r   r   r   r      r   zSkipLayerGuidance.__init__c                 O   r   r   r	   r   r   r   r   r      r   zSkipLayerGuidance.from_configc                 O   r   r   r	   r   r   r   r   r      r   z!SkipLayerGuidance.from_pretrainedNr   r   r   r   r   r&   }   r   r&   c                   @   r   )	SmoothedEnergyGuidancer   c                 O   r   r   r	   r
   r   r   r   r      r   zSmoothedEnergyGuidance.__init__c                 O   r   r   r	   r   r   r   r   r      r   z"SmoothedEnergyGuidance.from_configc                 O   r   r   r	   r   r   r   r   r      r   z&SmoothedEnergyGuidance.from_pretrainedNr   r   r   r   r   r'      r   r'   c                   @   r   )	 TangentialClassifierFreeGuidancer   c                 O   r   r   r	   r
   r   r   r   r      r   z)TangentialClassifierFreeGuidance.__init__c                 O   r   r   r	   r   r   r   r   r      r   z,TangentialClassifierFreeGuidance.from_configc                 O   r   r   r	   r   r   r   r   r      r   z0TangentialClassifierFreeGuidance.from_pretrainedNr   r   r   r   r   r(      r   r(   c                   @   r   )	FasterCacheConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zFasterCacheConfig.__init__c                 O   r   r   r	   r   r   r   r   r      r   zFasterCacheConfig.from_configc                 O   r   r   r	   r   r   r   r   r      r   z!FasterCacheConfig.from_pretrainedNr   r   r   r   r   r)      r   r)   c                   @   r   )	FirstBlockCacheConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zFirstBlockCacheConfig.__init__c                 O   r   r   r	   r   r   r   r   r      r   z!FirstBlockCacheConfig.from_configc                 O   r   r   r	   r   r   r   r   r      r   z%FirstBlockCacheConfig.from_pretrainedNr   r   r   r   r   r*      r   r*   c                   @   r   )	HookRegistryr   c                 O   r   r   r	   r
   r   r   r   r      r   zHookRegistry.__init__c                 O   r   r   r	   r   r   r   r   r      r   zHookRegistry.from_configc                 O   r   r   r	   r   r   r   r   r      r   zHookRegistry.from_pretrainedNr   r   r   r   r   r+      r   r+   c                   @   r   )	LayerSkipConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zLayerSkipConfig.__init__c                 O   r   r   r	   r   r   r   r   r      r   zLayerSkipConfig.from_configc                 O   r   r   r	   r   r   r   r   r      r   zLayerSkipConfig.from_pretrainedNr   r   r   r   r   r,      r   r,   c                   @   r   )	MagCacheConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   zMagCacheConfig.__init__c                 O   r   r   r	   r   r   r   r   r      r   zMagCacheConfig.from_configc                 O   r   r   r	   r   r   r   r   r      r   zMagCacheConfig.from_pretrainedNr   r   r   r   r   r-      r   r-   c                   @   r   )	PyramidAttentionBroadcastConfigr   c                 O   r   r   r	   r
   r   r   r   r      r   z(PyramidAttentionBroadcastConfig.__init__c                 O   r   r   r	   r   r   r   r   r      r   z+PyramidAttentionBroadcastConfig.from_configc                 O   r   r   r	   r   r   r   r   r      r   z/PyramidAttentionBroadcastConfig.from_pretrainedNr   r   r   r   r   r.      r   r.   c                   @   r   )	SmoothedEnergyGuidanceConfigr   c                 O   r   r   r	   r
   r   r   r   r     r   z%SmoothedEnergyGuidanceConfig.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z(SmoothedEnergyGuidanceConfig.from_configc                 O   r   r   r	   r   r   r   r   r     r   z,SmoothedEnergyGuidanceConfig.from_pretrainedNr   r   r   r   r   r/     r   r/   c                   @   r   )	TaylorSeerCacheConfigr   c                 O   r   r   r	   r
   r   r   r   r     r   zTaylorSeerCacheConfig.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!TaylorSeerCacheConfig.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%TaylorSeerCacheConfig.from_pretrainedNr   r   r   r   r   r0     r   r0   c                  O      t tdg d S r   )r   apply_faster_cacher   r   r   r   r   r2   "  r   r2   c                  O   r1   r   )r   apply_first_block_cacher3   r   r   r   r4   &  r   r4   c                  O   r1   r   )r   apply_layer_skipr3   r   r   r   r5   *  r   r5   c                  O   r1   r   )r   apply_mag_cacher3   r   r   r   r6   .  r   r6   c                  O   r1   r   )r   !apply_pyramid_attention_broadcastr3   r   r   r   r7   2  r   r7   c                  O   r1   r   )r   apply_taylorseer_cacher3   r   r   r   r8   6  r   r8   c                   @   r   )	AllegroTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   =  r   z"AllegroTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   @  r   z%AllegroTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   D  r   z)AllegroTransformer3DModel.from_pretrainedNr   r   r   r   r   r9   :  r   r9   c                   @   r   )	AsymmetricAutoencoderKLr   c                 O   r   r   r	   r
   r   r   r   r   L  r   z AsymmetricAutoencoderKL.__init__c                 O   r   r   r	   r   r   r   r   r   O  r   z#AsymmetricAutoencoderKL.from_configc                 O   r   r   r	   r   r   r   r   r   S  r   z'AsymmetricAutoencoderKL.from_pretrainedNr   r   r   r   r   r:   I  r   r:   c                   @   r   )	AttentionBackendNamer   c                 O   r   r   r	   r
   r   r   r   r   [  r   zAttentionBackendName.__init__c                 O   r   r   r	   r   r   r   r   r   ^  r   z AttentionBackendName.from_configc                 O   r   r   r	   r   r   r   r   r   b  r   z$AttentionBackendName.from_pretrainedNr   r   r   r   r   r;   X  r   r;   c                   @   r   )	AuraFlowTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   j  r   z#AuraFlowTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   m  r   z&AuraFlowTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   q  r   z*AuraFlowTransformer2DModel.from_pretrainedNr   r   r   r   r   r<   g  r   r<   c                   @   r   )	AutoencoderDCr   c                 O   r   r   r	   r
   r   r   r   r   y  r   zAutoencoderDC.__init__c                 O   r   r   r	   r   r   r   r   r   |  r   zAutoencoderDC.from_configc                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderDC.from_pretrainedNr   r   r   r   r   r=   v  r   r=   c                   @   r   )	AutoencoderKLr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKL.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderKL.from_configc                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderKL.from_pretrainedNr   r   r   r   r   r>     r   r>   c                   @   r   )	AutoencoderKLAllegror   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLAllegro.__init__c                 O   r   r   r	   r   r   r   r   r     r   z AutoencoderKLAllegro.from_configc                 O   r   r   r	   r   r   r   r   r     r   z$AutoencoderKLAllegro.from_pretrainedNr   r   r   r   r   r?     r   r?   c                   @   r   )	AutoencoderKLCogVideoXr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLCogVideoX.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"AutoencoderKLCogVideoX.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&AutoencoderKLCogVideoX.from_pretrainedNr   r   r   r   r   r@     r   r@   c                   @   r   )	AutoencoderKLCosmosr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLCosmos.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderKLCosmos.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#AutoencoderKLCosmos.from_pretrainedNr   r   r   r   r   rA     r   rA   c                   @   r   )	AutoencoderKLFlux2r   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLFlux2.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderKLFlux2.from_configc                 O   r   r   r	   r   r   r   r   r     r   z"AutoencoderKLFlux2.from_pretrainedNr   r   r   r   r   rB     r   rB   c                   @   r   )	AutoencoderKLHunyuanImager   c                 O   r   r   r	   r
   r   r   r   r     r   z"AutoencoderKLHunyuanImage.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%AutoencoderKLHunyuanImage.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)AutoencoderKLHunyuanImage.from_pretrainedNr   r   r   r   r   rC     r   rC   c                   @   r   )	 AutoencoderKLHunyuanImageRefinerr   c                 O   r   r   r	   r
   r   r   r   r     r   z)AutoencoderKLHunyuanImageRefiner.__init__c                 O   r   r   r	   r   r   r   r   r     r   z,AutoencoderKLHunyuanImageRefiner.from_configc                 O   r   r   r	   r   r   r   r   r     r   z0AutoencoderKLHunyuanImageRefiner.from_pretrainedNr   r   r   r   r   rD     r   rD   c                   @   r   )	AutoencoderKLHunyuanVideor   c                 O   r   r   r	   r
   r   r   r   r     r   z"AutoencoderKLHunyuanVideo.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%AutoencoderKLHunyuanVideo.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)AutoencoderKLHunyuanVideo.from_pretrainedNr   r   r   r   r   rE     r   rE   c                   @   r   )	AutoencoderKLHunyuanVideo15r   c                 O   r   r   r	   r
   r   r   r   r      r   z$AutoencoderKLHunyuanVideo15.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'AutoencoderKLHunyuanVideo15.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+AutoencoderKLHunyuanVideo15.from_pretrainedNr   r   r   r   r   rF     r   rF   c                   @   r   )	AutoencoderKLLTX2Audior   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLLTX2Audio.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"AutoencoderKLLTX2Audio.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&AutoencoderKLLTX2Audio.from_pretrainedNr   r   r   r   r   rG     r   rG   c                   @   r   )	AutoencoderKLLTX2Videor   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderKLLTX2Video.__init__c                 O   r   r   r	   r   r   r   r   r   !  r   z"AutoencoderKLLTX2Video.from_configc                 O   r   r   r	   r   r   r   r   r   %  r   z&AutoencoderKLLTX2Video.from_pretrainedNr   r   r   r   r   rH     r   rH   c                   @   r   )	AutoencoderKLLTXVideor   c                 O   r   r   r	   r
   r   r   r   r   -  r   zAutoencoderKLLTXVideo.__init__c                 O   r   r   r	   r   r   r   r   r   0  r   z!AutoencoderKLLTXVideo.from_configc                 O   r   r   r	   r   r   r   r   r   4  r   z%AutoencoderKLLTXVideo.from_pretrainedNr   r   r   r   r   rI   *  r   rI   c                   @   r   )	AutoencoderKLMagvitr   c                 O   r   r   r	   r
   r   r   r   r   <  r   zAutoencoderKLMagvit.__init__c                 O   r   r   r	   r   r   r   r   r   ?  r   zAutoencoderKLMagvit.from_configc                 O   r   r   r	   r   r   r   r   r   C  r   z#AutoencoderKLMagvit.from_pretrainedNr   r   r   r   r   rJ   9  r   rJ   c                   @   r   )	AutoencoderKLMochir   c                 O   r   r   r	   r
   r   r   r   r   K  r   zAutoencoderKLMochi.__init__c                 O   r   r   r	   r   r   r   r   r   N  r   zAutoencoderKLMochi.from_configc                 O   r   r   r	   r   r   r   r   r   R  r   z"AutoencoderKLMochi.from_pretrainedNr   r   r   r   r   rK   H  r   rK   c                   @   r   )	AutoencoderKLQwenImager   c                 O   r   r   r	   r
   r   r   r   r   Z  r   zAutoencoderKLQwenImage.__init__c                 O   r   r   r	   r   r   r   r   r   ]  r   z"AutoencoderKLQwenImage.from_configc                 O   r   r   r	   r   r   r   r   r   a  r   z&AutoencoderKLQwenImage.from_pretrainedNr   r   r   r   r   rL   W  r   rL   c                   @   r   )	AutoencoderKLTemporalDecoderr   c                 O   r   r   r	   r
   r   r   r   r   i  r   z%AutoencoderKLTemporalDecoder.__init__c                 O   r   r   r	   r   r   r   r   r   l  r   z(AutoencoderKLTemporalDecoder.from_configc                 O   r   r   r	   r   r   r   r   r   p  r   z,AutoencoderKLTemporalDecoder.from_pretrainedNr   r   r   r   r   rM   f  r   rM   c                   @   r   )	AutoencoderKLWanr   c                 O   r   r   r	   r
   r   r   r   r   x  r   zAutoencoderKLWan.__init__c                 O   r   r   r	   r   r   r   r   r   {  r   zAutoencoderKLWan.from_configc                 O   r   r   r	   r   r   r   r   r     r   z AutoencoderKLWan.from_pretrainedNr   r   r   r   r   rN   u  r   rN   c                   @   r   )	AutoencoderOobleckr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderOobleck.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderOobleck.from_configc                 O   r   r   r	   r   r   r   r   r     r   z"AutoencoderOobleck.from_pretrainedNr   r   r   r   r   rO     r   rO   c                   @   r   )	AutoencoderRAEr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderRAE.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderRAE.from_configc                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderRAE.from_pretrainedNr   r   r   r   r   rP     r   rP   c                   @   r   )	AutoencoderTinyr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoencoderTiny.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderTiny.from_configc                 O   r   r   r	   r   r   r   r   r     r   zAutoencoderTiny.from_pretrainedNr   r   r   r   r   rQ     r   rQ   c                   @   r   )		AutoModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   zAutoModel.from_pretrainedNr   r   r   r   r   rR     r   rR   c                   @   r   )	BriaFiboTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#BriaFiboTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z&BriaFiboTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z*BriaFiboTransformer2DModel.from_pretrainedNr   r   r   r   r   rS     r   rS   c                   @   r   )	BriaTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zBriaTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"BriaTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&BriaTransformer2DModel.from_pretrainedNr   r   r   r   r   rT     r   rT   c                   @   r   )	
CacheMixinr   c                 O   r   r   r	   r
   r   r   r   r     r   zCacheMixin.__init__c                 O   r   r   r	   r   r   r   r   r     r   zCacheMixin.from_configc                 O   r   r   r	   r   r   r   r   r     r   zCacheMixin.from_pretrainedNr   r   r   r   r   rU     r   rU   c                   @   r   )	ChromaTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!ChromaTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$ChromaTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(ChromaTransformer2DModel.from_pretrainedNr   r   r   r   r   rV     r   rV   c                   @   r   )	ChronoEditTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z%ChronoEditTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z(ChronoEditTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z,ChronoEditTransformer3DModel.from_pretrainedNr   r   r   r   r   rW     r   rW   c                   @   r   )	CogVideoXTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$CogVideoXTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'CogVideoXTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+CogVideoXTransformer3DModel.from_pretrainedNr   r   r   r   r   rX     r   rX   c                   @   r   )	CogView3PlusTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z'CogView3PlusTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r      r   z*CogView3PlusTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   $  r   z.CogView3PlusTransformer2DModel.from_pretrainedNr   r   r   r   r   rY     r   rY   c                   @   r   )	CogView4Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   ,  r   z#CogView4Transformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   /  r   z&CogView4Transformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   3  r   z*CogView4Transformer2DModel.from_pretrainedNr   r   r   r   r   rZ   )  r   rZ   c                   @   r   )	ConsisIDTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   ;  r   z#ConsisIDTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   >  r   z&ConsisIDTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   B  r   z*ConsisIDTransformer3DModel.from_pretrainedNr   r   r   r   r   r[   8  r   r[   c                   @   r   )	ConsistencyDecoderVAEr   c                 O   r   r   r	   r
   r   r   r   r   J  r   zConsistencyDecoderVAE.__init__c                 O   r   r   r	   r   r   r   r   r   M  r   z!ConsistencyDecoderVAE.from_configc                 O   r   r   r	   r   r   r   r   r   Q  r   z%ConsistencyDecoderVAE.from_pretrainedNr   r   r   r   r   r\   G  r   r\   c                   @   r   )	ContextParallelConfigr   c                 O   r   r   r	   r
   r   r   r   r   Y  r   zContextParallelConfig.__init__c                 O   r   r   r	   r   r   r   r   r   \  r   z!ContextParallelConfig.from_configc                 O   r   r   r	   r   r   r   r   r   `  r   z%ContextParallelConfig.from_pretrainedNr   r   r   r   r   r]   V  r   r]   c                   @   r   )	ControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   h  r   zControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   k  r   zControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   o  r   zControlNetModel.from_pretrainedNr   r   r   r   r   r^   e  r   r^   c                   @   r   )	ControlNetUnionModelr   c                 O   r   r   r	   r
   r   r   r   r   w  r   zControlNetUnionModel.__init__c                 O   r   r   r	   r   r   r   r   r   z  r   z ControlNetUnionModel.from_configc                 O   r   r   r	   r   r   r   r   r   ~  r   z$ControlNetUnionModel.from_pretrainedNr   r   r   r   r   r_   t  r   r_   c                   @   r   )	ControlNetXSAdapterr   c                 O   r   r   r	   r
   r   r   r   r     r   zControlNetXSAdapter.__init__c                 O   r   r   r	   r   r   r   r   r     r   zControlNetXSAdapter.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#ControlNetXSAdapter.from_pretrainedNr   r   r   r   r   r`     r   r`   c                   @   r   )	CosmosControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zCosmosControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!CosmosControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%CosmosControlNetModel.from_pretrainedNr   r   r   r   r   ra     r   ra   c                   @   r   )	CosmosTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!CosmosTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$CosmosTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(CosmosTransformer3DModel.from_pretrainedNr   r   r   r   r   rb     r   rb   c                   @   r   )	DiTTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zDiTTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!DiTTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%DiTTransformer2DModel.from_pretrainedNr   r   r   r   r   rc     r   rc   c                   @   r   )	EasyAnimateTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&EasyAnimateTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z)EasyAnimateTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z-EasyAnimateTransformer3DModel.from_pretrainedNr   r   r   r   r   rd     r   rd   c                   @   r   )	Flux2Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z Flux2Transformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z#Flux2Transformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z'Flux2Transformer2DModel.from_pretrainedNr   r   r   r   r   re     r   re   c                   @   r   )	FluxControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFluxControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zFluxControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#FluxControlNetModel.from_pretrainedNr   r   r   r   r   rf     r   rf   c                   @   r   )	FluxMultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!FluxMultiControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$FluxMultiControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(FluxMultiControlNetModel.from_pretrainedNr   r   r   r   r   rg     r   rg   c                   @   r   )	FluxTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zFluxTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"FluxTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&FluxTransformer2DModel.from_pretrainedNr   r   r   r   r   rh     r   rh   c                   @   r   )	GlmImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#GlmImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z&GlmImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z*GlmImageTransformer2DModel.from_pretrainedNr   r   r   r   r   ri   
  r   ri   c                   @   r   )	HeliosTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!HeliosTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$HeliosTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   #  r   z(HeliosTransformer3DModel.from_pretrainedNr   r   r   r   r   rj     r   rj   c                   @   r   )	HiDreamImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   +  r   z'HiDreamImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   .  r   z*HiDreamImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   2  r   z.HiDreamImageTransformer2DModel.from_pretrainedNr   r   r   r   r   rk   (  r   rk   c                   @   r   )	HunyuanDiT2DControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   :  r   z$HunyuanDiT2DControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   =  r   z'HunyuanDiT2DControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   A  r   z+HunyuanDiT2DControlNetModel.from_pretrainedNr   r   r   r   r   rl   7  r   rl   c                   @   r   )	HunyuanDiT2DModelr   c                 O   r   r   r	   r
   r   r   r   r   I  r   zHunyuanDiT2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   L  r   zHunyuanDiT2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   P  r   z!HunyuanDiT2DModel.from_pretrainedNr   r   r   r   r   rm   F  r   rm   c                   @   r   )	 HunyuanDiT2DMultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   X  r   z)HunyuanDiT2DMultiControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   [  r   z,HunyuanDiT2DMultiControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   _  r   z0HunyuanDiT2DMultiControlNetModel.from_pretrainedNr   r   r   r   r   rn   U  r   rn   c                   @   r   )	HunyuanImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   g  r   z'HunyuanImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   j  r   z*HunyuanImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   n  r   z.HunyuanImageTransformer2DModel.from_pretrainedNr   r   r   r   r   ro   d  r   ro   c                   @   r   )	 HunyuanVideo15Transformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   v  r   z)HunyuanVideo15Transformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   y  r   z,HunyuanVideo15Transformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   }  r   z0HunyuanVideo15Transformer3DModel.from_pretrainedNr   r   r   r   r   rp   s  r   rp   c                   @   r   )	'HunyuanVideoFramepackTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z0HunyuanVideoFramepackTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z3HunyuanVideoFramepackTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z7HunyuanVideoFramepackTransformer3DModel.from_pretrainedNr   r   r   r   r   rq     r   rq   c                   @   r   )	HunyuanVideoTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z'HunyuanVideoTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z*HunyuanVideoTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z.HunyuanVideoTransformer3DModel.from_pretrainedNr   r   r   r   r   rr     r   rr   c                   @   r   )	I2VGenXLUNetr   c                 O   r   r   r	   r
   r   r   r   r     r   zI2VGenXLUNet.__init__c                 O   r   r   r	   r   r   r   r   r     r   zI2VGenXLUNet.from_configc                 O   r   r   r	   r   r   r   r   r     r   zI2VGenXLUNet.from_pretrainedNr   r   r   r   r   rs     r   rs   c                   @   r   )	Kandinsky3UNetr   c                 O   r   r   r	   r
   r   r   r   r     r   zKandinsky3UNet.__init__c                 O   r   r   r	   r   r   r   r   r     r   zKandinsky3UNet.from_configc                 O   r   r   r	   r   r   r   r   r     r   zKandinsky3UNet.from_pretrainedNr   r   r   r   r   rt     r   rt   c                   @   r   )	Kandinsky5Transformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z%Kandinsky5Transformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z(Kandinsky5Transformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z,Kandinsky5Transformer3DModel.from_pretrainedNr   r   r   r   r   ru     r   ru   c                   @   r   )	LatteTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z LatteTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z#LatteTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z'LatteTransformer3DModel.from_pretrainedNr   r   r   r   r   rv     r   rv   c                   @   r   )	LongCatImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z'LongCatImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z*LongCatImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z.LongCatImageTransformer2DModel.from_pretrainedNr   r   r   r   r   rw     r   rw   c                   @   r   )	LTX2VideoTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$LTX2VideoTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'LTX2VideoTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+LTX2VideoTransformer3DModel.from_pretrainedNr   r   r   r   r   rx     r   rx   c                   @   r   )	LTXVideoTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z#LTXVideoTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r      r   z&LTXVideoTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z*LTXVideoTransformer3DModel.from_pretrainedNr   r   r   r   r   ry     r   ry   c                   @   r   )	Lumina2Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z"Lumina2Transformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%Lumina2Transformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)Lumina2Transformer2DModel.from_pretrainedNr   r   r   r   r   rz   	  r   rz   c                   @   r   )	LuminaNextDiT2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zLuminaNextDiT2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z LuminaNextDiT2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   "  r   z$LuminaNextDiT2DModel.from_pretrainedNr   r   r   r   r   r{     r   r{   c                   @   r   )	MochiTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   *  r   z MochiTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   -  r   z#MochiTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   1  r   z'MochiTransformer3DModel.from_pretrainedNr   r   r   r   r   r|   '  r   r|   c                   @   r   )	
ModelMixinr   c                 O   r   r   r	   r
   r   r   r   r   9  r   zModelMixin.__init__c                 O   r   r   r	   r   r   r   r   r   <  r   zModelMixin.from_configc                 O   r   r   r	   r   r   r   r   r   @  r   zModelMixin.from_pretrainedNr   r   r   r   r   r}   6  r   r}   c                   @   r   )	MotionAdapterr   c                 O   r   r   r	   r
   r   r   r   r   H  r   zMotionAdapter.__init__c                 O   r   r   r	   r   r   r   r   r   K  r   zMotionAdapter.from_configc                 O   r   r   r	   r   r   r   r   r   O  r   zMotionAdapter.from_pretrainedNr   r   r   r   r   r~   E  r   r~   c                   @   r   )	MultiAdapterr   c                 O   r   r   r	   r
   r   r   r   r   W  r   zMultiAdapter.__init__c                 O   r   r   r	   r   r   r   r   r   Z  r   zMultiAdapter.from_configc                 O   r   r   r	   r   r   r   r   r   ^  r   zMultiAdapter.from_pretrainedNr   r   r   r   r   r   T  r   r   c                   @   r   )	MultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   f  r   zMultiControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   i  r   z MultiControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   m  r   z$MultiControlNetModel.from_pretrainedNr   r   r   r   r   r   c  r   r   c                   @   r   )	OmniGenTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   u  r   z"OmniGenTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   x  r   z%OmniGenTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   |  r   z)OmniGenTransformer2DModel.from_pretrainedNr   r   r   r   r   r   r  r   r   c                   @   r   )	OvisImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$OvisImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'OvisImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+OvisImageTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ParallelConfigr   c                 O   r   r   r	   r
   r   r   r   r     r   zParallelConfig.__init__c                 O   r   r   r	   r   r   r   r   r     r   zParallelConfig.from_configc                 O   r   r   r	   r   r   r   r   r     r   zParallelConfig.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	PixArtTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!PixArtTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$PixArtTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(PixArtTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	PriorTransformerr   c                 O   r   r   r	   r
   r   r   r   r     r   zPriorTransformer.__init__c                 O   r   r   r	   r   r   r   r   r     r   zPriorTransformer.from_configc                 O   r   r   r	   r   r   r   r   r     r   z PriorTransformer.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	PRXTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zPRXTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!PRXTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%PRXTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	QwenImageControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!QwenImageControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$QwenImageControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(QwenImageControlNetModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	QwenImageMultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z&QwenImageMultiControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z)QwenImageMultiControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z-QwenImageMultiControlNetModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	QwenImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$QwenImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'QwenImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+QwenImageTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SanaControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zSanaControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zSanaControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#SanaControlNetModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SanaTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zSanaTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"SanaTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&SanaTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SanaVideoTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z$SanaVideoTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'SanaVideoTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   !  r   z+SanaVideoTransformer3DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SD3ControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   )  r   zSD3ControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   ,  r   zSD3ControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   0  r   z"SD3ControlNetModel.from_pretrainedNr   r   r   r   r   r   &  r   r   c                   @   r   )	SD3MultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   8  r   z SD3MultiControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   ;  r   z#SD3MultiControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   ?  r   z'SD3MultiControlNetModel.from_pretrainedNr   r   r   r   r   r   5  r   r   c                   @   r   )	SD3Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   G  r   zSD3Transformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   J  r   z!SD3Transformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   N  r   z%SD3Transformer2DModel.from_pretrainedNr   r   r   r   r   r   D  r   r   c                   @   r   )	SkyReelsV2Transformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   V  r   z%SkyReelsV2Transformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   Y  r   z(SkyReelsV2Transformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   ]  r   z,SkyReelsV2Transformer3DModel.from_pretrainedNr   r   r   r   r   r   S  r   r   c                   @   r   )	SparseControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   e  r   zSparseControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   h  r   z!SparseControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   l  r   z%SparseControlNetModel.from_pretrainedNr   r   r   r   r   r   b  r   r   c                   @   r   )	StableAudioDiTModelr   c                 O   r   r   r	   r
   r   r   r   r   t  r   zStableAudioDiTModel.__init__c                 O   r   r   r	   r   r   r   r   r   w  r   zStableAudioDiTModel.from_configc                 O   r   r   r	   r   r   r   r   r   {  r   z#StableAudioDiTModel.from_pretrainedNr   r   r   r   r   r   q  r   r   c                   @   r   )	
T2IAdapterr   c                 O   r   r   r	   r
   r   r   r   r     r   zT2IAdapter.__init__c                 O   r   r   r	   r   r   r   r   r     r   zT2IAdapter.from_configc                 O   r   r   r	   r   r   r   r   r     r   zT2IAdapter.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	T5FilmDecoderr   c                 O   r   r   r	   r
   r   r   r   r     r   zT5FilmDecoder.__init__c                 O   r   r   r	   r   r   r   r   r     r   zT5FilmDecoder.from_configc                 O   r   r   r	   r   r   r   r   r     r   zT5FilmDecoder.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z"Transformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	TransformerTemporalModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!TransformerTemporalModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$TransformerTemporalModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(TransformerTemporalModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNet1DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zUNet1DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zUNet1DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   zUNet1DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNet2DConditionModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zUNet2DConditionModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z UNet2DConditionModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z$UNet2DConditionModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNet2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zUNet2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zUNet2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   zUNet2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNet3DConditionModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zUNet3DConditionModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z UNet3DConditionModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z$UNet3DConditionModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNetControlNetXSModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zUNetControlNetXSModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!UNetControlNetXSModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%UNetControlNetXSModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UNetMotionModelr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zUNetMotionModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zUNetMotionModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   zUNetMotionModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	 UNetSpatioTemporalConditionModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z)UNetSpatioTemporalConditionModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z,UNetSpatioTemporalConditionModel.from_configc                 O   r   r   r	   r   r   r   r   r      r   z0UNetSpatioTemporalConditionModel.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UVit2DModelr   c                 O   r   r   r	   r
   r   r   r   r   (  r   zUVit2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   +  r   zUVit2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   /  r   zUVit2DModel.from_pretrainedNr   r   r   r   r   r   %  r   r   c                   @   r   )	VQModelr   c                 O   r   r   r	   r
   r   r   r   r   7  r   zVQModel.__init__c                 O   r   r   r	   r   r   r   r   r   :  r   zVQModel.from_configc                 O   r   r   r	   r   r   r   r   r   >  r   zVQModel.from_pretrainedNr   r   r   r   r   r   4  r   r   c                   @   r   )	WanAnimateTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   F  r   z%WanAnimateTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   I  r   z(WanAnimateTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   M  r   z,WanAnimateTransformer3DModel.from_pretrainedNr   r   r   r   r   r   C  r   r   c                   @   r   )	WanTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   U  r   zWanTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   X  r   z!WanTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   \  r   z%WanTransformer3DModel.from_pretrainedNr   r   r   r   r   r   R  r   r   c                   @   r   )	WanVACETransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   d  r   z"WanVACETransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   g  r   z%WanVACETransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   k  r   z)WanVACETransformer3DModel.from_pretrainedNr   r   r   r   r   r   a  r   r   c                   @   r   )	ZImageControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r   s  r   zZImageControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r   v  r   z!ZImageControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r   z  r   z%ZImageControlNetModel.from_pretrainedNr   r   r   r   r   r   p  r   r   c                   @   r   )	ZImageTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   z!ZImageTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$ZImageTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(ZImageTransformer2DModel.from_pretrainedNr   r   r   r   r   r     r   r   c                  O   r1   r   )r   attention_backendr3   r   r   r   r     r   r   c                   @   r   )	AutoPipelineBlocksr   c                 O   r   r   r	   r
   r   r   r   r     r   zAutoPipelineBlocks.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAutoPipelineBlocks.from_configc                 O   r   r   r	   r   r   r   r   r     r   z"AutoPipelineBlocks.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ComponentsManagerr   c                 O   r   r   r	   r
   r   r   r   r     r   zComponentsManager.__init__c                 O   r   r   r	   r   r   r   r   r     r   zComponentsManager.from_configc                 O   r   r   r	   r   r   r   r   r     r   z!ComponentsManager.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ComponentSpecr   c                 O   r   r   r	   r
   r   r   r   r     r   zComponentSpec.__init__c                 O   r   r   r	   r   r   r   r   r     r   zComponentSpec.from_configc                 O   r   r   r	   r   r   r   r   r     r   zComponentSpec.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ConditionalPipelineBlocksr   c                 O   r   r   r	   r
   r   r   r   r     r   z"ConditionalPipelineBlocks.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%ConditionalPipelineBlocks.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)ConditionalPipelineBlocks.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	
ConfigSpecr   c                 O   r   r   r	   r
   r   r   r   r     r   zConfigSpec.__init__c                 O   r   r   r	   r   r   r   r   r     r   zConfigSpec.from_configc                 O   r   r   r	   r   r   r   r   r     r   zConfigSpec.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	
InputParamr   c                 O   r   r   r	   r
   r   r   r   r     r   zInputParam.__init__c                 O   r   r   r	   r   r   r   r   r     r   zInputParam.from_configc                 O   r   r   r	   r   r   r   r   r     r   zInputParam.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	LoopSequentialPipelineBlocksr   c                 O   r   r   r	   r
   r   r   r   r     r   z%LoopSequentialPipelineBlocks.__init__c                 O   r   r   r	   r   r   r   r   r     r   z(LoopSequentialPipelineBlocks.from_configc                 O   r   r   r	   r   r   r   r   r     r   z,LoopSequentialPipelineBlocks.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ModularPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zModularPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zModularPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   zModularPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ModularPipelineBlocksr   c                 O   r   r   r	   r
   r   r   r   r     r   zModularPipelineBlocks.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!ModularPipelineBlocks.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%ModularPipelineBlocks.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	OutputParamr   c                 O   r   r   r	   r
   r   r   r   r     r   zOutputParam.__init__c                 O   r   r   r	   r   r   r   r   r     r   zOutputParam.from_configc                 O   r   r   r	   r   r   r   r   r   #  r   zOutputParam.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SequentialPipelineBlocksr   c                 O   r   r   r	   r
   r   r   r   r   +  r   z!SequentialPipelineBlocks.__init__c                 O   r   r   r	   r   r   r   r   r   .  r   z$SequentialPipelineBlocks.from_configc                 O   r   r   r	   r   r   r   r   r   2  r   z(SequentialPipelineBlocks.from_pretrainedNr   r   r   r   r   r   (  r   r   c                  O   r1   r   )r   get_constant_scheduler3   r   r   r   r   7  r   r   c                  O   r1   r   )r   !get_constant_schedule_with_warmupr3   r   r   r   r   ;  r   r   c                  O   r1   r   )r   get_cosine_schedule_with_warmupr3   r   r   r   r   ?  r   r   c                  O   r1   r   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr3   r   r   r   r   C  r   r   c                  O   r1   r   )r   get_linear_schedule_with_warmupr3   r   r   r   r   G  r   r   c                  O   r1   r   )r   )get_polynomial_decay_schedule_with_warmupr3   r   r   r   r   K  r   r   c                  O   r1   r   )r   get_schedulerr3   r   r   r   r   O  r   r   c                   @   r   )	AudioPipelineOutputr   c                 O   r   r   r	   r
   r   r   r   r   V  r   zAudioPipelineOutput.__init__c                 O   r   r   r	   r   r   r   r   r   Y  r   zAudioPipelineOutput.from_configc                 O   r   r   r	   r   r   r   r   r   ]  r   z#AudioPipelineOutput.from_pretrainedNr   r   r   r   r   r   S  r   r   c                   @   r   )	AutoPipelineForImage2Imager   c                 O   r   r   r	   r
   r   r   r   r   e  r   z#AutoPipelineForImage2Image.__init__c                 O   r   r   r	   r   r   r   r   r   h  r   z&AutoPipelineForImage2Image.from_configc                 O   r   r   r	   r   r   r   r   r   l  r   z*AutoPipelineForImage2Image.from_pretrainedNr   r   r   r   r   r   b  r   r   c                   @   r   )	AutoPipelineForInpaintingr   c                 O   r   r   r	   r
   r   r   r   r   t  r   z"AutoPipelineForInpainting.__init__c                 O   r   r   r	   r   r   r   r   r   w  r   z%AutoPipelineForInpainting.from_configc                 O   r   r   r	   r   r   r   r   r   {  r   z)AutoPipelineForInpainting.from_pretrainedNr   r   r   r   r   r   q  r   r   c                   @   r   )	AutoPipelineForText2Imager   c                 O   r   r   r	   r
   r   r   r   r     r   z"AutoPipelineForText2Image.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%AutoPipelineForText2Image.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)AutoPipelineForText2Image.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	BlipDiffusionControlNetPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   z(BlipDiffusionControlNetPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   z+BlipDiffusionControlNetPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z/BlipDiffusionControlNetPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	BlipDiffusionPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zBlipDiffusionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!BlipDiffusionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%BlipDiffusionPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	CLIPImageProjectionr   c                 O   r   r   r	   r
   r   r   r   r     r   zCLIPImageProjection.__init__c                 O   r   r   r	   r   r   r   r   r     r   zCLIPImageProjection.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#CLIPImageProjection.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ConsistencyModelPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   z!ConsistencyModelPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   z$ConsistencyModelPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z(ConsistencyModelPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	DanceDiffusionPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zDanceDiffusionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   z"DanceDiffusionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z&DanceDiffusionPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	DDIMPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zDDIMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zDDIMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   zDDIMPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	DDPMPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zDDPMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zDDPMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   zDDPMPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	DiffusionPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zDiffusionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zDiffusionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z!DiffusionPipeline.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	DiTPipeliner   c                 O   r   r   r	   r
   r   r   r   r   
	  r   zDiTPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   zDiTPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   zDiTPipeline.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	ImagePipelineOutputr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zImagePipelineOutput.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   zImagePipelineOutput.from_configc                 O   r   r   r	   r   r   r   r   r    	  r   z#ImagePipelineOutput.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	KarrasVePipeliner   c                 O   r   r   r	   r
   r   r   r   r   (	  r   zKarrasVePipeline.__init__c                 O   r   r   r	   r   r   r   r   r   +	  r   zKarrasVePipeline.from_configc                 O   r   r   r	   r   r   r   r   r   /	  r   z KarrasVePipeline.from_pretrainedNr   r   r   r   r   r   %	  r   r   c                   @   r   )	LDMPipeliner   c                 O   r   r   r	   r
   r   r   r   r   7	  r   zLDMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   :	  r   zLDMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   >	  r   zLDMPipeline.from_pretrainedNr   r   r   r   r   r   4	  r   r   c                   @   r   )	LDMSuperResolutionPipeliner   c                 O   r   r   r	   r
   r   r   r   r   F	  r   z#LDMSuperResolutionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   I	  r   z&LDMSuperResolutionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   M	  r   z*LDMSuperResolutionPipeline.from_pretrainedNr   r   r   r   r   r   C	  r   r   c                   @   r   )	PNDMPipeliner   c                 O   r   r   r	   r
   r   r   r   r   U	  r   zPNDMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   X	  r   zPNDMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   \	  r   zPNDMPipeline.from_pretrainedNr   r   r   r   r   r   R	  r   r   c                   @   r   )	RePaintPipeliner   c                 O   r   r   r	   r
   r   r   r   r   d	  r   zRePaintPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   g	  r   zRePaintPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   k	  r   zRePaintPipeline.from_pretrainedNr   r   r   r   r   r   a	  r   r   c                   @   r   )	ScoreSdeVePipeliner   c                 O   r   r   r	   r
   r   r   r   r   s	  r   zScoreSdeVePipeline.__init__c                 O   r   r   r	   r   r   r   r   r   v	  r   zScoreSdeVePipeline.from_configc                 O   r   r   r	   r   r   r   r   r   z	  r   z"ScoreSdeVePipeline.from_pretrainedNr   r   r   r   r   r   p	  r   r   c                   @   r   )	StableDiffusionMixinr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zStableDiffusionMixin.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z StableDiffusionMixin.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z$StableDiffusionMixin.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	DiffusersQuantizerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zDiffusersQuantizer.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   zDiffusersQuantizer.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z"DiffusersQuantizer.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	AmusedSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zAmusedScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   zAmusedScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   zAmusedScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	CMStochasticIterativeSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   z'CMStochasticIterativeScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z*CMStochasticIterativeScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z.CMStochasticIterativeScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	CogVideoXDDIMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zCogVideoXDDIMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z"CogVideoXDDIMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z&CogVideoXDDIMScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	CogVideoXDPMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zCogVideoXDPMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z!CogVideoXDPMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z%CogVideoXDPMScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	DDIMInverseSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zDDIMInverseScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z DDIMInverseScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z$DDIMInverseScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	DDIMParallelSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zDDIMParallelScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   z!DDIMParallelScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   	  r   z%DDIMParallelScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	DDIMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	  r   zDDIMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   	  r   zDDIMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   zDDIMScheduler.from_pretrainedNr   r   r   r   r   r   	  r   r   c                   @   r   )	DDPMParallelSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   	
  r   zDDPMParallelScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z!DDPMParallelScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z%DDPMParallelScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	DDPMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zDDPMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   zDDPMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   zDDPMScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	DDPMWuerstchenSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   '
  r   z DDPMWuerstchenScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   *
  r   z#DDPMWuerstchenScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   .
  r   z'DDPMWuerstchenScheduler.from_pretrainedNr   r   r   r   r   r   $
  r   r   c                   @   r   )	DEISMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   6
  r   zDEISMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   9
  r   z"DEISMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   =
  r   z&DEISMultistepScheduler.from_pretrainedNr   r   r   r   r   r   3
  r   r   c                   @   r   )	"DPMSolverMultistepInverseSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   E
  r   z+DPMSolverMultistepInverseScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   H
  r   z.DPMSolverMultistepInverseScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   L
  r   z2DPMSolverMultistepInverseScheduler.from_pretrainedNr   r   r   r   r   r   B
  r   r   c                   @   r   )	DPMSolverMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   T
  r   z$DPMSolverMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   W
  r   z'DPMSolverMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   [
  r   z+DPMSolverMultistepScheduler.from_pretrainedNr   r   r   r   r   r   Q
  r   r   c                   @   r   )	DPMSolverSinglestepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   c
  r   z%DPMSolverSinglestepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   f
  r   z(DPMSolverSinglestepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   j
  r   z,DPMSolverSinglestepScheduler.from_pretrainedNr   r   r   r   r   r   `
  r   r   c                   @   r   )	EDMDPMSolverMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   r
  r   z'EDMDPMSolverMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   u
  r   z*EDMDPMSolverMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   y
  r   z.EDMDPMSolverMultistepScheduler.from_pretrainedNr   r   r   r   r   r   o
  r   r   c                   @   r   )	EDMEulerSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zEDMEulerScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   zEDMEulerScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z!EDMEulerScheduler.from_pretrainedNr   r   r   r   r   r   ~
  r   r   c                   @   r   )	EulerAncestralDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   z(EulerAncestralDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z+EulerAncestralDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z/EulerAncestralDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	EulerDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zEulerDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z"EulerDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z&EulerDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	FlowMatchEulerDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   z(FlowMatchEulerDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z+FlowMatchEulerDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z/FlowMatchEulerDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	FlowMatchHeunDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   z'FlowMatchHeunDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z*FlowMatchHeunDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z.FlowMatchHeunDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	FlowMatchLCMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zFlowMatchLCMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z!FlowMatchLCMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z%FlowMatchLCMScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	HeliosDMDSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zHeliosDMDScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   zHeliosDMDScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   z"HeliosDMDScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	HeliosSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zHeliosScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   zHeliosScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   
  r   zHeliosScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	HeunDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   
  r   zHeunDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   
  r   z!HeunDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r      r   z%HeunDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   
  r   r   c                   @   r   )	IPNDMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zIPNDMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zIPNDMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   zIPNDMScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	KarrasVeSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zKarrasVeScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zKarrasVeScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z!KarrasVeScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	KDPM2AncestralDiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   &  r   z(KDPM2AncestralDiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   )  r   z+KDPM2AncestralDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   -  r   z/KDPM2AncestralDiscreteScheduler.from_pretrainedNr   r   r   r   r   r   #  r   r   c                   @   r   )	KDPM2DiscreteSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   5  r   zKDPM2DiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   8  r   z"KDPM2DiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   <  r   z&KDPM2DiscreteScheduler.from_pretrainedNr   r   r   r   r   r   2  r   r   c                   @   r   )	LCMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   D  r   zLCMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   G  r   zLCMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   K  r   zLCMScheduler.from_pretrainedNr   r   r   r   r   r   A  r   r   c                   @   r   )	LTXEulerAncestralRFSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   S  r   z%LTXEulerAncestralRFScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   V  r   z(LTXEulerAncestralRFScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   Z  r   z,LTXEulerAncestralRFScheduler.from_pretrainedNr   r   r   r   r   r   P  r   r   c                   @   r   )	PNDMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   b  r   zPNDMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   e  r   zPNDMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   i  r   zPNDMScheduler.from_pretrainedNr   r   r   r   r   r   _  r   r   c                   @   r   )	RePaintSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   q  r   zRePaintScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   t  r   zRePaintScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   x  r   z RePaintScheduler.from_pretrainedNr   r   r   r   r   r   n  r   r   c                   @   r   )	SASolverSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zSASolverScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zSASolverScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z!SASolverScheduler.from_pretrainedNr   r   r   r   r   r   }  r   r   c                   @   r   )	SchedulerMixinr   c                 O   r   r   r	   r
   r   r   r   r     r   zSchedulerMixin.__init__c                 O   r   r   r	   r   r   r   r   r     r   zSchedulerMixin.from_configc                 O   r   r   r	   r   r   r   r   r     r   zSchedulerMixin.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	SCMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zSCMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zSCMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   zSCMScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	ScoreSdeVeSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zScoreSdeVeScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zScoreSdeVeScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#ScoreSdeVeScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	TCDSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zTCDScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zTCDScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   zTCDScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UnCLIPSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zUnCLIPScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   zUnCLIPScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   zUnCLIPScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	UniPCMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   z UniPCMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   z#UniPCMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z'UniPCMultistepScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	VQDiffusionSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   zVQDiffusionScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   z VQDiffusionScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z$VQDiffusionScheduler.from_pretrainedNr   r   r   r   r   r     r   r   c                   @   r   )	EMAModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zEMAModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zEMAModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   zEMAModel.from_pretrainedNr   r   r   r   r   r     r   r   N)utilsr   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r2   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   ri   rj   rk   rl   rm   rn   ro   rp   rq   rr   rs   rt   ru   rv   rw   rx   ry   rz   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   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>   s  