
    bi                        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d! Zd" Zd# Zd$ Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Z G d1 d2e      Z G d3 d4e      Z G d5 d6e      Z G d7 d8e      Z G d9 d:e      Z  G d; d<e      Z! G d= d>e      Z" G d? d@e      Z# G dA dBe      Z$ G dC dDe      Z% G dE dFe      Z& G dG dHe      Z' G dI dJe      Z( G dK dLe      Z) G dM dNe      Z* G dO dPe      Z+ G dQ dRe      Z, G dS dTe      Z- G dU dVe      Z. G dW dXe      Z/ G dY dZe      Z0 G d[ d\e      Z1 G d] d^e      Z2 G d_ d`e      Z3 G da dbe      Z4 G dc dde      Z5 G de dfe      Z6 G dg dhe      Z7 G di dje      Z8 G dk dle      Z9 G dm dne      Z: G do dpe      Z; G dq dre      Z< G ds dte      Z= G du dve      Z> G dw dxe      Z? G dy dze      Z@ G d{ d|e      ZA G d} d~e      ZB G d de      ZC G d de      ZD G d de      ZE G d de      ZF G d de      ZG G d de      ZH G d de      ZI G d de      ZJ G d de      ZK G d de      ZL G d de      ZM G d de      ZN G d de      ZO G d de      ZP G d de      ZQ G d de      ZR G d de      ZS G d de      ZT G d de      ZU G d de      ZV G d de      ZW G d de      ZX G d de      ZY G d de      ZZ G d de      Z[ G d de      Z\ G d de      Z] G d de      Z^ G d de      Z_ G d de      Z` G d de      Za G d de      Zb G d de      Zc G d de      ZddÄ Ze G dĄ de      Zf G dƄ de      Zg G dȄ de      Zh G dʄ de      Zid̄ Zjd̈́ Zkd΄ Zldτ ZmdЄ Zndф Zod҄ Zp G dӄ de      Zq G dՄ de      Zr G dׄ de      Zs G dل de      Zt G dۄ de      Zu G d݄ de      Zv G d߄ de      Zw G d de      Zx G d de      Zy G d de      Zz G d de      Z{ G d de      Z| G d de      Z} G d de      Z~ G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d	 d
e      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d e      Z G d! d"e      Z G d# d$e      Z G d% d&e      Z G d' d(e      Z G d) d*e      Z G d+ d,e      Z G d- d.e      Z G d/ d0e      Z G d1 d2e      Z G d3 d4e      Z G d5 d6e      Z G d7 d8e      Z G d9 d:e      Z G d; d<e      Z G d= d>e      Z G d? d@e      Z G dA dBe      Z G dC dDe      Z G dE dFe      Z G dG dHe      Z G dI dJe      ZyK(L     )DummyObjectrequires_backendsc                   8    e Zd ZdgZd Zed        Zed        Zy)AdaptiveProjectedGuidancetorchc                     t        | dg       y Nr   r   selfargskwargss      [/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/utils/dummy_pt_objects.py__init__z"AdaptiveProjectedGuidance.__init__       $	*    c                     t        | dg       y r	   r
   clsr   r   s      r   from_configz%AdaptiveProjectedGuidance.from_config       #y)r   c                     t        | dg       y r	   r
   r   s      r   from_pretrainedz)AdaptiveProjectedGuidance.from_pretrained   r   r   N__name__
