o
    piH                     @   sB  d dl mZmZ G dd dedZG dd dedZG dd dedZG d	d
 d
edZG dd dedZG dd dedZG dd dedZ	G dd dedZ
G dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd dedZG dd  d edZG d!d" d"edZG d#d$ d$edZG d%d& d&edZG d'd( d(edZG d)d* d*edZG d+d, d,edZG d-d. d.edZG d/d0 d0edZG d1d2 d2edZG d3d4 d4edZG d5d6 d6edZG d7d8 d8edZG d9d: d:edZG d;d< d<edZG d=d> d>edZ G d?d@ d@edZ!G dAdB dBedZ"G dCdD dDedZ#G dEdF dFedZ$G dGdH dHedZ%G dIdJ dJedZ&G dKdL dLedZ'G dMdN dNedZ(G dOdP dPedZ)G dQdR dRedZ*G dSdT dTedZ+G dUdV dVedZ,dWdX Z-dYdZ Z.d[d\ Z/d]d^ Z0d_d` Z1dadb Z2dcdd 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ZldS )   )DummyObjectrequires_backendsc                   @   2   e Zd ZdgZdd Zedd Zedd ZdS )	AsymmetricAutoencoderKLtorchc                 O      t | dg d S Nr   r   selfargskwargs r   ^/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/utils/dummy_pt_objects.py__init__      z AsymmetricAutoencoderKL.__init__c                 O   r   r   r	   clsr   r   r   r   r   from_config      z#AsymmetricAutoencoderKL.from_configc                 O   r   r   r	   r   r   r   r   from_pretrained   r   z'AsymmetricAutoencoderKL.from_pretrainedN__name__
__module____qualname__	_backendsr   classmethodr   r   r   r   r   r   r          
r   )	metaclassc                   @   r   )	AuraFlowTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r      r   z#AuraFlowTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r      r   z&AuraFlowTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r      r   z*AuraFlowTransformer2DModel.from_pretrainedNr   r   r   r   r   r      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   )	AutoencoderKLCogVideoXr   c                 O   r   r   r	   r
   r   r   r   r   5   r   zAutoencoderKLCogVideoX.__init__c                 O   r   r   r	   r   r   r   r   r   8   r   z"AutoencoderKLCogVideoX.from_configc                 O   r   r   r	   r   r   r   r   r   <   r   z&AutoencoderKLCogVideoX.from_pretrainedNr   r   r   r   r   r!   2   r   r!   c                   @   r   )	AutoencoderKLTemporalDecoderr   c                 O   r   r   r	   r
   r   r   r   r   D   r   z%AutoencoderKLTemporalDecoder.__init__c                 O   r   r   r	   r   r   r   r   r   G   r   z(AutoencoderKLTemporalDecoder.from_configc                 O   r   r   r	   r   r   r   r   r   K   r   z,AutoencoderKLTemporalDecoder.from_pretrainedNr   r   r   r   r   r"   A   r   r"   c                   @   r   )	AutoencoderOobleckr   c                 O   r   r   r	   r
   r   r   r   r   S   r   zAutoencoderOobleck.__init__c                 O   r   r   r	   r   r   r   r   r   V   r   zAutoencoderOobleck.from_configc                 O   r   r   r	   r   r   r   r   r   Z   r   z"AutoencoderOobleck.from_pretrainedNr   r   r   r   r   r#   P   r   r#   c                   @   r   )	AutoencoderTinyr   c                 O   r   r   r	   r
   r   r   r   r   b   r   zAutoencoderTiny.__init__c                 O   r   r   r	   r   r   r   r   r   e   r   zAutoencoderTiny.from_configc                 O   r   r   r	   r   r   r   r   r   i   r   zAutoencoderTiny.from_pretrainedNr   r   r   r   r   r$   _   r   r$   c                   @   r   )	CogVideoXTransformer3DModelr   c                 O   r   r   r	   r
   r   r   r   r   q   r   z$CogVideoXTransformer3DModel.__init__c                 O   r   r   r	   r   r   r   r   r   t   r   z'CogVideoXTransformer3DModel.from_configc                 O   r   r   r	   r   r   r   r   r   x   r   z+CogVideoXTransformer3DModel.from_pretrainedNr   r   r   r   r   r%   n   r   r%   c                   @   r   )	ConsistencyDecoderVAEr   c                 O   r   r   r	   r
   r   r   r   r      r   zConsistencyDecoderVAE.__init__c                 O   r   r   r	   r   r   r   r   r      r   z!ConsistencyDecoderVAE.from_configc                 O   r   r   r	   r   r   r   r   r      r   z%ConsistencyDecoderVAE.from_pretrainedNr   r   r   r   r   r&   }   r   r&   c                   @   r   )	ControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r      r   zControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r      r   zControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r      r   zControlNetModel.from_pretrainedNr   r   r   r   r   r'      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   )	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   r)      r   r)   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   r*      r   r*   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      r   z+HunyuanDiT2DControlNetModel.from_pretrainedNr   r   r   r   r   r+      r   r+   c                   @   r   )	HunyuanDiT2DModelr   c                 O   r   r   r	   r
   r   r   r   r      r   zHunyuanDiT2DModel.__init__c                 O   r   r   r	   r   r   r   r   r      r   zHunyuanDiT2DModel.from_configc                 O   r   r   r	   r   r   r   r   r      r   z!HunyuanDiT2DModel.from_pretrainedNr   r   r   r   r   r,      r   r,   c                   @   r   )	 HunyuanDiT2DMultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r      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   r-      r   r-   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   r.      r   r.   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   r/     r   r/   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   r0     r   r0   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   r1   "  r   r1   c                   @   r   )	
