
    +h                        S SK JrJr   " S S\S9r " S S\S9r " S S\S9r " S	 S
\S9r " S S\S9r " S S\S9r " S S\S9r	 " S S\S9r
 " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S \S9rS! rS" rS# rS$ r " S% S&\S9r " S' S(\S9r " S) S*\S9r " S+ S,\S9r " S- S.\S9r " S/ S0\S9r " S1 S2\S9r " S3 S4\S9r " S5 S6\S9r " S7 S8\S9r " S9 S:\S9r  " S; S<\S9r! " S= S>\S9r" " S? S@\S9r# " SA SB\S9r$ " SC SD\S9r% " SE SF\S9r& " SG SH\S9r' " SI SJ\S9r( " SK SL\S9r) " SM SN\S9r* " SO SP\S9r+ " SQ SR\S9r, " SS ST\S9r- " SU SV\S9r. " SW SX\S9r/ " SY SZ\S9r0 " S[ S\\S9r1 " S] S^\S9r2 " S_ S`\S9r3 " Sa Sb\S9r4 " Sc Sd\S9r5 " Se Sf\S9r6 " Sg Sh\S9r7 " Si Sj\S9r8 " Sk Sl\S9r9 " Sm Sn\S9r: " So Sp\S9r; " Sq Sr\S9r< " Ss St\S9r= " Su Sv\S9r> " Sw Sx\S9r? " Sy Sz\S9r@ " S{ S|\S9rA " S} S~\S9rB " S S\S9rC " S S\S9rD " S S\S9rE " S S\S9rF " S S\S9rG " S S\S9rH " S S\S9rI " S S\S9rJ " S S\S9rK " S S\S9rL " S S\S9rM " S S\S9rN " S S\S9rO " S S\S9rP " S S\S9rQ " S S\S9rR " S S\S9rS " S S\S9rT " S S\S9rU " S S\S9rV " S S\S9rW " S S\S9rX " S S\S9rY " S S\S9rZ " S S\S9r[ " S S\S9r\ " S S\S9r] " S S\S9r^ " S S\S9r_ " S S\S9r` " S S\S9ra " S S\S9rb " S S\S9rc " S S\S9rdS re " S S\S9rf " S S\S9rg " S S\S9rh " S S\S9riS rjS rkS rlS rmS rnS roS rp " S S\S9rq " S S\S9rr " S S\S9rs " S S\S9rt " S S\S9ru " S S\S9rv " S S\S9rw " S S\S9rx " S S\S9ry " S S\S9rz " S S\S9r{ " S S\S9r| " S S\S9r} " S S\S9r~ " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S S\S9r " S GS \S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS	 GS
\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS\S9r " GS GS \S9r " GS! GS"\S9r " GS# GS$\S9r " GS% GS&\S9r " GS' GS(\S9r " GS) GS*\S9r " GS+ GS,\S9r " GS- GS.\S9r " GS/ GS0\S9r " GS1 GS2\S9r " GS3 GS4\S9r " GS5 GS6\S9r " GS7 GS8\S9r " GS9 GS:\S9r " GS; GS<\S9r " GS= GS>\S9r " GS? GS@\S9r " GSA GSB\S9r " GSC GSD\S9r " GSE GSF\S9r " GSG GSH\S9r " GSI GSJ\S9rGgK(L     )DummyObjectrequires_backendsc                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AdaptiveProjectedGuidance   torchc                     [        U S/5        g Nr   r   selfargskwargss      Z/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/utils/dummy_pt_objects.py__init__"AdaptiveProjectedGuidance.__init__       $	*    c                     [        U S/5        g r
   r   clsr   r   s      r   from_config%AdaptiveProjectedGuidance.from_config       #y)r   c                     [        U S/5        g r
   r   r   s      r   from_pretrained)AdaptiveProjectedGuidance.from_pretrained   r   r    N