__module____qualname__	_backendsr   classmethodr   r    r   r   r   r      6    	I+ * * * *r   r   )	metaclassc                   8    e Zd ZdgZd Zed        Zed        Zy)AutoGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   zAutoGuidance.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoGuidance.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoGuidance.from_pretrained   r   r   Nr   r    r   r   r$   r$      r!   r   r$   c                   8    e Zd ZdgZd Zed        Zed        Zy)ClassifierFreeGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   zClassifierFreeGuidance.__init__&   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"ClassifierFreeGuidance.from_config)   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&ClassifierFreeGuidance.from_pretrained-   r   r   Nr   r    r   r   r)   r)   #   r!   r   r)   c                   8    e Zd ZdgZd Zed        Zed        Zy)ClassifierFreeZeroStarGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   z'ClassifierFreeZeroStarGuidance.__init__5   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*ClassifierFreeZeroStarGuidance.from_config8   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.ClassifierFreeZeroStarGuidance.from_pretrained<   r   r   Nr   r    r   r   r.   r.   2   r!   r   r.   c                   8    e Zd ZdgZd Zed        Zed        Zy)FrequencyDecoupledGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   z#FrequencyDecoupledGuidance.__init__D   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&FrequencyDecoupledGuidance.from_configG   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*FrequencyDecoupledGuidance.from_pretrainedK   r   r   Nr   r    r   r   r3   r3   A   r!   r   r3   c                   8    e Zd ZdgZd Zed        Zed        Zy)PerturbedAttentionGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   z#PerturbedAttentionGuidance.__init__S   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&PerturbedAttentionGuidance.from_configV   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*PerturbedAttentionGuidance.from_pretrainedZ   r   r   Nr   r    r   r   r8   r8   P   r!   r   r8   c                   8    e Zd ZdgZd Zed        Zed        Zy)SkipLayerGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   zSkipLayerGuidance.__init__b   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSkipLayerGuidance.from_confige   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!SkipLayerGuidance.from_pretrainedi   r   r   Nr   r    r   r   r=   r=   _   r!   r   r=   c                   8    e Zd ZdgZd Zed        Zed        Zy)SmoothedEnergyGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   zSmoothedEnergyGuidance.__init__q   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"SmoothedEnergyGuidance.from_configt   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&SmoothedEnergyGuidance.from_pretrainedx   r   r   Nr   r    r   r   rB   rB   n   r!   r   rB   c                   8    e Zd ZdgZd Zed        Zed        Zy) TangentialClassifierFreeGuidancer   c                     t        | dg       y r	   r
   r   s      r   r   z)TangentialClassifierFreeGuidance.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,TangentialClassifierFreeGuidance.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z0TangentialClassifierFreeGuidance.from_pretrained   r   r   Nr   r    r   r   rG   rG   }   r!   r   rG   c                   8    e Zd ZdgZd Zed        Zed        Zy)FasterCacheConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zFasterCacheConfig.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zFasterCacheConfig.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!FasterCacheConfig.from_pretrained   r   r   Nr   r    r   r   rL   rL      r!   r   rL   c                   8    e Zd ZdgZd Zed        Zed        Zy)FirstBlockCacheConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zFirstBlockCacheConfig.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!FirstBlockCacheConfig.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%FirstBlockCacheConfig.from_pretrained   r   r   Nr   r    r   r   rQ   rQ      r!   r   rQ   c                   8    e Zd ZdgZd Zed        Zed        Zy)HookRegistryr   c                     t        | dg       y r	   r
   r   s      r   r   zHookRegistry.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zHookRegistry.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zHookRegistry.from_pretrained   r   r   Nr   r    r   r   rV   rV      r!   r   rV   c                   8    e Zd ZdgZd Zed        Zed        Zy)LayerSkipConfigr   c                     t        | dg       y r	   r
   r   s      r   r   zLayerSkipConfig.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLayerSkipConfig.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLayerSkipConfig.from_pretrained   r   r   Nr   r    r   r   r[   r[      r!   r   r[   c                   8    e Zd ZdgZd Zed        Zed        Zy)PyramidAttentionBroadcastConfigr   c                     t        | dg       y r	   r
   r   s      r   r   z(PyramidAttentionBroadcastConfig.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+PyramidAttentionBroadcastConfig.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z/PyramidAttentionBroadcastConfig.from_pretrained   r   r   Nr   r    r   r   r`   r`      r!   r   r`   c                   8    e Zd ZdgZd Zed        Zed        Zy)SmoothedEnergyGuidanceConfigr   c                     t        | dg       y r	   r
   r   s      r   r   z%SmoothedEnergyGuidanceConfig.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(SmoothedEnergyGuidanceConfig.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,SmoothedEnergyGuidanceConfig.from_pretrained   r   r   Nr   r    r   r   re   re      r!   r   re   c                  &    t        t        dg       y r	   )r   apply_faster_cacher   r   s     r   rj   rj      s    (7)4r   c                  &    t        t        dg       y r	   )r   apply_first_block_cacherk   s     r   rm   rm      s    -y9r   c                  &    t        t        dg       y r	   )r   apply_layer_skiprk   s     r   ro   ro      s    &	2r   c                  &    t        t        dg       y r	   )r   !apply_pyramid_attention_broadcastrk   s     r   rq   rq          7'Cr   c                   8    e Zd ZdgZd Zed        Zed        Zy)AllegroTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"AllegroTransformer3DModel.__init__   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%AllegroTransformer3DModel.from_config   r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)AllegroTransformer3DModel.from_pretrained   r   r   Nr   r    r   r   rt   rt      r!   r   rt   c                   8    e Zd ZdgZd Zed        Zed        Zy)AsymmetricAutoencoderKLr   c                     t        | dg       y r	   r
   r   s      r   r   z AsymmetricAutoencoderKL.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#AsymmetricAutoencoderKL.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'AsymmetricAutoencoderKL.from_pretrained  r   r   Nr   r    r   r   ry   ry     r!   r   ry   c                   8    e Zd ZdgZd Zed        Zed        Zy)AttentionBackendNamer   c                     t        | dg       y r	   r
   r   s      r   r   zAttentionBackendName.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z AttentionBackendName.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$AttentionBackendName.from_pretrained  r   r   Nr   r    r   r   r~   r~     r!   r   r~   c                   8    e Zd ZdgZd Zed        Zed        Zy)AuraFlowTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#AuraFlowTransformer2DModel.__init__&  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&AuraFlowTransformer2DModel.from_config)  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*AuraFlowTransformer2DModel.from_pretrained-  r   r   Nr   r    r   r   r   r   #  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderDCr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderDC.__init__5  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderDC.from_config8  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderDC.from_pretrained<  r   r   Nr   r    r   r   r   r   2  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKL.__init__D  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKL.from_configG  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKL.from_pretrainedK  r   r   Nr   r    r   r   r   r   A  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLAllegror   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLAllegro.__init__S  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z AutoencoderKLAllegro.from_configV  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$AutoencoderKLAllegro.from_pretrainedZ  r   r   Nr   r    r   r   r   r   P  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLCogVideoXr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLCogVideoX.__init__b  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoencoderKLCogVideoX.from_confige  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoencoderKLCogVideoX.from_pretrainedi  r   r   Nr   r    r   r   r   r   _  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLCosmosr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLCosmos.__init__q  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLCosmos.from_configt  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoencoderKLCosmos.from_pretrainedx  r   r   Nr   r    r   r   r   r   n  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLHunyuanVideor   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoencoderKLHunyuanVideo.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%AutoencoderKLHunyuanVideo.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)AutoencoderKLHunyuanVideo.from_pretrained  r   r   Nr   r    r   r   r   r   }  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLLTXVideor   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLLTXVideo.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!AutoencoderKLLTXVideo.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%AutoencoderKLLTXVideo.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLMagvitr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLMagvit.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLMagvit.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoencoderKLMagvit.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLMochir   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLMochi.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLMochi.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoencoderKLMochi.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLQwenImager   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLQwenImage.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoencoderKLQwenImage.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoencoderKLQwenImage.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLTemporalDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   z%AutoencoderKLTemporalDecoder.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(AutoencoderKLTemporalDecoder.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,AutoencoderKLTemporalDecoder.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderKLWanr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLWan.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderKLWan.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z AutoencoderKLWan.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderOobleckr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderOobleck.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderOobleck.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoencoderOobleck.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoencoderTinyr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderTiny.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderTiny.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoencoderTiny.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)	AutoModelr   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModel.from_config
  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAutoModel.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)
CacheMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zCacheMixin.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zCacheMixin.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zCacheMixin.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ChromaTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!ChromaTransformer2DModel.__init__%  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$ChromaTransformer2DModel.from_config(  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(ChromaTransformer2DModel.from_pretrained,  r   r   Nr   r    r   r   r   r   "  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)CogVideoXTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$CogVideoXTransformer3DModel.__init__4  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'CogVideoXTransformer3DModel.from_config7  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+CogVideoXTransformer3DModel.from_pretrained;  r   r   Nr   r    r   r   r   r   1  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)CogView3PlusTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'CogView3PlusTransformer2DModel.__init__C  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*CogView3PlusTransformer2DModel.from_configF  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.CogView3PlusTransformer2DModel.from_pretrainedJ  r   r   Nr   r    r   r   r   r   @  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)CogView4Transformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#CogView4Transformer2DModel.__init__R  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&CogView4Transformer2DModel.from_configU  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*CogView4Transformer2DModel.from_pretrainedY  r   r   Nr   r    r   r   r   r   O  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ConsisIDTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#ConsisIDTransformer3DModel.__init__a  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&ConsisIDTransformer3DModel.from_configd  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*ConsisIDTransformer3DModel.from_pretrainedh  r   r   Nr   r    r   r   r   r   ^  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ConsistencyDecoderVAEr   c                     t        | dg       y r	   r
   r   s      r   r   zConsistencyDecoderVAE.__init__p  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!ConsistencyDecoderVAE.from_configs  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%ConsistencyDecoderVAE.from_pretrainedw  r   r   Nr   r    r   r   r   r   m  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetModel.from_pretrained  r   r   Nr   r    r   r   r   r   |  r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ControlNetUnionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetUnionModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z ControlNetUnionModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$ControlNetUnionModel.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ControlNetXSAdapterr   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetXSAdapter.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zControlNetXSAdapter.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#ControlNetXSAdapter.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)CosmosTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!CosmosTransformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$CosmosTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(CosmosTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DiTTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zDiTTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!DiTTransformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%DiTTransformer2DModel.from_pretrained  r   r   Nr   r    r   r   r
  r
    r!   r   r
  c                   8    e Zd ZdgZd Zed        Zed        Zy)EasyAnimateTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z&EasyAnimateTransformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)EasyAnimateTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z-EasyAnimateTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FluxControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFluxControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zFluxControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#FluxControlNetModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FluxMultiControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!FluxMultiControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$FluxMultiControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(FluxMultiControlNetModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FluxTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zFluxTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"FluxTransformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&FluxTransformer2DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)HiDreamImageTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'HiDreamImageTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*HiDreamImageTransformer2DModel.from_config	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.HiDreamImageTransformer2DModel.from_pretrained  r   r   Nr   r    r   r   r#  r#    r!   r   r#  c                   8    e Zd ZdgZd Zed        Zed        Zy)HunyuanDiT2DControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$HunyuanDiT2DControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'HunyuanDiT2DControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+HunyuanDiT2DControlNetModel.from_pretrained  r   r   Nr   r    r   r   r(  r(    r!   r   r(  c                   8    e Zd ZdgZd Zed        Zed        Zy)HunyuanDiT2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zHunyuanDiT2DModel.__init__$  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zHunyuanDiT2DModel.from_config'  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!HunyuanDiT2DModel.from_pretrained+  r   r   Nr   r    r   r   r-  r-  !  r!   r   r-  c                   8    e Zd ZdgZd Zed        Zed        Zy) HunyuanDiT2DMultiControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   z)HunyuanDiT2DMultiControlNetModel.__init__3  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,HunyuanDiT2DMultiControlNetModel.from_config6  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z0HunyuanDiT2DMultiControlNetModel.from_pretrained:  r   r   Nr   r    r   r   r2  r2  0  r!   r   r2  c                   8    e Zd ZdgZd Zed        Zed        Zy)'HunyuanVideoFramepackTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z0HunyuanVideoFramepackTransformer3DModel.__init__B  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z3HunyuanVideoFramepackTransformer3DModel.from_configE  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z7HunyuanVideoFramepackTransformer3DModel.from_pretrainedI  r   r   Nr   r    r   r   r7  r7  ?  r!   r   r7  c                   8    e Zd ZdgZd Zed        Zed        Zy)HunyuanVideoTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z'HunyuanVideoTransformer3DModel.__init__Q  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*HunyuanVideoTransformer3DModel.from_configT  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.HunyuanVideoTransformer3DModel.from_pretrainedX  r   r   Nr   r    r   r   r<  r<  N  r!   r   r<  c                   8    e Zd ZdgZd Zed        Zed        Zy)I2VGenXLUNetr   c                     t        | dg       y r	   r
   r   s      r   r   zI2VGenXLUNet.__init__`  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zI2VGenXLUNet.from_configc  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zI2VGenXLUNet.from_pretrainedg  r   r   Nr   r    r   r   rA  rA  ]  r!   r   rA  c                   8    e Zd ZdgZd Zed        Zed        Zy)Kandinsky3UNetr   c                     t        | dg       y r	   r
   r   s      r   r   zKandinsky3UNet.__init__o  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zKandinsky3UNet.from_configr  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zKandinsky3UNet.from_pretrainedv  r   r   Nr   r    r   r   rF  rF  l  r!   r   rF  c                   8    e Zd ZdgZd Zed        Zed        Zy)LatteTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z LatteTransformer3DModel.__init__~  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#LatteTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'LatteTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   rK  rK  {  r!   r   rK  c                   8    e Zd ZdgZd Zed        Zed        Zy)LTXVideoTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z#LTXVideoTransformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&LTXVideoTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*LTXVideoTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   rP  rP    r!   r   rP  c                   8    e Zd ZdgZd Zed        Zed        Zy)Lumina2Transformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"Lumina2Transformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%Lumina2Transformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)Lumina2Transformer2DModel.from_pretrained  r   r   Nr   r    r   r   rU  rU    r!   r   rU  c                   8    e Zd ZdgZd Zed        Zed        Zy)LuminaNextDiT2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zLuminaNextDiT2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z LuminaNextDiT2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$LuminaNextDiT2DModel.from_pretrained  r   r   Nr   r    r   r   rZ  rZ    r!   r   rZ  c                   8    e Zd ZdgZd Zed        Zed        Zy)MochiTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z MochiTransformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#MochiTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'MochiTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   r_  r_    r!   r   r_  c                   8    e Zd ZdgZd Zed        Zed        Zy)
ModelMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zModelMixin.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zModelMixin.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zModelMixin.from_pretrained  r   r   Nr   r    r   r   rd  rd    r!   r   rd  c                   8    e Zd ZdgZd Zed        Zed        Zy)MotionAdapterr   c                     t        | dg       y r	   r
   r   s      r   r   zMotionAdapter.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zMotionAdapter.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zMotionAdapter.from_pretrained  r   r   Nr   r    r   r   ri  ri    r!   r   ri  c                   8    e Zd ZdgZd Zed        Zed        Zy)MultiAdapterr   c                     t        | dg       y r	   r
   r   s      r   r   zMultiAdapter.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zMultiAdapter.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zMultiAdapter.from_pretrained  r   r   Nr   r    r   r   rn  rn    r!   r   rn  c                   8    e Zd ZdgZd Zed        Zed        Zy)MultiControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zMultiControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z MultiControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$MultiControlNetModel.from_pretrained  r   r   Nr   r    r   r   rs  rs    r!   r   rs  c                   8    e Zd ZdgZd Zed        Zed        Zy)OmniGenTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"OmniGenTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%OmniGenTransformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)OmniGenTransformer2DModel.from_pretrained  r   r   Nr   r    r   r   rx  rx    r!   r   rx  c                   8    e Zd ZdgZd Zed        Zed        Zy)PixArtTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!PixArtTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$PixArtTransformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(PixArtTransformer2DModel.from_pretrained  r   r   Nr   r    r   r   r}  r}    r!   r   r}  c                   8    e Zd ZdgZd Zed        Zed        Zy)PriorTransformerr   c                     t        | dg       y r	   r
   r   s      r   r   zPriorTransformer.__init__#  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zPriorTransformer.from_config&  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z PriorTransformer.from_pretrained*  r   r   Nr   r    r   r   r  r     r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)QwenImageTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z$QwenImageTransformer2DModel.__init__2  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'QwenImageTransformer2DModel.from_config5  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+QwenImageTransformer2DModel.from_pretrained9  r   r   Nr   r    r   r   r  r  /  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SanaControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSanaControlNetModel.__init__A  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSanaControlNetModel.from_configD  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#SanaControlNetModel.from_pretrainedH  r   r   Nr   r    r   r   r  r  >  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SanaTransformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSanaTransformer2DModel.__init__P  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"SanaTransformer2DModel.from_configS  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&SanaTransformer2DModel.from_pretrainedW  r   r   Nr   r    r   r   r  r  M  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SD3ControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSD3ControlNetModel.__init___  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSD3ControlNetModel.from_configb  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"SD3ControlNetModel.from_pretrainedf  r   r   Nr   r    r   r   r  r  \  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SD3MultiControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   z SD3MultiControlNetModel.__init__n  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#SD3MultiControlNetModel.from_configq  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'SD3MultiControlNetModel.from_pretrainedu  r   r   Nr   r    r   r   r  r  k  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SD3Transformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSD3Transformer2DModel.__init__}  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!SD3Transformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%SD3Transformer2DModel.from_pretrained  r   r   Nr   r    r   r   r  r  z  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SkyReelsV2Transformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z%SkyReelsV2Transformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(SkyReelsV2Transformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,SkyReelsV2Transformer3DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SparseControlNetModelr   c                     t        | dg       y r	   r
   r   s      r   r   zSparseControlNetModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!SparseControlNetModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%SparseControlNetModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)StableAudioDiTModelr   c                     t        | dg       y r	   r
   r   s      r   r   zStableAudioDiTModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zStableAudioDiTModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#StableAudioDiTModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)
