
    wi|                     	   d dl mZmZ  G d de          Z G d de          Z G d de          Z G d	 d
e          Z G d de          Z G d de          Z G d de          Z	 G d de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d d e          Z G d! d"e          Z G d# d$e          Z G d% d&e          Z G d' d(e          Z G d) d*e          Z G d+ d,e          Z G d- d.e          Z G d/ d0e          Z G d1 d2e          Z G d3 d4e          Z G d5 d6e          Z G d7 d8e          Z G d9 d:e          Z G d; d<e          Zd= Z d> Z!d? Z"d@ Z#dA Z$dB Z%dC Z& G dD dEe          Z' G dF dGe          Z( G dH dIe          Z) G dJ dKe          Z* G dL dMe          Z+ G dN dOe          Z, G dP dQe          Z- G dR dSe          Z. G dT dUe          Z/ G dV dWe          Z0 G dX dYe          Z1 G dZ d[e          Z2 G d\ d]e          Z3 G d^ d_e          Z4 G d` dae          Z5 G db dce          Z6 G dd dee          Z7 G df dge          Z8 G dh die          Z9 G dj dke          Z: G dl dme          Z; G dn doe          Z< G dp dqe          Z= G dr dse          Z> G dt due          Z? G dv dwe          Z@ G dx dye          ZA G dz d{e          ZB G d| d}e          ZC G d~ de          ZD G d de          ZE G d de          ZF G d de          ZG G d de          ZH G d de          ZI G d de          ZJ G d de          ZK G d de          ZL G d de          ZM G d de          ZN G d de          ZO G d de          ZP G d de          ZQ G d de          ZR G d de          ZS G d de          ZT G d de          ZU G d de          ZV G d de          ZW G d de          ZX G d de          ZY G d de          ZZ G d de          Z[ G d de          Z\dS )   )DummyObjectrequires_backendsc                   F    e Zd ZdgZd Zed             Zed             ZdS )AsymmetricAutoencoderKLtorchc                 (    t          | dg           d S Nr   r   selfargskwargss      p/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/utils/dummy_pt_objects.py__init__z AsymmetricAutoencoderKL.__init__       $	*****    c                 (    t          | dg           d S r	   r
   clsr   r   s      r   from_configz#AsymmetricAutoencoderKL.from_config       #y)))))r   c                 (    t          | dg           d S r	   r
   r   s      r   from_pretrainedz'AsymmetricAutoencoderKL.from_pretrained   r   r   N__name__