__name__
__module____qualname____firstlineno__	_backendsr   classmethodr   r   __static_attributes__r   r   r   r   r      6    	I+ * * * *r   r   )	metaclassc                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoGuidance   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoGuidance.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoGuidance.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoGuidance.from_pretrained   r   r   r   Nr   r   r   r   r*   r*      r'   r   r*   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ClassifierFreeGuidance#   r   c                     [        U S/5        g r
   r   r   s      r   r   ClassifierFreeGuidance.__init__&   r   r   c                     [        U S/5        g r
   r   r   s      r   r   "ClassifierFreeGuidance.from_config)   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &ClassifierFreeGuidance.from_pretrained-   r   r   r   Nr   r   r   r   r3   r3   #   r'   r   r3   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ClassifierFreeZeroStarGuidance2   r   c                     [        U S/5        g r
   r   r   s      r   r   'ClassifierFreeZeroStarGuidance.__init__5   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *ClassifierFreeZeroStarGuidance.from_config8   r   r   c                     [        U S/5        g r
   r   r   s      r   r   .ClassifierFreeZeroStarGuidance.from_pretrained<   r   r   r   Nr   r   r   r   r<   r<   2   r'   r   r<   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FrequencyDecoupledGuidanceA   r   c                     [        U S/5        g r
   r   r   s      r   r   #FrequencyDecoupledGuidance.__init__D   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &FrequencyDecoupledGuidance.from_configG   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *FrequencyDecoupledGuidance.from_pretrainedK   r   r   r   Nr   r   r   r   rE   rE   A   r'   r   rE   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PerturbedAttentionGuidanceP   r   c                     [        U S/5        g r
   r   r   s      r   r   #PerturbedAttentionGuidance.__init__S   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &PerturbedAttentionGuidance.from_configV   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *PerturbedAttentionGuidance.from_pretrainedZ   r   r   r   Nr   r   r   r   rN   rN   P   r'   r   rN   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SkipLayerGuidance_   r   c                     [        U S/5        g r
   r   r   s      r   r   SkipLayerGuidance.__init__b   r   r   c                     [        U S/5        g r
   r   r   s      r   r   SkipLayerGuidance.from_confige   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !SkipLayerGuidance.from_pretrainedi   r   r   r   Nr   r   r   r   rW   rW   _   r'   r   rW   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SmoothedEnergyGuidancen   r   c                     [        U S/5        g r
   r   r   s      r   r   SmoothedEnergyGuidance.__init__q   r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SmoothedEnergyGuidance.from_configt   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &SmoothedEnergyGuidance.from_pretrainedx   r   r   r   Nr   r   r   r   r`   r`   n   r'   r   r`   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g) TangentialClassifierFreeGuidance}   r   c                     [        U S/5        g r
   r   r   s      r   r   )TangentialClassifierFreeGuidance.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,TangentialClassifierFreeGuidance.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   0TangentialClassifierFreeGuidance.from_pretrained   r   r   r   Nr   r   r   r   ri   ri   }   r'   r   ri   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FasterCacheConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   FasterCacheConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   FasterCacheConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FasterCacheConfig.from_pretrained   r   r   r   Nr   r   r   r   rr   rr      r'   r   rr   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FirstBlockCacheConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   FirstBlockCacheConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FirstBlockCacheConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   %FirstBlockCacheConfig.from_pretrained   r   r   r   Nr   r   r   r   r{   r{      r'   r   r{   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HookRegistry   r   c                     [        U S/5        g r
   r   r   s      r   r   HookRegistry.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   HookRegistry.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   HookRegistry.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LayerSkipConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   LayerSkipConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   LayerSkipConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   LayerSkipConfig.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PyramidAttentionBroadcastConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   (PyramidAttentionBroadcastConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   +PyramidAttentionBroadcastConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   /PyramidAttentionBroadcastConfig.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SmoothedEnergyGuidanceConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   %SmoothedEnergyGuidanceConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   (SmoothedEnergyGuidanceConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,SmoothedEnergyGuidanceConfig.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                  &    [        [        S/5        g r
   )r   apply_faster_cacher   r   s     r   r   r      s    (7)4r   c                  &    [        [        S/5        g r
   )r   apply_first_block_cacher   s     r   r   r      s    -y9r   c                  &    [        [        S/5        g r
   )r   apply_layer_skipr   s     r   r   r      s    &	2r   c                  &    [        [        S/5        g r
   )r   !apply_pyramid_attention_broadcastr   s     r   r   r          7'Cr   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AllegroTransformer3DModel   r   c                     [        U S/5        g r
   r   r   s      r   r   "AllegroTransformer3DModel.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AllegroTransformer3DModel.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AllegroTransformer3DModel.from_pretrained   r   r   r   Nr   r   r   r   r   r      r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AsymmetricAutoencoderKLi  r   c                     [        U S/5        g r
   r   r   s      r   r    AsymmetricAutoencoderKL.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #AsymmetricAutoencoderKL.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'AsymmetricAutoencoderKL.from_pretrained  r   r   r   Nr   r   r   r   r   r     r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AttentionBackendNamei  r   c                     [        U S/5        g r
   r   r   s      r   r   AttentionBackendName.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    AttentionBackendName.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $AttentionBackendName.from_pretrained  r   r   r   Nr   r   r   r   r   r     r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AuraFlowTransformer2DModeli#  r   c                     [        U S/5        g r
   r   r   s      r   r   #AuraFlowTransformer2DModel.__init__&  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AuraFlowTransformer2DModel.from_config)  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *AuraFlowTransformer2DModel.from_pretrained-  r   r   r   Nr   r   r   r   r   r   #  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderDCi2  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.__init__5  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.from_config8  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.from_pretrained<  r   r   r   Nr   r   r   r   r   r   2  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLiA  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.__init__D  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.from_configG  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.from_pretrainedK  r   r   r   Nr   r   r   r   r   r   A  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLAllegroiP  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLAllegro.__init__S  r   r   c                     [        U S/5        g r
   r   r   s      r   r    AutoencoderKLAllegro.from_configV  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $AutoencoderKLAllegro.from_pretrainedZ  r   r   r   Nr   r   r   r   r   r   P  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLCogVideoXi_  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLCogVideoX.__init__b  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLCogVideoX.from_confige  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLCogVideoX.from_pretrainedi  r   r   r   Nr   r   r   r   r   r   _  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLCosmosin  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLCosmos.__init__q  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLCosmos.from_configt  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #AutoencoderKLCosmos.from_pretrainedx  r   r   r   Nr   r   r   r   r   r   n  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLHunyuanVideoi}  r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLHunyuanVideo.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoencoderKLHunyuanVideo.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoencoderKLHunyuanVideo.from_pretrained  r   r   r   Nr   r   r   r   r   r   }  r'   r   r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLLTXVideoi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLLTXVideo.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !AutoencoderKLLTXVideo.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoencoderKLLTXVideo.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLMagviti  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLMagvit.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLMagvit.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #AutoencoderKLMagvit.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLMochii  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLMochi.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLMochi.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLMochi.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLQwenImagei  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLQwenImage.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLQwenImage.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLQwenImage.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLTemporalDecoderi  r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoencoderKLTemporalDecoder.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (AutoencoderKLTemporalDecoder.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,AutoencoderKLTemporalDecoder.from_pretrained  r   r   r   Nr   r   r   r   r#  r#    r'   r   r#  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLWani  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLWan.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLWan.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r    AutoencoderKLWan.from_pretrained  r   r   r   Nr   r   r   r   r+  r+    r'   r   r+  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderOoblecki  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderOobleck.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderOobleck.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderOobleck.from_pretrained  r   r   r   Nr   r   r   r   r3  r3    r'   r   r3  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderTinyi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.from_pretrained  r   r   r   Nr   r   r   r   r;  r;    r'   r   r;  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)	AutoModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.from_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.from_pretrained  r   r   r   Nr   r   r   r   rC  rC    r'   r   rC  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)
CacheMixini  r   c                     [        U S/5        g r
   r   r   s      r   r   CacheMixin.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   CacheMixin.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   CacheMixin.from_pretrained  r   r   r   Nr   r   r   r   rK  rK    r'   r   rK  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ChromaTransformer2DModeli"  r   c                     [        U S/5        g r
   r   r   s      r   r   !ChromaTransformer2DModel.__init__%  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $ChromaTransformer2DModel.from_config(  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ChromaTransformer2DModel.from_pretrained,  r   r   r   Nr   r   r   r   rS  rS  "  r'   r   rS  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CogVideoXTransformer3DModeli1  r   c                     [        U S/5        g r
   r   r   s      r   r   $CogVideoXTransformer3DModel.__init__4  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'CogVideoXTransformer3DModel.from_config7  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +CogVideoXTransformer3DModel.from_pretrained;  r   r   r   Nr   r   r   r   r[  r[  1  r'   r   r[  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CogView3PlusTransformer2DModeli@  r   c                     [        U S/5        g r
   r   r   s      r   r   'CogView3PlusTransformer2DModel.__init__C  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CogView3PlusTransformer2DModel.from_configF  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .CogView3PlusTransformer2DModel.from_pretrainedJ  r   r   r   Nr   r   r   r   rc  rc  @  r'   r   rc  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CogView4Transformer2DModeliO  r   c                     [        U S/5        g r
   r   r   s      r   r   #CogView4Transformer2DModel.__init__R  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &CogView4Transformer2DModel.from_configU  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CogView4Transformer2DModel.from_pretrainedY  r   r   r   Nr   r   r   r   rk  rk  O  r'   r   rk  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ConsisIDTransformer3DModeli^  r   c                     [        U S/5        g r
   r   r   s      r   r   #ConsisIDTransformer3DModel.__init__a  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &ConsisIDTransformer3DModel.from_configd  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *ConsisIDTransformer3DModel.from_pretrainedh  r   r   r   Nr   r   r   r   rs  rs  ^  r'   r   rs  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ConsistencyDecoderVAEim  r   c                     [        U S/5        g r
   r   r   s      r   r   ConsistencyDecoderVAE.__init__p  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ConsistencyDecoderVAE.from_configs  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ConsistencyDecoderVAE.from_pretrainedw  r   r   r   Nr   r   r   r   r{  r{  m  r'   r   r{  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ControlNetModeli|  r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   r  r  |  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ControlNetUnionModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetUnionModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    ControlNetUnionModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $ControlNetUnionModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ControlNetXSAdapteri  r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetXSAdapter.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ControlNetXSAdapter.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #ControlNetXSAdapter.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CosmosTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !CosmosTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $CosmosTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (CosmosTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DiTTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   DiTTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !DiTTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %DiTTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EasyAnimateTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   &EasyAnimateTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )EasyAnimateTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   -EasyAnimateTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FluxControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   FluxControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   FluxControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #FluxControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FluxMultiControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !FluxMultiControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $FluxMultiControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (FluxMultiControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FluxTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   FluxTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "FluxTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &FluxTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HiDreamImageTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   'HiDreamImageTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *HiDreamImageTransformer2DModel.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .HiDreamImageTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HunyuanDiT2DControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   $HunyuanDiT2DControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'HunyuanDiT2DControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +HunyuanDiT2DControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HunyuanDiT2DModeli!  r   c                     [        U S/5        g r
   r   r   s      r   r   HunyuanDiT2DModel.__init__$  r   r   c                     [        U S/5        g r
   r   r   s      r   r   HunyuanDiT2DModel.from_config'  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !HunyuanDiT2DModel.from_pretrained+  r   r   r   Nr   r   r   r   r  r  !  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g) HunyuanDiT2DMultiControlNetModeli0  r   c                     [        U S/5        g r
   r   r   s      r   r   )HunyuanDiT2DMultiControlNetModel.__init__3  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,HunyuanDiT2DMultiControlNetModel.from_config6  r   r   c                     [        U S/5        g r
   r   r   s      r   r   0HunyuanDiT2DMultiControlNetModel.from_pretrained:  r   r   r   Nr   r   r   r   r  r  0  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)'HunyuanVideoFramepackTransformer3DModeli?  r   c                     [        U S/5        g r
   r   r   s      r   r   0HunyuanVideoFramepackTransformer3DModel.__init__B  r   r   c                     [        U S/5        g r
   r   r   s      r   r   3HunyuanVideoFramepackTransformer3DModel.from_configE  r   r   c                     [        U S/5        g r
   r   r   s      r   r   7HunyuanVideoFramepackTransformer3DModel.from_pretrainedI  r   r   r   Nr   r   r   r   r  r  ?  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HunyuanVideoTransformer3DModeliN  r   c                     [        U S/5        g r
   r   r   s      r   r   'HunyuanVideoTransformer3DModel.__init__Q  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *HunyuanVideoTransformer3DModel.from_configT  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .HunyuanVideoTransformer3DModel.from_pretrainedX  r   r   r   Nr   r   r   r   r  r  N  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)I2VGenXLUNeti]  r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.__init__`  r   r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.from_configc  r   r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.from_pretrainedg  r   r   r   Nr   r   r   r   r  r  ]  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)Kandinsky3UNetil  r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.__init__o  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.from_configr  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.from_pretrainedv  r   r   r   Nr   r   r   r   r  r  l  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LatteTransformer3DModeli{  r   c                     [        U S/5        g r
   r   r   s      r   r    LatteTransformer3DModel.__init__~  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #LatteTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'LatteTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r  {  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LTXVideoTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   #LTXVideoTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &LTXVideoTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *LTXVideoTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)Lumina2Transformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   "Lumina2Transformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %Lumina2Transformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )Lumina2Transformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LuminaNextDiT2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   LuminaNextDiT2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    LuminaNextDiT2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $LuminaNextDiT2DModel.from_pretrained  r   r   r   Nr   r   r   r   r#  r#    r'   r   r#  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)MochiTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r    MochiTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #MochiTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'MochiTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r+  r+    r'   r   r+  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)
ModelMixini  r   c                     [        U S/5        g r
   r   r   s      r   r   ModelMixin.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ModelMixin.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ModelMixin.from_pretrained  r   r   r   Nr   r   r   r   r3  r3    r'   r   r3  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)MotionAdapteri  r   c                     [        U S/5        g r
   r   r   s      r   r   MotionAdapter.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   MotionAdapter.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   MotionAdapter.from_pretrained  r   r   r   Nr   r   r   r   r;  r;    r'   r   r;  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)MultiAdapteri  r   c                     [        U S/5        g r
   r   r   s      r   r   MultiAdapter.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   MultiAdapter.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   MultiAdapter.from_pretrained  r   r   r   Nr   r   r   r   rC  rC    r'   r   rC  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)MultiControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   MultiControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    MultiControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $MultiControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   rK  rK    r'   r   rK  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)OmniGenTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   "OmniGenTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %OmniGenTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )OmniGenTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   rS  rS    r'   r   rS  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PixArtTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !PixArtTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $PixArtTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (PixArtTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r[  r[    r'   r   r[  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PriorTransformeri   r   c                     [        U S/5        g r
   r   r   s      r   r   PriorTransformer.__init__#  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PriorTransformer.from_config&  r   r   c                     [        U S/5        g r
   r   r   s      r   r    PriorTransformer.from_pretrained*  r   r   r   Nr   r   r   r   rc  rc     r'   r   rc  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)QwenImageTransformer2DModeli/  r   c                     [        U S/5        g r
   r   r   s      r   r   $QwenImageTransformer2DModel.__init__2  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'QwenImageTransformer2DModel.from_config5  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +QwenImageTransformer2DModel.from_pretrained9  r   r   r   Nr   r   r   r   rk  rk  /  r'   r   rk  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SanaControlNetModeli>  r   c                     [        U S/5        g r
   r   r   s      r   r   SanaControlNetModel.__init__A  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SanaControlNetModel.from_configD  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #SanaControlNetModel.from_pretrainedH  r   r   r   Nr   r   r   r   rs  rs  >  r'   r   rs  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SanaTransformer2DModeliM  r   c                     [        U S/5        g r
   r   r   s      r   r   SanaTransformer2DModel.__init__P  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SanaTransformer2DModel.from_configS  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &SanaTransformer2DModel.from_pretrainedW  r   r   r   Nr   r   r   r   r{  r{  M  r'   r   r{  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SD3ControlNetModeli\  r   c                     [        U S/5        g r
   r   r   s      r   r   SD3ControlNetModel.__init___  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SD3ControlNetModel.from_configb  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SD3ControlNetModel.from_pretrainedf  r   r   r   Nr   r   r   r   r  r  \  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SD3MultiControlNetModelik  r   c                     [        U S/5        g r
   r   r   s      r   r    SD3MultiControlNetModel.__init__n  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #SD3MultiControlNetModel.from_configq  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'SD3MultiControlNetModel.from_pretrainedu  r   r   r   Nr   r   r   r   r  r  k  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SD3Transformer2DModeliz  r   c                     [        U S/5        g r
   r   r   s      r   r   SD3Transformer2DModel.__init__}  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !SD3Transformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %SD3Transformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r  z  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SkyReelsV2Transformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   %SkyReelsV2Transformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (SkyReelsV2Transformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,SkyReelsV2Transformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SparseControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   SparseControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !SparseControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %SparseControlNetModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)StableAudioDiTModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   StableAudioDiTModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   StableAudioDiTModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #StableAudioDiTModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)
T2IAdapteri  r   c                     [        U S/5        g r
   r   r   s      r   r   T2IAdapter.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T2IAdapter.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T2IAdapter.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)T5FilmDecoderi  r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)Transformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   Transformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Transformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "Transformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)TransformerTemporalModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !TransformerTemporalModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $TransformerTemporalModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (TransformerTemporalModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNet1DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   UNet1DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNet1DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNet1DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNet2DConditionModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   UNet2DConditionModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    UNet2DConditionModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $UNet2DConditionModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNet2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   UNet2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNet2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNet2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNet3DConditionModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   UNet3DConditionModel.__init__"  r   r   c                     [        U S/5        g r
   r   r   s      r   r    UNet3DConditionModel.from_config%  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $UNet3DConditionModel.from_pretrained)  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNetControlNetXSModeli.  r   c                     [        U S/5        g r
   r   r   s      r   r   UNetControlNetXSModel.__init__1  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !UNetControlNetXSModel.from_config4  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %UNetControlNetXSModel.from_pretrained8  r   r   r   Nr   r   r   r   r  r  .  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UNetMotionModeli=  r   c                     [        U S/5        g r
   r   r   s      r   r   UNetMotionModel.__init__@  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNetMotionModel.from_configC  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNetMotionModel.from_pretrainedG  r   r   r   Nr   r   r   r   r  r  =  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g) UNetSpatioTemporalConditionModeliL  r   c                     [        U S/5        g r
   r   r   s      r   r   )UNetSpatioTemporalConditionModel.__init__O  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,UNetSpatioTemporalConditionModel.from_configR  r   r   c                     [        U S/5        g r
   r   r   s      r   r   0UNetSpatioTemporalConditionModel.from_pretrainedV  r   r   r   Nr   r   r   r   r  r  L  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UVit2DModeli[  r   c                     [        U S/5        g r
   r   r   s      r   r   UVit2DModel.__init__^  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UVit2DModel.from_configa  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UVit2DModel.from_pretrainede  r   r   r   Nr   r   r   r   r  r  [  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)VQModelij  r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.__init__m  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.from_configp  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.from_pretrainedt  r   r   r   Nr   r   r   r   r  r  j  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)WanTransformer3DModeliy  r   c                     [        U S/5        g r
   r   r   s      r   r   WanTransformer3DModel.__init__|  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !WanTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %WanTransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r  y  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)WanVACETransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   "WanVACETransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %WanVACETransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )WanVACETransformer3DModel.from_pretrained  r   r   r   Nr   r   r   r   r#  r#    r'   r   r#  c                  &    [        [        S/5        g r
   )r   attention_backendr   s     r   r+  r+    s    ''3r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ComponentsManageri  r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentsManager.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentsManager.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ComponentsManager.from_pretrained  r   r   r   Nr   r   r   r   r-  r-    r'   r   r-  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ComponentSpeci  r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentSpec.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentSpec.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentSpec.from_pretrained  r   r   r   Nr   r   r   r   r5  r5    r'   r   r5  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ModularPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   ModularPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ModularPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ModularPipeline.from_pretrained  r   r   r   Nr   r   r   r   r=  r=    r'   r   r=  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ModularPipelineBlocksi  r   c                     [        U S/5        g r
   r   r   s      r   r   ModularPipelineBlocks.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ModularPipelineBlocks.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ModularPipelineBlocks.from_pretrained  r   r   r   Nr   r   r   r   rE  rE    r'   r   rE  c                  &    [        [        S/5        g r
   )r   get_constant_scheduler   s     r   rM  rM    s    +gY7r   c                  &    [        [        S/5        g r
   )r   !get_constant_schedule_with_warmupr   s     r   rO  rO    r   r   c                  &    [        [        S/5        g r
   )r   get_cosine_schedule_with_warmupr   s     r   rQ  rQ        5yAr   c                  &    [        [        S/5        g r
   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr   s     r   rT  rT    s    H7)Tr   c                  &    [        [        S/5        g r
   )r   get_linear_schedule_with_warmupr   s     r   rV  rV    rR  r   c                  &    [        [        S/5        g r
   )r   )get_polynomial_decay_schedule_with_warmupr   s     r   rX  rX    s    ?'Kr   c                  &    [        [        S/5        g r
   )r   get_schedulerr   s     r   rZ  rZ    s    mgY/r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AudioPipelineOutputi  r   c                     [        U S/5        g r
   r   r   s      r   r   AudioPipelineOutput.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AudioPipelineOutput.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #AudioPipelineOutput.from_pretrained  r   r   r   Nr   r   r   r   r\  r\    r'   r   r\  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineForImage2Imagei  r   c                     [        U S/5        g r
   r   r   s      r   r   #AutoPipelineForImage2Image.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoPipelineForImage2Image.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *AutoPipelineForImage2Image.from_pretrained  r   r   r   Nr   r   r   r   rd  rd    r'   r   rd  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineForInpaintingi  r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoPipelineForInpainting.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoPipelineForInpainting.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoPipelineForInpainting.from_pretrained  r   r   r   Nr   r   r   r   rl  rl    r'   r   rl  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineForText2Imagei   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoPipelineForText2Image.__init__#  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoPipelineForText2Image.from_config&  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoPipelineForText2Image.from_pretrained*  r   r   r   Nr   r   r   r   rt  rt     r'   r   rt  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BlipDiffusionControlNetPipelinei/  r   c                     [        U S/5        g r
   r   r   s      r   r   (BlipDiffusionControlNetPipeline.__init__2  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +BlipDiffusionControlNetPipeline.from_config5  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /BlipDiffusionControlNetPipeline.from_pretrained9  r   r   r   Nr   r   r   r   r|  r|  /  r'   r   r|  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BlipDiffusionPipelinei>  r   c                     [        U S/5        g r
   r   r   s      r   r   BlipDiffusionPipeline.__init__A  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !BlipDiffusionPipeline.from_configD  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %BlipDiffusionPipeline.from_pretrainedH  r   r   r   Nr   r   r   r   r  r  >  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CLIPImageProjectioniM  r   c                     [        U S/5        g r
   r   r   s      r   r   CLIPImageProjection.__init__P  r   r   c                     [        U S/5        g r
   r   r   s      r   r   CLIPImageProjection.from_configS  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #CLIPImageProjection.from_pretrainedW  r   r   r   Nr   r   r   r   r  r  M  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ConsistencyModelPipelinei\  r   c                     [        U S/5        g r
   r   r   s      r   r   !ConsistencyModelPipeline.__init___  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $ConsistencyModelPipeline.from_configb  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ConsistencyModelPipeline.from_pretrainedf  r   r   r   Nr   r   r   r   r  r  \  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DanceDiffusionPipelineik  r   c                     [        U S/5        g r
   r   r   s      r   r   DanceDiffusionPipeline.__init__n  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "DanceDiffusionPipeline.from_configq  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &DanceDiffusionPipeline.from_pretrainedu  r   r   r   Nr   r   r   r   r  r  k  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDIMPipelineiz  r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMPipeline.__init__}  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r  z  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDPMPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DiffusionPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusionPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusionPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !DiffusionPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DiTPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   DiTPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DiTPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DiTPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ImagePipelineOutputi  r   c                     [        U S/5        g r
   r   r   s      r   r   ImagePipelineOutput.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ImagePipelineOutput.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #ImagePipelineOutput.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)KarrasVePipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   KarrasVePipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   KarrasVePipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r    KarrasVePipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LDMPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   LDMPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   LDMPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   LDMPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LDMSuperResolutionPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   #LDMSuperResolutionPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &LDMSuperResolutionPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *LDMSuperResolutionPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PNDMPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)RePaintPipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintPipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintPipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintPipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ScoreSdeVePipelinei  r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVePipeline.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVePipeline.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "ScoreSdeVePipeline.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)StableDiffusionMixini  r   c                     [        U S/5        g r
   r   r   s      r   r   StableDiffusionMixin.__init__"  r   r   c                     [        U S/5        g r
   r   r   s      r   r    StableDiffusionMixin.from_config%  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $StableDiffusionMixin.from_pretrained)  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DiffusersQuantizeri.  r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusersQuantizer.__init__1  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusersQuantizer.from_config4  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "DiffusersQuantizer.from_pretrained8  r   r   r   Nr   r   r   r   r  r  .  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AmusedScheduleri=  r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.__init__@  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.from_configC  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.from_pretrainedG  r   r   r   Nr   r   r   r   r  r  =  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CMStochasticIterativeScheduleriL  r   c                     [        U S/5        g r
   r   r   s      r   r   'CMStochasticIterativeScheduler.__init__O  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CMStochasticIterativeScheduler.from_configR  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .CMStochasticIterativeScheduler.from_pretrainedV  r   r   r   Nr   r   r   r   r  r  L  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CogVideoXDDIMScheduleri[  r   c                     [        U S/5        g r
   r   r   s      r   r   CogVideoXDDIMScheduler.__init__^  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "CogVideoXDDIMScheduler.from_configa  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &CogVideoXDDIMScheduler.from_pretrainede  r   r   r   Nr   r   r   r   r  r  [  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CogVideoXDPMSchedulerij  r   c                     [        U S/5        g r
   r   r   s      r   r   CogVideoXDPMScheduler.__init__m  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !CogVideoXDPMScheduler.from_configp  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %CogVideoXDPMScheduler.from_pretrainedt  r   r   r   Nr   r   r   r   r$  r$  j  r'   r   r$  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDIMInverseScheduleriy  r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMInverseScheduler.__init__|  r   r   c                     [        U S/5        g r
   r   r   s      r   r    DDIMInverseScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $DDIMInverseScheduler.from_pretrained  r   r   r   Nr   r   r   r   r,  r,  y  r'   r   r,  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDIMParallelScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMParallelScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !DDIMParallelScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %DDIMParallelScheduler.from_pretrained  r   r   r   Nr   r   r   r   r4  r4    r'   r   r4  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDIMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDIMScheduler.from_pretrained  r   r   r   Nr   r   r   r   r<  r<    r'   r   r<  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDPMParallelScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMParallelScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !DDPMParallelScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %DDPMParallelScheduler.from_pretrained  r   r   r   Nr   r   r   r   rD  rD    r'   r   rD  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDPMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DDPMScheduler.from_pretrained  r   r   r   Nr   r   r   r   rL  rL    r'   r   rL  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DDPMWuerstchenScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r    DDPMWuerstchenScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #DDPMWuerstchenScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'DDPMWuerstchenScheduler.from_pretrained  r   r   r   Nr   r   r   r   rT  rT    r'   r   rT  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DEISMultistepScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   DEISMultistepScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "DEISMultistepScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &DEISMultistepScheduler.from_pretrained  r   r   r   Nr   r   r   r   r\  r\    r'   r   r\  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)"DPMSolverMultistepInverseScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   +DPMSolverMultistepInverseScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .DPMSolverMultistepInverseScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   2DPMSolverMultistepInverseScheduler.from_pretrained  r   r   r   Nr   r   r   r   rd  rd    r'   r   rd  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DPMSolverMultistepScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   $DPMSolverMultistepScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'DPMSolverMultistepScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +DPMSolverMultistepScheduler.from_pretrained  r   r   r   Nr   r   r   r   rl  rl    r'   r   rl  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DPMSolverSinglestepScheduleri   r   c                     [        U S/5        g r
   r   r   s      r   r   %DPMSolverSinglestepScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (DPMSolverSinglestepScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,DPMSolverSinglestepScheduler.from_pretrained
  r   r   r   Nr   r   r   r   rt  rt     r'   r   rt  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EDMDPMSolverMultistepScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   'EDMDPMSolverMultistepScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *EDMDPMSolverMultistepScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .EDMDPMSolverMultistepScheduler.from_pretrained  r   r   r   Nr   r   r   r   r|  r|    r'   r   r|  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EDMEulerScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   EDMEulerScheduler.__init__!  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EDMEulerScheduler.from_config$  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !EDMEulerScheduler.from_pretrained(  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EulerAncestralDiscreteScheduleri-  r   c                     [        U S/5        g r
   r   r   s      r   r   (EulerAncestralDiscreteScheduler.__init__0  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +EulerAncestralDiscreteScheduler.from_config3  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /EulerAncestralDiscreteScheduler.from_pretrained7  r   r   r   Nr   r   r   r   r  r  -  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EulerDiscreteScheduleri<  r   c                     [        U S/5        g r
   r   r   s      r   r   EulerDiscreteScheduler.__init__?  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "EulerDiscreteScheduler.from_configB  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &EulerDiscreteScheduler.from_pretrainedF  r   r   r   Nr   r   r   r   r  r  <  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlowMatchEulerDiscreteScheduleriK  r   c                     [        U S/5        g r
   r   r   s      r   r   (FlowMatchEulerDiscreteScheduler.__init__N  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +FlowMatchEulerDiscreteScheduler.from_configQ  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /FlowMatchEulerDiscreteScheduler.from_pretrainedU  r   r   r   Nr   r   r   r   r  r  K  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlowMatchHeunDiscreteScheduleriZ  r   c                     [        U S/5        g r
   r   r   s      r   r   'FlowMatchHeunDiscreteScheduler.__init__]  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *FlowMatchHeunDiscreteScheduler.from_config`  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .FlowMatchHeunDiscreteScheduler.from_pretrainedd  r   r   r   Nr   r   r   r   r  r  Z  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)FlowMatchLCMSchedulerii  r   c                     [        U S/5        g r
   r   r   s      r   r   FlowMatchLCMScheduler.__init__l  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlowMatchLCMScheduler.from_configo  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %FlowMatchLCMScheduler.from_pretraineds  r   r   r   Nr   r   r   r   r  r  i  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HeunDiscreteSchedulerix  r   c                     [        U S/5        g r
   r   r   s      r   r   HeunDiscreteScheduler.__init__{  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !HeunDiscreteScheduler.from_config~  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %HeunDiscreteScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r  x  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)IPNDMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   IPNDMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   IPNDMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   IPNDMScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)KarrasVeScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   KarrasVeScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   KarrasVeScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !KarrasVeScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)KDPM2AncestralDiscreteScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   (KDPM2AncestralDiscreteScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +KDPM2AncestralDiscreteScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /KDPM2AncestralDiscreteScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)KDPM2DiscreteScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   KDPM2DiscreteScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "KDPM2DiscreteScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &KDPM2DiscreteScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LCMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   LCMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   LCMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   LCMScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PNDMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)RePaintScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r    RePaintScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SASolverScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   SASolverScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SASolverScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !SASolverScheduler.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SchedulerMixini  r   c                     [        U S/5        g r
   r   r   s      r   r   SchedulerMixin.__init__	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SchedulerMixin.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SchedulerMixin.from_pretrained		  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SCMScheduleri	  r   c                     [        U S/5        g r
   r   r   s      r   r   SCMScheduler.__init__	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SCMScheduler.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SCMScheduler.from_pretrained	  r   r   r   Nr   r   r   r   r  r  	  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ScoreSdeVeScheduleri	  r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVeScheduler.__init__ 	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVeScheduler.from_config#	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #ScoreSdeVeScheduler.from_pretrained'	  r   r   r   Nr   r   r   r   r  r  	  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)TCDScheduleri,	  r   c                     [        U S/5        g r
   r   r   s      r   r   TCDScheduler.__init__/	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   TCDScheduler.from_config2	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   TCDScheduler.from_pretrained6	  r   r   r   Nr   r   r   r   r  r  ,	  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UnCLIPScheduleri;	  r   c                     [        U S/5        g r
   r   r   s      r   r   UnCLIPScheduler.__init__>	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UnCLIPScheduler.from_configA	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UnCLIPScheduler.from_pretrainedE	  r   r   r   Nr   r   r   r   r  r  ;	  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)UniPCMultistepScheduleriJ	  r   c                     [        U S/5        g r
   r   r   s      r   r    UniPCMultistepScheduler.__init__M	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #UniPCMultistepScheduler.from_configP	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'UniPCMultistepScheduler.from_pretrainedT	  r   r   r   Nr   r   r   r   r$  r$  J	  r'   r   r$  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)VQDiffusionScheduleriY	  r   c                     [        U S/5        g r
   r   r   s      r   r   VQDiffusionScheduler.__init__\	  r   r   c                     [        U S/5        g r
   r   r   s      r   r    VQDiffusionScheduler.from_config_	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $VQDiffusionScheduler.from_pretrainedc	  r   r   r   Nr   r   r   r   r,  r,  Y	  r'   r   r,  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EMAModelih	  r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.__init__k	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.from_confign	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.from_pretrainedr	  r   r   r   Nr   r   r   r   r4  r4  h	  r'   r   r4  N)utilsr   r   r   r*   r3   r<   rE   rN   rW   r`   ri   rr   r{   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  r;  rC  rK  rS  r[  rc  rk  rs  r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r#  r+  r3  r;  rC  rK  rS  r[  rc  rk  rs  r{  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r#  r+  r-  r5  r=  rE  rM  rO  rQ  rT  rV  rX  rZ  r\  rd  rl  rt  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r$  r,  r4  r<  rD  rL  rT  r\  rd  rl  rt  r|  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r$  r,  r4  r   r   r   <module>r<     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   