T2IAdapterr   c                     t        | dg       y r	   r
   r   s      r   r   zT2IAdapter.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zT2IAdapter.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zT2IAdapter.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)T5FilmDecoderr   c                     t        | dg       y r	   r
   r   s      r   r   zT5FilmDecoder.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zT5FilmDecoder.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zT5FilmDecoder.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)Transformer2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zTransformer2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zTransformer2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"Transformer2DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)TransformerTemporalModelr   c                     t        | dg       y r	   r
   r   s      r   r   z!TransformerTemporalModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$TransformerTemporalModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(TransformerTemporalModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNet1DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNet1DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNet1DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNet1DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNet2DConditionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNet2DConditionModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z UNet2DConditionModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$UNet2DConditionModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNet2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNet2DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNet2DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNet2DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNet3DConditionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNet3DConditionModel.__init__"  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z UNet3DConditionModel.from_config%  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$UNet3DConditionModel.from_pretrained)  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNetControlNetXSModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNetControlNetXSModel.__init__1  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!UNetControlNetXSModel.from_config4  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%UNetControlNetXSModel.from_pretrained8  r   r   Nr   r    r   r   r  r  .  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UNetMotionModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUNetMotionModel.__init__@  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNetMotionModel.from_configC  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUNetMotionModel.from_pretrainedG  r   r   Nr   r    r   r   r  r  =  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy) UNetSpatioTemporalConditionModelr   c                     t        | dg       y r	   r
   r   s      r   r   z)UNetSpatioTemporalConditionModel.__init__O  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,UNetSpatioTemporalConditionModel.from_configR  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z0UNetSpatioTemporalConditionModel.from_pretrainedV  r   r   Nr   r    r   r   r  r  L  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)UVit2DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zUVit2DModel.__init__^  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUVit2DModel.from_configa  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUVit2DModel.from_pretrainede  r   r   Nr   r    r   r   r  r  [  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)VQModelr   c                     t        | dg       y r	   r
   r   s      r   r   zVQModel.__init__m  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zVQModel.from_configp  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zVQModel.from_pretrainedt  r   r   Nr   r    r   r   r  r  j  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)WanTransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   zWanTransformer3DModel.__init__|  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!WanTransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%WanTransformer3DModel.from_pretrained  r   r   Nr   r    r   r   r  r  y  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)WanVACETransformer3DModelr   c                     t        | dg       y r	   r
   r   s      r   r   z"WanVACETransformer3DModel.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%WanVACETransformer3DModel.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)WanVACETransformer3DModel.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                  &    t        t        dg       y r	   )r   attention_backendrk   s     r   r  r    s    ''3r   c                   8    e Zd ZdgZd Zed        Zed        Zy)ComponentsManagerr   c                     t        | dg       y r	   r
   r   s      r   r   zComponentsManager.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zComponentsManager.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!ComponentsManager.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)ComponentSpecr   c                     t        | dg       y r	   r
   r   s      r   r   zComponentSpec.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zComponentSpec.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zComponentSpec.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)ModularPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zModularPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zModularPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zModularPipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)ModularPipelineBlocksr   c                     t        | dg       y r	   r
   r   s      r   r   zModularPipelineBlocks.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!ModularPipelineBlocks.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%ModularPipelineBlocks.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                  &    t        t        dg       y r	   )r   get_constant_schedulerk   s     r   r  r    s    +gY7r   c                  &    t        t        dg       y r	   )r   !get_constant_schedule_with_warmuprk   s     r   r  r    rr   r   c                  &    t        t        dg       y r	   )r   get_cosine_schedule_with_warmuprk   s     r   r  r        5yAr   c                  &    t        t        dg       y r	   )r   2get_cosine_with_hard_restarts_schedule_with_warmuprk   s     r   r  r    s    H7)Tr   c                  &    t        t        dg       y r	   )r   get_linear_schedule_with_warmuprk   s     r   r  r    r  r   c                  &    t        t        dg       y r	   )r   )get_polynomial_decay_schedule_with_warmuprk   s     r   r   r     s    ?'Kr   c                  &    t        t        dg       y r	   )r   get_schedulerrk   s     r   r"  r"    s    mgY/r   c                   8    e Zd ZdgZd Zed        Zed        Zy)AudioPipelineOutputr   c                     t        | dg       y r	   r
   r   s      r   r   zAudioPipelineOutput.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAudioPipelineOutput.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#AudioPipelineOutput.from_pretrained  r   r   Nr   r    r   r   r$  r$    r!   r   r$  c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoPipelineForImage2Imager   c                     t        | dg       y r	   r
   r   s      r   r   z#AutoPipelineForImage2Image.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&AutoPipelineForImage2Image.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*AutoPipelineForImage2Image.from_pretrained  r   r   Nr   r    r   r   r)  r)    r!   r   r)  c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoPipelineForInpaintingr   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoPipelineForInpainting.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%AutoPipelineForInpainting.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)AutoPipelineForInpainting.from_pretrained  r   r   Nr   r    r   r   r.  r.    r!   r   r.  c                   8    e Zd ZdgZd Zed        Zed        Zy)AutoPipelineForText2Imager   c                     t        | dg       y r	   r
   r   s      r   r   z"AutoPipelineForText2Image.__init__#  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%AutoPipelineForText2Image.from_config&  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z)AutoPipelineForText2Image.from_pretrained*  r   r   Nr   r    r   r   r3  r3     r!   r   r3  c                   8    e Zd ZdgZd Zed        Zed        Zy)BlipDiffusionControlNetPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   z(BlipDiffusionControlNetPipeline.__init__2  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+BlipDiffusionControlNetPipeline.from_config5  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z/BlipDiffusionControlNetPipeline.from_pretrained9  r   r   Nr   r    r   r   r8  r8  /  r!   r   r8  c                   8    e Zd ZdgZd Zed        Zed        Zy)BlipDiffusionPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zBlipDiffusionPipeline.__init__A  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!BlipDiffusionPipeline.from_configD  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%BlipDiffusionPipeline.from_pretrainedH  r   r   Nr   r    r   r   r=  r=  >  r!   r   r=  c                   8    e Zd ZdgZd Zed        Zed        Zy)CLIPImageProjectionr   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPImageProjection.__init__P  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zCLIPImageProjection.from_configS  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#CLIPImageProjection.from_pretrainedW  r   r   Nr   r    r   r   rB  rB  M  r!   r   rB  c                   8    e Zd ZdgZd Zed        Zed        Zy)ConsistencyModelPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   z!ConsistencyModelPipeline.__init___  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$ConsistencyModelPipeline.from_configb  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(ConsistencyModelPipeline.from_pretrainedf  r   r   Nr   r    r   r   rG  rG  \  r!   r   rG  c                   8    e Zd ZdgZd Zed        Zed        Zy)DanceDiffusionPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zDanceDiffusionPipeline.__init__n  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"DanceDiffusionPipeline.from_configq  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&DanceDiffusionPipeline.from_pretrainedu  r   r   Nr   r    r   r   rL  rL  k  r!   r   rL  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDIMPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMPipeline.__init__}  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMPipeline.from_pretrained  r   r   Nr   r    r   r   rQ  rQ  z  r!   r   rQ  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDPMPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMPipeline.from_pretrained  r   r   Nr   r    r   r   rV  rV    r!   r   rV  c                   8    e Zd ZdgZd Zed        Zed        Zy)DiffusionPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zDiffusionPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDiffusionPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!DiffusionPipeline.from_pretrained  r   r   Nr   r    r   r   r[  r[    r!   r   r[  c                   8    e Zd ZdgZd Zed        Zed        Zy)DiTPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zDiTPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDiTPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDiTPipeline.from_pretrained  r   r   Nr   r    r   r   r`  r`    r!   r   r`  c                   8    e Zd ZdgZd Zed        Zed        Zy)ImagePipelineOutputr   c                     t        | dg       y r	   r
   r   s      r   r   zImagePipelineOutput.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zImagePipelineOutput.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#ImagePipelineOutput.from_pretrained  r   r   Nr   r    r   r   re  re    r!   r   re  c                   8    e Zd ZdgZd Zed        Zed        Zy)KarrasVePipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zKarrasVePipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zKarrasVePipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z KarrasVePipeline.from_pretrained  r   r   Nr   r    r   r   rj  rj    r!   r   rj  c                   8    e Zd ZdgZd Zed        Zed        Zy)LDMPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zLDMPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLDMPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLDMPipeline.from_pretrained  r   r   Nr   r    r   r   ro  ro    r!   r   ro  c                   8    e Zd ZdgZd Zed        Zed        Zy)LDMSuperResolutionPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   z#LDMSuperResolutionPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&LDMSuperResolutionPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*LDMSuperResolutionPipeline.from_pretrained  r   r   Nr   r    r   r   rt  rt    r!   r   rt  c                   8    e Zd ZdgZd Zed        Zed        Zy)PNDMPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMPipeline.from_pretrained  r   r   Nr   r    r   r   ry  ry    r!   r   ry  c                   8    e Zd ZdgZd Zed        Zed        Zy)RePaintPipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zRePaintPipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zRePaintPipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zRePaintPipeline.from_pretrained  r   r   Nr   r    r   r   r~  r~    r!   r   r~  c                   8    e Zd ZdgZd Zed        Zed        Zy)ScoreSdeVePipeliner   c                     t        | dg       y r	   r
   r   s      r   r   zScoreSdeVePipeline.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zScoreSdeVePipeline.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"ScoreSdeVePipeline.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)StableDiffusionMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zStableDiffusionMixin.__init__"  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z StableDiffusionMixin.from_config%  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$StableDiffusionMixin.from_pretrained)  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DiffusersQuantizerr   c                     t        | dg       y r	   r
   r   s      r   r   zDiffusersQuantizer.__init__1  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDiffusersQuantizer.from_config4  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"DiffusersQuantizer.from_pretrained8  r   r   Nr   r    r   r   r  r  .  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)AmusedSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zAmusedScheduler.__init__@  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAmusedScheduler.from_configC  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zAmusedScheduler.from_pretrainedG  r   r   Nr   r    r   r   r  r  =  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)CMStochasticIterativeSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z'CMStochasticIterativeScheduler.__init__O  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*CMStochasticIterativeScheduler.from_configR  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.CMStochasticIterativeScheduler.from_pretrainedV  r   r   Nr   r    r   r   r  r  L  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)CogVideoXDDIMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zCogVideoXDDIMScheduler.__init__^  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"CogVideoXDDIMScheduler.from_configa  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&CogVideoXDDIMScheduler.from_pretrainede  r   r   Nr   r    r   r   r  r  [  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)CogVideoXDPMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zCogVideoXDPMScheduler.__init__m  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!CogVideoXDPMScheduler.from_configp  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%CogVideoXDPMScheduler.from_pretrainedt  r   r   Nr   r    r   r   r  r  j  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDIMInverseSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMInverseScheduler.__init__|  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z DDIMInverseScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$DDIMInverseScheduler.from_pretrained  r   r   Nr   r    r   r   r  r  y  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDIMParallelSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMParallelScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!DDIMParallelScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%DDIMParallelScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDIMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDIMScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDPMParallelSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMParallelScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!DDPMParallelScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%DDPMParallelScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDPMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zDDPMScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DDPMWuerstchenSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z DDPMWuerstchenScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#DDPMWuerstchenScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'DDPMWuerstchenScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DEISMultistepSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zDEISMultistepScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"DEISMultistepScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&DEISMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)"DPMSolverMultistepInverseSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z+DPMSolverMultistepInverseScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.DPMSolverMultistepInverseScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z2DPMSolverMultistepInverseScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DPMSolverMultistepSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z$DPMSolverMultistepScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'DPMSolverMultistepScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+DPMSolverMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)DPMSolverSinglestepSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z%DPMSolverSinglestepScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z(DPMSolverSinglestepScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z,DPMSolverSinglestepScheduler.from_pretrained
  r   r   Nr   r    r   r   r  r     r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)EDMDPMSolverMultistepSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z'EDMDPMSolverMultistepScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*EDMDPMSolverMultistepScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.EDMDPMSolverMultistepScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)EDMEulerSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zEDMEulerScheduler.__init__!  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zEDMEulerScheduler.from_config$  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!EDMEulerScheduler.from_pretrained(  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)EulerAncestralDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z(EulerAncestralDiscreteScheduler.__init__0  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+EulerAncestralDiscreteScheduler.from_config3  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z/EulerAncestralDiscreteScheduler.from_pretrained7  r   r   Nr   r    r   r   r  r  -  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)EulerDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zEulerDiscreteScheduler.__init__?  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"EulerDiscreteScheduler.from_configB  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&EulerDiscreteScheduler.from_pretrainedF  r   r   Nr   r    r   r   r  r  <  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FlowMatchEulerDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z(FlowMatchEulerDiscreteScheduler.__init__N  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+FlowMatchEulerDiscreteScheduler.from_configQ  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z/FlowMatchEulerDiscreteScheduler.from_pretrainedU  r   r   Nr   r    r   r   r  r  K  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FlowMatchHeunDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z'FlowMatchHeunDiscreteScheduler.__init__]  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z*FlowMatchHeunDiscreteScheduler.from_config`  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z.FlowMatchHeunDiscreteScheduler.from_pretrainedd  r   r   Nr   r    r   r   r  r  Z  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)FlowMatchLCMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zFlowMatchLCMScheduler.__init__l  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!FlowMatchLCMScheduler.from_configo  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%FlowMatchLCMScheduler.from_pretraineds  r   r   Nr   r    r   r   r  r  i  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)HeunDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zHeunDiscreteScheduler.__init__{  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!HeunDiscreteScheduler.from_config~  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z%HeunDiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   r  r  x  r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)IPNDMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zIPNDMScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zIPNDMScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zIPNDMScheduler.from_pretrained  r   r   Nr   r    r   r   r   r     r!   r   r   c                   8    e Zd ZdgZd Zed        Zed        Zy)KarrasVeSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zKarrasVeScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zKarrasVeScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!KarrasVeScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)KDPM2AncestralDiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z(KDPM2AncestralDiscreteScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z+KDPM2AncestralDiscreteScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z/KDPM2AncestralDiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   r
  r
    r!   r   r
  c                   8    e Zd ZdgZd Zed        Zed        Zy)KDPM2DiscreteSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zKDPM2DiscreteScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z"KDPM2DiscreteScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z&KDPM2DiscreteScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)LCMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zLCMScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLCMScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zLCMScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)PNDMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zPNDMScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)RePaintSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zRePaintScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zRePaintScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z RePaintScheduler.from_pretrained  r   r   Nr   r    r   r   r  r    r!   r   r  c                   8    e Zd ZdgZd Zed        Zed        Zy)SASolverSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zSASolverScheduler.__init__  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSASolverScheduler.from_config  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z!SASolverScheduler.from_pretrained  r   r   Nr   r    r   r   r#  r#    r!   r   r#  c                   8    e Zd ZdgZd Zed        Zed        Zy)SchedulerMixinr   c                     t        | dg       y r	   r
   r   s      r   r   zSchedulerMixin.__init__	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSchedulerMixin.from_config	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSchedulerMixin.from_pretrained		  r   r   Nr   r    r   r   r(  r(    r!   r   r(  c                   8    e Zd ZdgZd Zed        Zed        Zy)SCMSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zSCMScheduler.__init__	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSCMScheduler.from_config	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zSCMScheduler.from_pretrained	  r   r   Nr   r    r   r   r-  r-  	  r!   r   r-  c                   8    e Zd ZdgZd Zed        Zed        Zy)ScoreSdeVeSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zScoreSdeVeScheduler.__init__ 	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zScoreSdeVeScheduler.from_config#	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#ScoreSdeVeScheduler.from_pretrained'	  r   r   Nr   r    r   r   r2  r2  	  r!   r   r2  c                   8    e Zd ZdgZd Zed        Zed        Zy)TCDSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zTCDScheduler.__init__/	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zTCDScheduler.from_config2	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zTCDScheduler.from_pretrained6	  r   r   Nr   r    r   r   r7  r7  ,	  r!   r   r7  c                   8    e Zd ZdgZd Zed        Zed        Zy)UnCLIPSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zUnCLIPScheduler.__init__>	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUnCLIPScheduler.from_configA	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zUnCLIPScheduler.from_pretrainedE	  r   r   Nr   r    r   r   r<  r<  ;	  r!   r   r<  c                   8    e Zd ZdgZd Zed        Zed        Zy)UniPCMultistepSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   z UniPCMultistepScheduler.__init__M	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z#UniPCMultistepScheduler.from_configP	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z'UniPCMultistepScheduler.from_pretrainedT	  r   r   Nr   r    r   r   rA  rA  J	  r!   r   rA  c                   8    e Zd ZdgZd Zed        Zed        Zy)VQDiffusionSchedulerr   c                     t        | dg       y r	   r
   r   s      r   r   zVQDiffusionScheduler.__init__\	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z VQDiffusionScheduler.from_config_	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   z$VQDiffusionScheduler.from_pretrainedc	  r   r   Nr   r    r   r   rF  rF  Y	  r!   r   rF  c                   8    e Zd ZdgZd Zed        Zed        Zy)EMAModelr   c                     t        | dg       y r	   r
   r   s      r   r   zEMAModel.__init__k	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zEMAModel.from_confign	  r   r   c                     t        | dg       y r	   r
   r   s      r   r   zEMAModel.from_pretrainedr	  r   r   Nr   r    r   r   rK  rK  h	  r!   r   rK  N)utilsr   r   r   r$   r)   r.   r3   r8   r=   rB   rG   rL   rQ   rV   r[   r`   re   rj   rm   ro   rq   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-  r2  r7  r<  rA  rF  rK  rP  rU  rZ  r_  rd  ri  rn  rs  rx  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r   r"  r$  r)  r.  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-  r2  r7  r<  rA  rF  rK  r    r   r   <module>rP     s   2*+ **[ **{ **{ **; **; **+ **{ ** **+ **k **[ ** ** **[ *5:3D*+ ** **[ **; **k **k **[ **{ **K **+ **k **K **; **{ **[ ** **; ** **+ **; ** **K **{ **; **; **k ** **[ **K ** **k **k **K ** **{ **{ **K **+ ** ** **{ **[ **{ ** **; **+ **[ ** **; **k **[ **[ **+ ** ** **K **K **{ **; ** **k **[ **k **K **; **k **; ** **K **[ **K **[ **k ** ** **K ** **k **+ *4*+ **k ** **k *8DBUBL0*K **; **+ **+ ** **k **K ** **{ **[ **[ **+ **K **K ** **K **; **[ ** **; **[ **; ** **{ **{ **k **[ **k **k **k **k ** **{ **; **K **[ **{ **+ ** **{ ** **{ **k **k **{ **+ ** **{ **[ **k ** **+ **{ **[ **K **[ ** ** **[ ** *r   