__module____qualname__	_backendsr   classmethodr   r    r   r   r   r      `        	I+ + + * * [* * * [* * *r   r   )	metaclassc                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoencoderKLr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderKL.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderKL.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderKL.from_pretrained   r   r   Nr   r    r   r   r$   r$      r!   r   r$   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoencoderKLTemporalDecoderr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%AutoencoderKLTemporalDecoder.__init__&   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(AutoencoderKLTemporalDecoder.from_config)   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z,AutoencoderKLTemporalDecoder.from_pretrained-   r   r   Nr   r    r   r   r)   r)   #   r!   r   r)   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoencoderTinyr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderTiny.__init__5   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderTiny.from_config8   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAutoencoderTiny.from_pretrained<   r   r   Nr   r    r   r   r.   r.   2   r!   r   r.   c                   F    e Zd ZdgZd Zed             Zed             ZdS )ConsistencyDecoderVAEr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zConsistencyDecoderVAE.__init__D   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!ConsistencyDecoderVAE.from_configG   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%ConsistencyDecoderVAE.from_pretrainedK   r   r   Nr   r    r   r   r3   r3   A   r!   r   r3   c                   F    e Zd ZdgZd Zed             Zed             ZdS )ControlNetModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zControlNetModel.__init__S   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zControlNetModel.from_configV   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zControlNetModel.from_pretrainedZ   r   r   Nr   r    r   r   r8   r8   P   r!   r   r8   c                   F    e Zd ZdgZd Zed             Zed             ZdS )ControlNetXSAdapterr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zControlNetXSAdapter.__init__b   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zControlNetXSAdapter.from_confige   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#ControlNetXSAdapter.from_pretrainedi   r   r   Nr   r    r   r   r=   r=   _   r!   r   r=   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DiTTransformer2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiTTransformer2DModel.__init__q   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!DiTTransformer2DModel.from_configt   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%DiTTransformer2DModel.from_pretrainedx   r   r   Nr   r    r   r   rB   rB   n   r!   r   rB   c                   F    e Zd ZdgZd Zed             Zed             ZdS )HunyuanDiT2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zHunyuanDiT2DModel.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zHunyuanDiT2DModel.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!HunyuanDiT2DModel.from_pretrained   r   r   Nr   r    r   r   rG   rG   }   r!   r   rG   c                   F    e Zd ZdgZd Zed             Zed             ZdS )I2VGenXLUNetr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zI2VGenXLUNet.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zI2VGenXLUNet.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zI2VGenXLUNet.from_pretrained   r   r   Nr   r    r   r   rL   rL      r!   r   rL   c                   F    e Zd ZdgZd Zed             Zed             ZdS )Kandinsky3UNetr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKandinsky3UNet.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKandinsky3UNet.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKandinsky3UNet.from_pretrained   r   r   Nr   r    r   r   rQ   rQ      r!   r   rQ   c                   F    e Zd ZdgZd Zed             Zed             ZdS )
ModelMixinr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zModelMixin.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zModelMixin.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zModelMixin.from_pretrained   r   r   Nr   r    r   r   rV   rV      r!   r   rV   c                   F    e Zd ZdgZd Zed             Zed             ZdS )MotionAdapterr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMotionAdapter.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMotionAdapter.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMotionAdapter.from_pretrained   r   r   Nr   r    r   r   r[   r[      r!   r   r[   c                   F    e Zd ZdgZd Zed             Zed             ZdS )MultiAdapterr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMultiAdapter.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMultiAdapter.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zMultiAdapter.from_pretrained   r   r   Nr   r    r   r   r`   r`      r!   r   r`   c                   F    e Zd ZdgZd Zed             Zed             ZdS )PixArtTransformer2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!PixArtTransformer2DModel.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$PixArtTransformer2DModel.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(PixArtTransformer2DModel.from_pretrained   r   r   Nr   r    r   r   re   re      r!   r   re   c                   F    e Zd ZdgZd Zed             Zed             ZdS )PriorTransformerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPriorTransformer.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPriorTransformer.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z PriorTransformer.from_pretrained   r   r   Nr   r    r   r   rj   rj      r!   r   rj   c                   F    e Zd ZdgZd Zed             Zed             ZdS )SD3Transformer2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSD3Transformer2DModel.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!SD3Transformer2DModel.from_config   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%SD3Transformer2DModel.from_pretrained   r   r   Nr   r    r   r   ro   ro      r!   r   ro   c                   F    e Zd ZdgZd Zed             Zed             ZdS )
T2IAdapterr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT2IAdapter.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT2IAdapter.from_config
  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT2IAdapter.from_pretrained  r   r   Nr   r    r   r   rt   rt     r!   r   rt   c                   F    e Zd ZdgZd Zed             Zed             ZdS )T5FilmDecoderr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT5FilmDecoder.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT5FilmDecoder.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zT5FilmDecoder.from_pretrained  r   r   Nr   r    r   r   ry   ry     r!   r   ry   c                   F    e Zd ZdgZd Zed             Zed             ZdS )Transformer2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zTransformer2DModel.__init__%  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zTransformer2DModel.from_config(  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"Transformer2DModel.from_pretrained,  r   r   Nr   r    r   r   r~   r~   "  r!   r   r~   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNet1DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet1DModel.__init__4  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet1DModel.from_config7  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet1DModel.from_pretrained;  r   r   Nr   r    r   r   r   r   1  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNet2DConditionModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet2DConditionModel.__init__C  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z UNet2DConditionModel.from_configF  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$UNet2DConditionModel.from_pretrainedJ  r   r   Nr   r    r   r   r   r   @  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNet2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet2DModel.__init__R  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet2DModel.from_configU  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet2DModel.from_pretrainedY  r   r   Nr   r    r   r   r   r   O  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNet3DConditionModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNet3DConditionModel.__init__a  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z UNet3DConditionModel.from_configd  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$UNet3DConditionModel.from_pretrainedh  r   r   Nr   r    r   r   r   r   ^  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNetControlNetXSModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNetControlNetXSModel.__init__p  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!UNetControlNetXSModel.from_configs  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%UNetControlNetXSModel.from_pretrainedw  r   r   Nr   r    r   r   r   r   m  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UNetMotionModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNetMotionModel.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNetMotionModel.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUNetMotionModel.from_pretrained  r   r   Nr   r    r   r   r   r   |  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS ) UNetSpatioTemporalConditionModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z)UNetSpatioTemporalConditionModel.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z,UNetSpatioTemporalConditionModel.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z0UNetSpatioTemporalConditionModel.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )UVit2DModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUVit2DModel.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUVit2DModel.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUVit2DModel.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )VQModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zVQModel.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zVQModel.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zVQModel.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                  2    t          t          dg           d S r	   )r   get_constant_scheduler   r   s     r   r   r     s    +gY77777r   c                  2    t          t          dg           d S r	   )r   !get_constant_schedule_with_warmupr   s     r   r   r     s    7'CCCCCr   c                  2    t          t          dg           d S r	   )r   get_cosine_schedule_with_warmupr   s     r   r   r         5yAAAAAr   c                  2    t          t          dg           d S r	   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr   s     r   r   r     s    H7)TTTTTr   c                  2    t          t          dg           d S r	   )r   get_linear_schedule_with_warmupr   s     r   r   r     r   r   c                  2    t          t          dg           d S r	   )r   )get_polynomial_decay_schedule_with_warmupr   s     r   r   r     s    ?'KKKKKr   c                  2    t          t          dg           d S r	   )r   get_schedulerr   s     r   r   r     s    mgY/////r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AudioPipelineOutputr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAudioPipelineOutput.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAudioPipelineOutput.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#AudioPipelineOutput.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoPipelineForImage2Imager   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#AutoPipelineForImage2Image.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&AutoPipelineForImage2Image.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z*AutoPipelineForImage2Image.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoPipelineForInpaintingr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"AutoPipelineForInpainting.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%AutoPipelineForInpainting.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z)AutoPipelineForInpainting.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )AutoPipelineForText2Imager   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"AutoPipelineForText2Image.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%AutoPipelineForText2Image.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z)AutoPipelineForText2Image.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )BlipDiffusionControlNetPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(BlipDiffusionControlNetPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+BlipDiffusionControlNetPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z/BlipDiffusionControlNetPipeline.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )BlipDiffusionPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zBlipDiffusionPipeline.__init__"  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!BlipDiffusionPipeline.from_config%  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%BlipDiffusionPipeline.from_pretrained)  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )CLIPImageProjectionr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zCLIPImageProjection.__init__1  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zCLIPImageProjection.from_config4  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#CLIPImageProjection.from_pretrained8  r   r   Nr   r    r   r   r   r   .  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )ConsistencyModelPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!ConsistencyModelPipeline.__init__@  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$ConsistencyModelPipeline.from_configC  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(ConsistencyModelPipeline.from_pretrainedG  r   r   Nr   r    r   r   r   r   =  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DanceDiffusionPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDanceDiffusionPipeline.__init__O  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"DanceDiffusionPipeline.from_configR  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&DanceDiffusionPipeline.from_pretrainedV  r   r   Nr   r    r   r   r   r   L  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDIMPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMPipeline.__init__^  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMPipeline.from_configa  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMPipeline.from_pretrainede  r   r   Nr   r    r   r   r   r   [  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDPMPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMPipeline.__init__m  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMPipeline.from_configp  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMPipeline.from_pretrainedt  r   r   Nr   r    r   r   r   r   j  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DiffusionPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiffusionPipeline.__init__|  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiffusionPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!DiffusionPipeline.from_pretrained  r   r   Nr   r    r   r   r   r   y  r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )DiTPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiTPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiTPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDiTPipeline.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   F    e Zd ZdgZd Zed             Zed             ZdS )ImagePipelineOutputr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zImagePipelineOutput.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zImagePipelineOutput.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#ImagePipelineOutput.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )KarrasVePipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKarrasVePipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKarrasVePipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z KarrasVePipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )LDMPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLDMPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLDMPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLDMPipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )LDMSuperResolutionPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#LDMSuperResolutionPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&LDMSuperResolutionPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z*LDMSuperResolutionPipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )PNDMPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMPipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )RePaintPipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zRePaintPipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zRePaintPipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zRePaintPipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )ScoreSdeVePipeliner   c                 (    t          | dg           d S r	   r
   r   s      r   r   zScoreSdeVePipeline.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zScoreSdeVePipeline.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"ScoreSdeVePipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )StableDiffusionMixinr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zStableDiffusionMixin.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z StableDiffusionMixin.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$StableDiffusionMixin.from_pretrained
  r   r   Nr   r    r   r   r$  r$     r!   r   r$  c                   F    e Zd ZdgZd Zed             Zed             ZdS )AmusedSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAmusedScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAmusedScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zAmusedScheduler.from_pretrained  r   r   Nr   r    r   r   r)  r)    r!   r   r)  c                   F    e Zd ZdgZd Zed             Zed             ZdS )CMStochasticIterativeSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'CMStochasticIterativeScheduler.__init__!  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z*CMStochasticIterativeScheduler.from_config$  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z.CMStochasticIterativeScheduler.from_pretrained(  r   r   Nr   r    r   r   r.  r.    r!   r   r.  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDIMInverseSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMInverseScheduler.__init__0  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z DDIMInverseScheduler.from_config3  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$DDIMInverseScheduler.from_pretrained7  r   r   Nr   r    r   r   r3  r3  -  r!   r   r3  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDIMParallelSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMParallelScheduler.__init__?  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!DDIMParallelScheduler.from_configB  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%DDIMParallelScheduler.from_pretrainedF  r   r   Nr   r    r   r   r8  r8  <  r!   r   r8  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDIMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMScheduler.__init__N  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMScheduler.from_configQ  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDIMScheduler.from_pretrainedU  r   r   Nr   r    r   r   r=  r=  K  r!   r   r=  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDPMParallelSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMParallelScheduler.__init__]  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!DDPMParallelScheduler.from_config`  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%DDPMParallelScheduler.from_pretrainedd  r   r   Nr   r    r   r   rB  rB  Z  r!   r   rB  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDPMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMScheduler.__init__l  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMScheduler.from_configo  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDDPMScheduler.from_pretraineds  r   r   Nr   r    r   r   rG  rG  i  r!   r   rG  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DDPMWuerstchenSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z DDPMWuerstchenScheduler.__init__{  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#DDPMWuerstchenScheduler.from_config~  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'DDPMWuerstchenScheduler.from_pretrained  r   r   Nr   r    r   r   rL  rL  x  r!   r   rL  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DEISMultistepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zDEISMultistepScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"DEISMultistepScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&DEISMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   rQ  rQ    r!   r   rQ  c                   F    e Zd ZdgZd Zed             Zed             ZdS )"DPMSolverMultistepInverseSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+DPMSolverMultistepInverseScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z.DPMSolverMultistepInverseScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z2DPMSolverMultistepInverseScheduler.from_pretrained  r   r   Nr   r    r   r   rV  rV    r!   r   rV  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DPMSolverMultistepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$DPMSolverMultistepScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'DPMSolverMultistepScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+DPMSolverMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   r[  r[    r!   r   r[  c                   F    e Zd ZdgZd Zed             Zed             ZdS )DPMSolverSinglestepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%DPMSolverSinglestepScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(DPMSolverSinglestepScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z,DPMSolverSinglestepScheduler.from_pretrained  r   r   Nr   r    r   r   r`  r`    r!   r   r`  c                   F    e Zd ZdgZd Zed             Zed             ZdS )EDMDPMSolverMultistepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'EDMDPMSolverMultistepScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z*EDMDPMSolverMultistepScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z.EDMDPMSolverMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   re  re    r!   r   re  c                   F    e Zd ZdgZd Zed             Zed             ZdS )EDMEulerSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEDMEulerScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEDMEulerScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!EDMEulerScheduler.from_pretrained  r   r   Nr   r    r   r   rj  rj    r!   r   rj  c                   F    e Zd ZdgZd Zed             Zed             ZdS )EulerAncestralDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(EulerAncestralDiscreteScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+EulerAncestralDiscreteScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z/EulerAncestralDiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   ro  ro    r!   r   ro  c                   F    e Zd ZdgZd Zed             Zed             ZdS )EulerDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEulerDiscreteScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"EulerDiscreteScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&EulerDiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   rt  rt    r!   r   rt  c                   F    e Zd ZdgZd Zed             Zed             ZdS )FlowMatchEulerDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(FlowMatchEulerDiscreteScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+FlowMatchEulerDiscreteScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z/FlowMatchEulerDiscreteScheduler.from_pretrained	  r   r   Nr   r    r   r   ry  ry    r!   r   ry  c                   F    e Zd ZdgZd Zed             Zed             ZdS )HeunDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zHeunDiscreteScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!HeunDiscreteScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z%HeunDiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   r~  r~    r!   r   r~  c                   F    e Zd ZdgZd Zed             Zed             ZdS )IPNDMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zIPNDMScheduler.__init__   r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zIPNDMScheduler.from_config#  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zIPNDMScheduler.from_pretrained'  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )KarrasVeSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKarrasVeScheduler.__init__/  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKarrasVeScheduler.from_config2  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!KarrasVeScheduler.from_pretrained6  r   r   Nr   r    r   r   r  r  ,  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )KDPM2AncestralDiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z(KDPM2AncestralDiscreteScheduler.__init__>  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z+KDPM2AncestralDiscreteScheduler.from_configA  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z/KDPM2AncestralDiscreteScheduler.from_pretrainedE  r   r   Nr   r    r   r   r  r  ;  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )KDPM2DiscreteSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zKDPM2DiscreteScheduler.__init__M  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z"KDPM2DiscreteScheduler.from_configP  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z&KDPM2DiscreteScheduler.from_pretrainedT  r   r   Nr   r    r   r   r  r  J  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )LCMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLCMScheduler.__init__\  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLCMScheduler.from_config_  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zLCMScheduler.from_pretrainedc  r   r   Nr   r    r   r   r  r  Y  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )PNDMSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMScheduler.__init__k  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMScheduler.from_confign  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zPNDMScheduler.from_pretrainedr  r   r   Nr   r    r   r   r  r  h  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )RePaintSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zRePaintScheduler.__init__z  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zRePaintScheduler.from_config}  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z RePaintScheduler.from_pretrained  r   r   Nr   r    r   r   r  r  w  r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )SASolverSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSASolverScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSASolverScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z!SASolverScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )SchedulerMixinr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSchedulerMixin.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSchedulerMixin.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zSchedulerMixin.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )ScoreSdeVeSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zScoreSdeVeScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zScoreSdeVeScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#ScoreSdeVeScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )TCDSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zTCDScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zTCDScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zTCDScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )UnCLIPSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUnCLIPScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUnCLIPScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zUnCLIPScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )UniPCMultistepSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   z UniPCMultistepScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z#UniPCMultistepScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z'UniPCMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )VQDiffusionSchedulerr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zVQDiffusionScheduler.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z VQDiffusionScheduler.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   z$VQDiffusionScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   F    e Zd ZdgZd Zed             Zed             ZdS )EMAModelr   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEMAModel.__init__  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEMAModel.from_config  r   r   c                 (    t          | dg           d S r	   r
   r   s      r   r   zEMAModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  N)]utilsr   r   r   r$   r)   r.   r3   r8   r=   rB   rG   rL   rQ   rV   r[   r`   re   rj   ro   rt   ry   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  r  r$  r)  r.  r3  r8  r=  rB  rG  rL  rQ  rV  r[  r`  re  rj  ro  rt  ry  r~  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r    r   r   <module>r     s   2 2 2 2 2 2 2 2* * * * * * * * ** * * * *k * * * ** * * * *[ * * * ** * * * * * * * ** * * * *k * * * ** * * * * * * * ** * * * *K * * * ** * * * *k * * * ** * * * *+ * * * ** * * * *[ * * * ** * * * *{ * * * ** * * * *; * * * ** * * * *k * * * ** * * * *[ * * * ** * * * * * * * ** * * * * * * * ** * * * *k * * * ** * * * *; * * * ** * * * *k * * * ** * * * *; * * * ** * * * *K * * * ** * * * *[ * * * ** * * * *K * * * ** * * * *[ * * * ** * * * *k * * * ** * * * * * * * ** * * * * * * * ** * * * *K * * * ** * * * * * * * *8 8 8D D DB B BU U UB B BL L L0 0 0* * * * *K * * * ** * * * *; * * * ** * * * *+ * * * ** * * * *+ * * * ** * * * * * * * ** * * * *k * * * ** * * * *K * * * ** * * * * * * * ** * * * *{ * * * ** * * * *[ * * * ** * * * *[ * * * ** * * * *+ * * * ** * * * *K * * * ** * * * *K * * * ** * * * * * * * ** * * * *K * * * ** * * * *; * * * ** * * * *[ * * * ** * * * * * * * ** * * * *; * * * ** * * * *[ * * * ** * * * * * * * ** * * * *{ * * * ** * * * *[ * * * ** * * * *k * * * ** * * * *k * * * ** * * * *k * * * ** * * * *k * * * ** * * * * * * * ** * * * *{ * * * ** * * * *; * * * ** * * * *K * * * ** * * * *[ * * * ** * * * *{ * * * ** * * * *+ * * * ** * * * * * * * ** * * * *{ * * * ** * * * * * * * ** * * * *k * * * ** * * * *{ * * * ** * * * *+ * * * ** * * * * * * * ** * * * *{ * * * ** * * * *[ * * * ** * * * *k * * * ** * * * * * * * ** * * * *+ * * * ** * * * *{ * * * ** * * * *K * * * ** * * * *[ * * * ** * * * * * * * ** * * * * * * * ** * * * *[ * * * ** * * * * * * * * * *r   