ModelMixinr   c                 O   r   r   r	   r
   r   r   r   r   4  r   zModelMixin.__init__c                 O   r   r   r	   r   r   r   r   r   7  r   zModelMixin.from_configc                 O   r   r   r	   r   r   r   r   r   ;  r   zModelMixin.from_pretrainedNr   r   r   r   r   r2   1  r   r2   c                   @   r   )	MotionAdapterr   c                 O   r   r   r	   r
   r   r   r   r   C  r   zMotionAdapter.__init__c                 O   r   r   r	   r   r   r   r   r   F  r   zMotionAdapter.from_configc                 O   r   r   r	   r   r   r   r   r   J  r   zMotionAdapter.from_pretrainedNr   r   r   r   r   r3   @  r   r3   c                   @   r   )	MultiAdapterr   c                 O   r   r   r	   r
   r   r   r   r   R  r   zMultiAdapter.__init__c                 O   r   r   r	   r   r   r   r   r   U  r   zMultiAdapter.from_configc                 O   r   r   r	   r   r   r   r   r   Y  r   zMultiAdapter.from_pretrainedNr   r   r   r   r   r4   O  r   r4   c                   @   r   )	PixArtTransformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r   a  r   z!PixArtTransformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r   d  r   z$PixArtTransformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   h  r   z(PixArtTransformer2DModel.from_pretrainedNr   r   r   r   r   r5   ^  r   r5   c                   @   r   )	PriorTransformerr   c                 O   r   r   r	   r
   r   r   r   r   p  r   zPriorTransformer.__init__c                 O   r   r   r	   r   r   r   r   r   s  r   zPriorTransformer.from_configc                 O   r   r   r	   r   r   r   r   r   w  r   z PriorTransformer.from_pretrainedNr   r   r   r   r   r6   m  r   r6   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     r   z"SD3ControlNetModel.from_pretrainedNr   r   r   r   r   r7   |  r   r7   c                   @   r   )	SD3MultiControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     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   r8     r   r8   c                   @   r   )	SD3Transformer2DModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zSD3Transformer2DModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!SD3Transformer2DModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%SD3Transformer2DModel.from_pretrainedNr   r   r   r   r   r9     r   r9   c                   @   r   )	SparseControlNetModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zSparseControlNetModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   z!SparseControlNetModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z%SparseControlNetModel.from_pretrainedNr   r   r   r   r   r:     r   r:   c                   @   r   )	StableAudioDiTModelr   c                 O   r   r   r	   r
   r   r   r   r     r   zStableAudioDiTModel.__init__c                 O   r   r   r	   r   r   r   r   r     r   zStableAudioDiTModel.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#StableAudioDiTModel.from_pretrainedNr   r   r   r   r   r;     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   )	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   rA     r   rA   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   rB   !  r   rB   c                   @   r   )	UNetControlNetXSModelr   c                 O   r   r   r	   r
   r   r   r   r   3  r   zUNetControlNetXSModel.__init__c                 O   r   r   r	   r   r   r   r   r   6  r   z!UNetControlNetXSModel.from_configc                 O   r   r   r	   r   r   r   r   r   :  r   z%UNetControlNetXSModel.from_pretrainedNr   r   r   r   r   rC   0  r   rC   c                   @   r   )	UNetMotionModelr   c                 O   r   r   r	   r
   r   r   r   r   B  r   zUNetMotionModel.__init__c                 O   r   r   r	   r   r   r   r   r   E  r   zUNetMotionModel.from_configc                 O   r   r   r	   r   r   r   r   r   I  r   zUNetMotionModel.from_pretrainedNr   r   r   r   r   rD   ?  r   rD   c                   @   r   )	 UNetSpatioTemporalConditionModelr   c                 O   r   r   r	   r
   r   r   r   r   Q  r   z)UNetSpatioTemporalConditionModel.__init__c                 O   r   r   r	   r   r   r   r   r   T  r   z,UNetSpatioTemporalConditionModel.from_configc                 O   r   r   r	   r   r   r   r   r   X  r   z0UNetSpatioTemporalConditionModel.from_pretrainedNr   r   r   r   r   rE   N  r   rE   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   c  r   zUVit2DModel.from_configc                 O   r   r   r	   r   r   r   r   r   g  r   zUVit2DModel.from_pretrainedNr   r   r   r   r   rF   ]  r   rF   c                   @   r   )	VQModelr   c                 O   r   r   r	   r
   r   r   r   r   o  r   zVQModel.__init__c                 O   r   r   r	   r   r   r   r   r   r  r   zVQModel.from_configc                 O   r   r   r	   r   r   r   r   r   v  r   zVQModel.from_pretrainedNr   r   r   r   r   rG   l  r   rG   c                  O      t tdg d S r   )r   get_constant_scheduler   r   r   r   r   rI   {  r   rI   c                  O   rH   r   )r   !get_constant_schedule_with_warmuprJ   r   r   r   rK     r   rK   c                  O   rH   r   )r   get_cosine_schedule_with_warmuprJ   r   r   r   rL     r   rL   c                  O   rH   r   )r   2get_cosine_with_hard_restarts_schedule_with_warmuprJ   r   r   r   rM     r   rM   c                  O   rH   r   )r   get_linear_schedule_with_warmuprJ   r   r   r   rN     r   rN   c                  O   rH   r   )r   )get_polynomial_decay_schedule_with_warmuprJ   r   r   r   rO     r   rO   c                  O   rH   r   )r   get_schedulerrJ   r   r   r   rP     r   rP   c                   @   r   )	AudioPipelineOutputr   c                 O   r   r   r	   r
   r   r   r   r     r   zAudioPipelineOutput.__init__c                 O   r   r   r	   r   r   r   r   r     r   zAudioPipelineOutput.from_configc                 O   r   r   r	   r   r   r   r   r     r   z#AudioPipelineOutput.from_pretrainedNr   r   r   r   r   rQ     r   rQ   c                   @   r   )	AutoPipelineForImage2Imager   c                 O   r   r   r	   r
   r   r   r   r     r   z#AutoPipelineForImage2Image.__init__c                 O   r   r   r	   r   r   r   r   r     r   z&AutoPipelineForImage2Image.from_configc                 O   r   r   r	   r   r   r   r   r     r   z*AutoPipelineForImage2Image.from_pretrainedNr   r   r   r   r   rR     r   rR   c                   @   r   )	AutoPipelineForInpaintingr   c                 O   r   r   r	   r
   r   r   r   r     r   z"AutoPipelineForInpainting.__init__c                 O   r   r   r	   r   r   r   r   r     r   z%AutoPipelineForInpainting.from_configc                 O   r   r   r	   r   r   r   r   r     r   z)AutoPipelineForInpainting.from_pretrainedNr   r   r   r   r   rS     r   rS   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   rT     r   rT   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   rU     r   rU   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   rV     r   rV   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   rW     r   rW   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   rX      r   rX   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   rY     r   rY   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   rZ     r   rZ   c                   @   r   )	DDPMPipeliner   c                 O   r   r   r	   r
   r   r   r   r   0  r   zDDPMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   3  r   zDDPMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   7  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   B  r   zDiffusionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   F  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   N  r   zDiTPipeline.__init__c                 O   r   r   r	   r   r   r   r   r   Q  r   zDiTPipeline.from_configc                 O   r   r   r	   r   r   r   r   r   U  r   zDiTPipeline.from_pretrainedNr   r   r   r   r   r]   K  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   d  r   z#ImagePipelineOutput.from_pretrainedNr   r   r   r   r   r^   Z  r   r^   c                   @   r   )	KarrasVePipeliner   c                 O   r   r   r	   r
   r   r   r   r   l  r   zKarrasVePipeline.__init__c                 O   r   r   r	   r   r   r   r   r   o  r   zKarrasVePipeline.from_configc                 O   r   r   r	   r   r   r   r   r   s  r   z KarrasVePipeline.from_pretrainedNr   r   r   r   r   r_   i  r   r_   c                   @   r   )	LDMPipeliner   c                 O   r   r   r	   r
   r   r   r   r   {  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`   x  r   r`   c                   @   r   )	LDMSuperResolutionPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   z#LDMSuperResolutionPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   z&LDMSuperResolutionPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z*LDMSuperResolutionPipeline.from_pretrainedNr   r   r   r   r   ra     r   ra   c                   @   r   )	PNDMPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zPNDMPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zPNDMPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   zPNDMPipeline.from_pretrainedNr   r   r   r   r   rb     r   rb   c                   @   r   )	RePaintPipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zRePaintPipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zRePaintPipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   zRePaintPipeline.from_pretrainedNr   r   r   r   r   rc     r   rc   c                   @   r   )	ScoreSdeVePipeliner   c                 O   r   r   r	   r
   r   r   r   r     r   zScoreSdeVePipeline.__init__c                 O   r   r   r	   r   r   r   r   r     r   zScoreSdeVePipeline.from_configc                 O   r   r   r	   r   r   r   r   r     r   z"ScoreSdeVePipeline.from_pretrainedNr   r   r   r   r   rd     r   rd   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   re     r   re   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   rf     r   rf   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   rg     r   rg   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   rh     r   rh   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   ri     r   ri   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   rj     r   rj   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   rk     r   rk   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   2  r   zDDIMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   6  r   zDDIMScheduler.from_pretrainedNr   r   r   r   r   rl   ,  r   rl   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   A  r   z!DDPMParallelScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   E  r   z%DDPMParallelScheduler.from_pretrainedNr   r   r   r   r   rm   ;  r   rm   c                   @   r   )	DDPMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   M  r   zDDPMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   P  r   zDDPMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   T  r   zDDPMScheduler.from_pretrainedNr   r   r   r   r   rn   J  r   rn   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   c  r   z'DDPMWuerstchenScheduler.from_pretrainedNr   r   r   r   r   ro   Y  r   ro   c                   @   r   )	DEISMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   k  r   zDEISMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   n  r   z"DEISMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   r  r   z&DEISMultistepScheduler.from_pretrainedNr   r   r   r   r   rp   h  r   rp   c                   @   r   )	"DPMSolverMultistepInverseSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   z  r   z+DPMSolverMultistepInverseScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   }  r   z.DPMSolverMultistepInverseScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z2DPMSolverMultistepInverseScheduler.from_pretrainedNr   r   r   r   r   rq   w  r   rq   c                   @   r   )	DPMSolverMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   z$DPMSolverMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   z'DPMSolverMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z+DPMSolverMultistepScheduler.from_pretrainedNr   r   r   r   r   rr     r   rr   c                   @   r   )	DPMSolverSinglestepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   z%DPMSolverSinglestepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   z(DPMSolverSinglestepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z,DPMSolverSinglestepScheduler.from_pretrainedNr   r   r   r   r   rs     r   rs   c                   @   r   )	EDMDPMSolverMultistepSchedulerr   c                 O   r   r   r	   r
   r   r   r   r     r   z'EDMDPMSolverMultistepScheduler.__init__c                 O   r   r   r	   r   r   r   r   r     r   z*EDMDPMSolverMultistepScheduler.from_configc                 O   r   r   r	   r   r   r   r   r     r   z.EDMDPMSolverMultistepScheduler.from_pretrainedNr   r   r   r   r   rt     r   rt   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   ru     r   ru   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   rv     r   rv   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   rw     r   rw   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   rx     r   rx   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   ry     r   ry   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   rz     r   rz   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   1  r   z+KDPM2AncestralDiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   5  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   =  r   zKDPM2DiscreteScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   @  r   z"KDPM2DiscreteScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   D  r   z&KDPM2DiscreteScheduler.from_pretrainedNr   r   r   r   r   r~   :  r   r~   c                   @   r   )	LCMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   L  r   zLCMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   O  r   zLCMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   S  r   zLCMScheduler.from_pretrainedNr   r   r   r   r   r   I  r   r   c                   @   r   )	PNDMSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   [  r   zPNDMScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   ^  r   zPNDMScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   b  r   zPNDMScheduler.from_pretrainedNr   r   r   r   r   r   X  r   r   c                   @   r   )	RePaintSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   j  r   zRePaintScheduler.__init__c                 O   r   r   r	   r   r   r   r   r   m  r   zRePaintScheduler.from_configc                 O   r   r   r	   r   r   r   r   r   q  r   z RePaintScheduler.from_pretrainedNr   r   r   r   r   r   g  r   r   c                   @   r   )	SASolverSchedulerr   c                 O   r   r   r	   r
   r   r   r   r   y  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   v  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   )	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)mutilsr   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rI   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   <module>   s   