
    
3jKF                    P   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! 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/ rS0 rS1 r " S2 S3\S9r " S4 S5\S9r " S6 S7\S9r  " S8 S9\S9r! " S: S;\S9r" " S< S=\S9r# " S> S?\S9r$ " S@ SA\S9r% " SB SC\S9r& " SD SE\S9r' " SF SG\S9r( " SH SI\S9r) " SJ SK\S9r* " SL SM\S9r+ " SN SO\S9r, " SP SQ\S9r- " SR SS\S9r. " ST SU\S9r/ " SV SW\S9r0 " SX SY\S9r1 " SZ S[\S9r2 " S\ S]\S9r3 " S^ S_\S9r4 " S` Sa\S9r5 " Sb Sc\S9r6 " Sd Se\S9r7 " Sf Sg\S9r8 " Sh Si\S9r9 " Sj Sk\S9r: " Sl Sm\S9r; " Sn So\S9r< " Sp Sq\S9r= " Sr Ss\S9r> " St Su\S9r? " Sv Sw\S9r@ " Sx Sy\S9rA " Sz 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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r " S S\S9r " S S\S9r " S 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rGS$ 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rGS; rGS< rGS= rGS> rGS? rGS@ rGSA r " GSB GSC\S9r " GSD GSE\S9r " GSF GSG\S9r " GSH GSI\S9r " GSJ GSK\S9r " GSL GSM\S9r " GSN GSO\S9r " GSP GSQ\S9r " GSR GSS\S9r " GST GSU\S9r " GSV GSW\S9r " GSX GSY\S9r " GSZ GS[\S9r " GS\ GS]\S9r " GS^ GS_\S9r " GS` GSa\S9r " GSb GSc\S9r " GSd GSe\S9r " GSf GSg\S9r " GSh GSi\S9r " GSj GSk\S9r " GSl GSm\S9r " GSn GSo\S9r " GSp GSq\S9r " GSr GSs\S9r " GSt GSu\S9r " GSv GSw\S9r " GSx GSy\S9r " GSz 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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rGg(     )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/wildlama/miniconda3/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)AdaptiveProjectedMixGuidance   r   c                     [        U S/5        g r
   r   r   s      r   r   %AdaptiveProjectedMixGuidance.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   (AdaptiveProjectedMixGuidance.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,AdaptiveProjectedMixGuidance.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)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   r3   r3   #   r'   r   r3   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BaseGuidance2   r   c                     [        U S/5        g r
   r   r   s      r   r   BaseGuidance.__init__5   r   r   c                     [        U S/5        g r
   r   r   s      r   r   BaseGuidance.from_config8   r   r   c                     [        U S/5        g r
   r   r   s      r   r   BaseGuidance.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)ClassifierFreeGuidanceA   r   c                     [        U S/5        g r
   r   r   s      r   r   ClassifierFreeGuidance.__init__D   r   r   c                     [        U S/5        g r
   r   r   s      r   r   "ClassifierFreeGuidance.from_configG   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &ClassifierFreeGuidance.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)ClassifierFreeZeroStarGuidanceP   r   c                     [        U S/5        g r
   r   r   s      r   r   'ClassifierFreeZeroStarGuidance.__init__S   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *ClassifierFreeZeroStarGuidance.from_configV   r   r   c                     [        U S/5        g r
   r   r   s      r   r   .ClassifierFreeZeroStarGuidance.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)FrequencyDecoupledGuidance_   r   c                     [        U S/5        g r
   r   r   s      r   r   #FrequencyDecoupledGuidance.__init__b   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &FrequencyDecoupledGuidance.from_confige   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *FrequencyDecoupledGuidance.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)PerturbedAttentionGuidancen   r   c                     [        U S/5        g r
   r   r   s      r   r   #PerturbedAttentionGuidance.__init__q   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &PerturbedAttentionGuidance.from_configt   r   r   c                     [        U S/5        g r
   r   r   s      r   r   *PerturbedAttentionGuidance.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)SkipLayerGuidance}   r   c                     [        U S/5        g r
   r   r   s      r   r   SkipLayerGuidance.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   SkipLayerGuidance.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   !SkipLayerGuidance.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)SmoothedEnergyGuidance   r   c                     [        U S/5        g r
   r   r   s      r   r   SmoothedEnergyGuidance.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SmoothedEnergyGuidance.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   &SmoothedEnergyGuidance.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) 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   r{   r{      r'   r   r{   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   r   r      r'   r   r   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)MagCacheConfig   r   c                     [        U S/5        g r
   r   r   s      r   r   MagCacheConfig.__init__   r   r   c                     [        U S/5        g r
   r   r   s      r   r   MagCacheConfig.from_config   r   r   c                     [        U S/5        g r
   r   r   s      r   r   MagCacheConfig.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)SmoothedEnergyGuidanceConfigi  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                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)TaylorSeerCacheConfigi  r   c                     [        U S/5        g r
   r   r   s      r   r   TaylorSeerCacheConfig.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !TaylorSeerCacheConfig.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %TaylorSeerCacheConfig.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)TextKVCacheConfigi"  r   c                     [        U S/5        g r
   r   r   s      r   r   TextKVCacheConfig.__init__%  r   r   c                     [        U S/5        g r
   r   r   s      r   r   TextKVCacheConfig.from_config(  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !TextKVCacheConfig.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   1  s    (7)4r   c                  &    [        [        S/5        g r
   )r   apply_first_block_cacher   s     r   r   r   5  s    -y9r   c                  &    [        [        S/5        g r
   )r   apply_layer_skipr   s     r   r   r   9  s    &	2r   c                  &    [        [        S/5        g r
   )r   apply_mag_cacher   s     r   r   r   =  s    oy1r   c                  &    [        [        S/5        g r
   )r   !apply_pyramid_attention_broadcastr   s     r   r   r   A      7'Cr   c                  &    [        [        S/5        g r
   )r   apply_taylorseer_cacher   s     r   r   r   E  s    ,wi8r   c                  &    [        [        S/5        g r
   )r   apply_text_kv_cacher   s     r   r   r   I  s    )G95r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)InpaintProcessoriM  r   c                     [        U S/5        g r
   r   r   s      r   r   InpaintProcessor.__init__P  r   r   c                     [        U S/5        g r
   r   r   s      r   r   InpaintProcessor.from_configS  r   r   c                     [        U S/5        g r
   r   r   s      r   r    InpaintProcessor.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)IPAdapterMaskProcessori\  r   c                     [        U S/5        g r
   r   r   s      r   r   IPAdapterMaskProcessor.__init___  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "IPAdapterMaskProcessor.from_configb  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &IPAdapterMaskProcessor.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)PixArtImageProcessorik  r   c                     [        U S/5        g r
   r   r   s      r   r   PixArtImageProcessor.__init__n  r   r   c                     [        U S/5        g r
   r   r   s      r   r    PixArtImageProcessor.from_configq  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $PixArtImageProcessor.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)VaeImageProcessoriz  r   c                     [        U S/5        g r
   r   r   s      r   r   VaeImageProcessor.__init__}  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VaeImageProcessor.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !VaeImageProcessor.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)VaeImageProcessorLDM3Di  r   c                     [        U S/5        g r
   r   r   s      r   r   VaeImageProcessorLDM3D.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "VaeImageProcessorLDM3D.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &VaeImageProcessorLDM3D.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)AceStepTransformer1DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   "AceStepTransformer1DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AceStepTransformer1DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AceStepTransformer1DModel.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)AllegroTransformer3DModeli  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)AutoencoderDCi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderDC.from_pretrained  r   r   r   Nr   r   r   r   r2  r2    r'   r   r2  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKL.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)AutoencoderKLAllegroi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLAllegro.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r    AutoencoderKLAllegro.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $AutoencoderKLAllegro.from_pretrained  r   r   r   Nr   r   r   r   rB  rB    r'   r   rB  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__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLCogVideoX.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLCogVideoX.from_pretrained  r   r   r   Nr   r   r   r   rJ  rJ    r'   r   rJ  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLCosmosi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLCosmos.__init__"  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLCosmos.from_config%  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #AutoencoderKLCosmos.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)AutoencoderKLFlux2i.  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLFlux2.__init__1  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLFlux2.from_config4  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLFlux2.from_pretrained8  r   r   r   Nr   r   r   r   rZ  rZ  .  r'   r   rZ  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLHunyuanImagei=  r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLHunyuanImage.__init__@  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoencoderKLHunyuanImage.from_configC  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoencoderKLHunyuanImage.from_pretrainedG  r   r   r   Nr   r   r   r   rb  rb  =  r'   r   rb  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g) AutoencoderKLHunyuanImageRefineriL  r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoencoderKLHunyuanImageRefiner.__init__O  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,AutoencoderKLHunyuanImageRefiner.from_configR  r   r   c                     [        U S/5        g r
   r   r   s      r   r   0AutoencoderKLHunyuanImageRefiner.from_pretrainedV  r   r   r   Nr   r   r   r   rj  rj  L  r'   r   rj  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_configa  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoencoderKLHunyuanVideo.from_pretrainede  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)AutoencoderKLHunyuanVideo15ij  r   c                     [        U S/5        g r
   r   r   s      r   r   $AutoencoderKLHunyuanVideo15.__init__m  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'AutoencoderKLHunyuanVideo15.from_configp  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +AutoencoderKLHunyuanVideo15.from_pretrainedt  r   r   r   Nr   r   r   r   rz  rz  j  r'   r   rz  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderKLKVAEiy  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLKVAE.__init__|  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLKVAE.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !AutoencoderKLKVAE.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)AutoencoderKLKVAEVideoi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLKVAEVideo.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLKVAEVideo.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLKVAEVideo.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)AutoencoderKLLTX2Audioi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLLTX2Audio.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLLTX2Audio.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLLTX2Audio.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)AutoencoderKLLTX2Videoi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderKLLTX2Video.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoencoderKLLTX2Video.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &AutoencoderKLLTX2Video.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   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoencoderRAEi  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderRAE.__init__!  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderRAE.from_config$  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderRAE.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)AutoencoderTinyi-  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.__init__0  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.from_config3  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderTiny.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)AutoencoderVidToki<  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderVidTok.__init__?  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoencoderVidTok.from_configB  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !AutoencoderVidTok.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)	AutoModeliK  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.__init__N  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.from_configQ  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoModel.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)BriaFiboTransformer2DModeliZ  r   c                     [        U S/5        g r
   r   r   s      r   r   #BriaFiboTransformer2DModel.__init__]  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &BriaFiboTransformer2DModel.from_config`  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *BriaFiboTransformer2DModel.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)BriaTransformer2DModelii  r   c                     [        U S/5        g r
   r   r   s      r   r   BriaTransformer2DModel.__init__l  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "BriaTransformer2DModel.from_configo  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &BriaTransformer2DModel.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)
CacheMixinix  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   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)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   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ChronoEditTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   %ChronoEditTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ChronoEditTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,ChronoEditTransformer3DModel.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)CogVideoXTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   $CogVideoXTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'CogVideoXTransformer3DModel.from_config  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"    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__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CogView3PlusTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .CogView3PlusTransformer2DModel.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)CogView4Transformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   #CogView4Transformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &CogView4Transformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CogView4Transformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r2  r2    r'   r   r2  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__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &ConsisIDTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *ConsisIDTransformer3DModel.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)ConsistencyDecoderVAEi  r   c                     [        U S/5        g r
   r   r   s      r   r   ConsistencyDecoderVAE.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ConsistencyDecoderVAE.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ConsistencyDecoderVAE.from_pretrained  r   r   r   Nr   r   r   r   rB  rB    r'   r   rB  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ContextParallelConfigi  r   c                     [        U S/5        g r
   r   r   s      r   r   ContextParallelConfig.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ContextParallelConfig.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ContextParallelConfig.from_pretrained  r   r   r   Nr   r   r   r   rJ  rJ    r'   r   rJ  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   rR  rR    r'   r   rR  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   rZ  rZ    r'   r   rZ  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   rb  rb    r'   r   rb  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CosmosControlNetModeli,  r   c                     [        U S/5        g r
   r   r   s      r   r   CosmosControlNetModel.__init__/  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !CosmosControlNetModel.from_config2  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %CosmosControlNetModel.from_pretrained6  r   r   r   Nr   r   r   r   rj  rj  ,  r'   r   rj  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_configA  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (CosmosTransformer3DModel.from_pretrainedE  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)DiTTransformer2DModeliJ  r   c                     [        U S/5        g r
   r   r   s      r   r   DiTTransformer2DModel.__init__M  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !DiTTransformer2DModel.from_configP  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %DiTTransformer2DModel.from_pretrainedT  r   r   r   Nr   r   r   r   rz  rz  J  r'   r   rz  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EasyAnimateTransformer3DModeliY  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_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)ErnieImageTransformer2DModelih  r   c                     [        U S/5        g r
   r   r   s      r   r   %ErnieImageTransformer2DModel.__init__k  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ErnieImageTransformer2DModel.from_confign  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,ErnieImageTransformer2DModel.from_pretrainedr  r   r   r   Nr   r   r   r   r  r  h  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)Flux2Transformer2DModeliw  r   c                     [        U S/5        g r
   r   r   s      r   r    Flux2Transformer2DModel.__init__z  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #Flux2Transformer2DModel.from_config}  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'Flux2Transformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r  r  w  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)GlmImageTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   #GlmImageTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &GlmImageTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *GlmImageTransformer2DModel.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)HeliosTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !HeliosTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $HeliosTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (HeliosTransformer3DModel.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) HunyuanDiT2DMultiControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   )HunyuanDiT2DMultiControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,HunyuanDiT2DMultiControlNetModel.from_config  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    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)HunyuanImageTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   'HunyuanImageTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *HunyuanImageTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .HunyuanImageTransformer2DModel.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) HunyuanVideo15Transformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   )HunyuanVideo15Transformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,HunyuanVideo15Transformer3DModel.from_config"  r   r   c                     [        U S/5        g r
   r   r   s      r   r   0HunyuanVideo15Transformer3DModel.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)'HunyuanVideoFramepackTransformer3DModeli+  r   c                     [        U S/5        g r
   r   r   s      r   r   0HunyuanVideoFramepackTransformer3DModel.__init__.  r   r   c                     [        U S/5        g r
   r   r   s      r   r   3HunyuanVideoFramepackTransformer3DModel.from_config1  r   r   c                     [        U S/5        g r
   r   r   s      r   r   7HunyuanVideoFramepackTransformer3DModel.from_pretrained5  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)HunyuanVideoTransformer3DModeli:  r   c                     [        U S/5        g r
   r   r   s      r   r   'HunyuanVideoTransformer3DModel.__init__=  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *HunyuanVideoTransformer3DModel.from_config@  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .HunyuanVideoTransformer3DModel.from_pretrainedD  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)I2VGenXLUNetiI  r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.__init__L  r   r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.from_configO  r   r   c                     [        U S/5        g r
   r   r   s      r   r   I2VGenXLUNet.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)Kandinsky3UNetiX  r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.__init__[  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.from_config^  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Kandinsky3UNet.from_pretrainedb  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)Kandinsky5Transformer3DModelig  r   c                     [        U S/5        g r
   r   r   s      r   r   %Kandinsky5Transformer3DModel.__init__j  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (Kandinsky5Transformer3DModel.from_configm  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,Kandinsky5Transformer3DModel.from_pretrainedq  r   r   r   Nr   r   r   r   r  r  g  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LatteTransformer3DModeliv  r   c                     [        U S/5        g r
   r   r   s      r   r    LatteTransformer3DModel.__init__y  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  v  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LongCatAudioDiTTransformeri  r   c                     [        U S/5        g r
   r   r   s      r   r   #LongCatAudioDiTTransformer.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &LongCatAudioDiTTransformer.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *LongCatAudioDiTTransformer.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)LongCatAudioDiTVaei  r   c                     [        U S/5        g r
   r   r   s      r   r   LongCatAudioDiTVae.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   LongCatAudioDiTVae.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "LongCatAudioDiTVae.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)LongCatImageTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   'LongCatImageTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *LongCatImageTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .LongCatImageTransformer2DModel.from_pretrained  r   r   r   Nr   r   r   r   r2  r2    r'   r   r2  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LTX2VideoTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   $LTX2VideoTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'LTX2VideoTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +LTX2VideoTransformer3DModel.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   rB  rB    r'   r   rB  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   rJ  rJ    r'   r   rJ  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   rR  rR    r'   r   rR  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   rZ  rZ    r'   r   rZ  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   rb  rb    r'   r   rb  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   rj  rj    r'   r   rj  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   rr  rr    r'   r   rr  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_config0  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $MultiControlNetModel.from_pretrained4  r   r   r   Nr   r   r   r   rz  rz  *  r'   r   rz  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)!NucleusMoEImageTransformer2DModeli9  r   c                     [        U S/5        g r
   r   r   s      r   r   *NucleusMoEImageTransformer2DModel.__init__<  r   r   c                     [        U S/5        g r
   r   r   s      r   r   -NucleusMoEImageTransformer2DModel.from_config?  r   r   c                     [        U S/5        g r
   r   r   s      r   r   1NucleusMoEImageTransformer2DModel.from_pretrainedC  r   r   r   Nr   r   r   r   r  r  9  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)OmniGenTransformer2DModeliH  r   c                     [        U S/5        g r
   r   r   s      r   r   "OmniGenTransformer2DModel.__init__K  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %OmniGenTransformer2DModel.from_configN  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )OmniGenTransformer2DModel.from_pretrainedR  r   r   r   Nr   r   r   r   r  r  H  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)OvisImageTransformer2DModeliW  r   c                     [        U S/5        g r
   r   r   s      r   r   $OvisImageTransformer2DModel.__init__Z  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'OvisImageTransformer2DModel.from_config]  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +OvisImageTransformer2DModel.from_pretraineda  r   r   r   Nr   r   r   r   r  r  W  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ParallelConfigif  r   c                     [        U S/5        g r
   r   r   s      r   r   ParallelConfig.__init__i  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ParallelConfig.from_configl  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ParallelConfig.from_pretrainedp  r   r   r   Nr   r   r   r   r  r  f  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PixArtTransformer2DModeliu  r   c                     [        U S/5        g r
   r   r   s      r   r   !PixArtTransformer2DModel.__init__x  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  u  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   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PRXTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   PRXTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !PRXTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %PRXTransformer2DModel.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)QwenImageControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   !QwenImageControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $QwenImageControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (QwenImageControlNetModel.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)QwenImageMultiControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   &QwenImageMultiControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )QwenImageMultiControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   -QwenImageMultiControlNetModel.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)QwenImageTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   $QwenImageTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'QwenImageTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +QwenImageTransformer2DModel.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)SanaControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   SanaControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SanaControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #SanaControlNetModel.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)SanaTransformer2DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   SanaTransformer2DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SanaTransformer2DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &SanaTransformer2DModel.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)SanaVideoTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   $SanaVideoTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'SanaVideoTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +SanaVideoTransformer3DModel.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)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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "SD3ControlNetModel.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)SD3MultiControlNetModeli  r   c                     [        U S/5        g r
   r   r   s      r   r    SD3MultiControlNetModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #SD3MultiControlNetModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'SD3MultiControlNetModel.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)SD3Transformer2DModeli  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    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_pretrained3  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)SparseControlNetModeli8  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_pretrainedB  r   r   r   Nr   r   r   r   r
  r
  8  r'   r   r
  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)StableAudioDiTModeliG  r   c                     [        U S/5        g r
   r   r   s      r   r   StableAudioDiTModel.__init__J  r   r   c                     [        U S/5        g r
   r   r   s      r   r   StableAudioDiTModel.from_configM  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #StableAudioDiTModel.from_pretrainedQ  r   r   r   Nr   r   r   r   r  r  G  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)
T2IAdapteriV  r   c                     [        U S/5        g r
   r   r   s      r   r   T2IAdapter.__init__Y  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  V  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)T5FilmDecoderie  r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.__init__h  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.from_configk  r   r   c                     [        U S/5        g r
   r   r   s      r   r   T5FilmDecoder.from_pretrainedo  r   r   r   Nr   r   r   r   r"  r"  e  r'   r   r"  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)Transformer2DModelit  r   c                     [        U S/5        g r
   r   r   s      r   r   Transformer2DModel.__init__w  r   r   c                     [        U S/5        g r
   r   r   s      r   r   Transformer2DModel.from_configz  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*  t  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   r2  r2    r'   r   r2  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   rB  rB    r'   r   rB  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   rJ  rJ    r'   r   rJ  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   rR  rR    r'   r   rR  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__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !UNetControlNetXSModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %UNetControlNetXSModel.from_pretrained  r   r   r   Nr   r   r   r   rZ  rZ    r'   r   rZ  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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UNetMotionModel.from_pretrained  r   r   r   Nr   r   r   r   rb  rb    r'   r   rb  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g) UNetSpatioTemporalConditionModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   )UNetSpatioTemporalConditionModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,UNetSpatioTemporalConditionModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   0UNetSpatioTemporalConditionModel.from_pretrained  r   r   r   Nr   r   r   r   rj  rj    r'   r   rj  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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UVit2DModel.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)VQModeli
  r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VQModel.from_pretrained  r   r   r   Nr   r   r   r   rz  rz  
  r'   r   rz  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)WanAnimateTransformer3DModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   %WanAnimateTransformer3DModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (WanAnimateTransformer3DModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,WanAnimateTransformer3DModel.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)WanTransformer3DModeli(  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_pretrained2  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)WanVACETransformer3DModeli7  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_pretrainedA  r   r   r   Nr   r   r   r   r  r  7  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ZImageControlNetModeliF  r   c                     [        U S/5        g r
   r   r   s      r   r   ZImageControlNetModel.__init__I  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !ZImageControlNetModel.from_configL  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ZImageControlNetModel.from_pretrainedP  r   r   r   Nr   r   r   r   r  r  F  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ZImageTransformer2DModeliU  r   c                     [        U S/5        g r
   r   r   s      r   r   !ZImageTransformer2DModel.__init__X  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $ZImageTransformer2DModel.from_config[  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ZImageTransformer2DModel.from_pretrained_  r   r   r   Nr   r   r   r   r  r  U  r'   r   r  c                  &    [        [        S/5        g r
   )r   attention_backendr   s     r   r  r  d  s    ''3r   c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineBlocksih  r   c                     [        U S/5        g r
   r   r   s      r   r   AutoPipelineBlocks.__init__k  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AutoPipelineBlocks.from_confign  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoPipelineBlocks.from_pretrainedr  r   r   r   Nr   r   r   r   r  r  h  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ComponentsManageriw  r   c                     [        U S/5        g r
   r   r   s      r   r   ComponentsManager.__init__z  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  w  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   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ConditionalPipelineBlocksi  r   c                     [        U S/5        g r
   r   r   s      r   r   "ConditionalPipelineBlocks.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %ConditionalPipelineBlocks.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )ConditionalPipelineBlocks.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)
ConfigSpeci  r   c                     [        U S/5        g r
   r   r   s      r   r   ConfigSpec.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ConfigSpec.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ConfigSpec.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)
InputParami  r   c                     [        U S/5        g r
   r   r   s      r   r   InputParam.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   InputParam.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   InputParam.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)LoopSequentialPipelineBlocksi  r   c                     [        U S/5        g r
   r   r   s      r   r   %LoopSequentialPipelineBlocks.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (LoopSequentialPipelineBlocks.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,LoopSequentialPipelineBlocks.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)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   r  r    r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)OutputParami  r   c                     [        U S/5        g r
   r   r   s      r   r   OutputParam.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   OutputParam.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   OutputParam.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)SequentialPipelineBlocksi  r   c                     [        U S/5        g r
   r   r   s      r   r   !SequentialPipelineBlocks.__init__	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $SequentialPipelineBlocks.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (SequentialPipelineBlocks.from_pretrained	  r   r   r   Nr   r   r   r   r  r    r'   r   r  c                  &    [        [        S/5        g r
   )r   get_constant_scheduler   s     r   r  r  	  s    +gY7r   c                  &    [        [        S/5        g r
   )r   !get_constant_schedule_with_warmupr   s     r   r  r  	  r   r   c                  &    [        [        S/5        g r
   )r   get_cosine_schedule_with_warmupr   s     r   r  r  	      5yAr   c                  &    [        [        S/5        g r
   )r   2get_cosine_with_hard_restarts_schedule_with_warmupr   s     r   r  r  	  s    H7)Tr   c                  &    [        [        S/5        g r
   )r   get_linear_schedule_with_warmupr   s     r   r  r  	  r	  r   c                  &    [        [        S/5        g r
   )r   )get_polynomial_decay_schedule_with_warmupr   s     r   r  r  !	  s    ?'Kr   c                  &    [        [        S/5        g r
   )r   get_schedulerr   s     r   r  r  %	  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_pretrained3	  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)AutoPipelineForImage2Imagei8	  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_pretrainedB	  r   r   r   Nr   r   r   r   r  r  8	  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineForInpaintingiG	  r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoPipelineForInpainting.__init__J	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %AutoPipelineForInpainting.from_configM	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   )AutoPipelineForInpainting.from_pretrainedQ	  r   r   r   Nr   r   r   r   r#  r#  G	  r'   r   r#  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AutoPipelineForText2ImageiV	  r   c                     [        U S/5        g r
   r   r   s      r   r   "AutoPipelineForText2Image.__init__Y	  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   r+  r+  V	  r'   r   r+  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BlipDiffusionControlNetPipelineie	  r   c                     [        U S/5        g r
   r   r   s      r   r   (BlipDiffusionControlNetPipeline.__init__h	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +BlipDiffusionControlNetPipeline.from_configk	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /BlipDiffusionControlNetPipeline.from_pretrainedo	  r   r   r   Nr   r   r   r   r3  r3  e	  r'   r   r3  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BlipDiffusionPipelineit	  r   c                     [        U S/5        g r
   r   r   s      r   r   BlipDiffusionPipeline.__init__w	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !BlipDiffusionPipeline.from_configz	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %BlipDiffusionPipeline.from_pretrained~	  r   r   r   Nr   r   r   r   r;  r;  t	  r'   r   r;  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)CLIPImageProjectioni	  r   c                     [        U S/5        g r
   r   r   s      r   r   CLIPImageProjection.__init__	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   CLIPImageProjection.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #CLIPImageProjection.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)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_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (ConsistencyModelPipeline.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)DanceDiffusionPipelinei	  r   c                     [        U S/5        g r
   r   r   s      r   r   DanceDiffusionPipeline.__init__	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "DanceDiffusionPipeline.from_config	  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &DanceDiffusionPipeline.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)DDIMPipelinei	  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[  	  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   rc  rc  	  r'   r   rc  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   rk  rk  	  r'   r   rk  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   rs  rs  	  r'   r   rs  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_pretrained2
  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)RePaintPipelinei7
  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_pretrainedA
  r   r   r   Nr   r   r   r   r  r  7
  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)ScoreSdeVePipelineiF
  r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVePipeline.__init__I
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ScoreSdeVePipeline.from_configL
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "ScoreSdeVePipeline.from_pretrainedP
  r   r   r   Nr   r   r   r   r  r  F
  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)StableDiffusionMixiniU
  r   c                     [        U S/5        g r
   r   r   s      r   r   StableDiffusionMixin.__init__X
  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  U
  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DiffusersQuantizerid
  r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusersQuantizer.__init__g
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   DiffusersQuantizer.from_configj
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "DiffusersQuantizer.from_pretrainedn
  r   r   r   Nr   r   r   r   r  r  d
  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)AmusedScheduleris
  r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.__init__v
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.from_configy
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   AmusedScheduler.from_pretrained}
  r   r   r   Nr   r   r   r   r  r  s
  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)BlockRefinementScheduleri
  r   c                     [        U S/5        g r
   r   r   s      r   r   !BlockRefinementScheduler.__init__
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   $BlockRefinementScheduler.from_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (BlockRefinementScheduler.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)BlockRefinementSchedulerOutputi
  r   c                     [        U S/5        g r
   r   r   s      r   r   'BlockRefinementSchedulerOutput.__init__
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *BlockRefinementSchedulerOutput.from_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .BlockRefinementSchedulerOutput.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)CMStochasticIterativeScheduleri
  r   c                     [        U S/5        g r
   r   r   s      r   r   'CMStochasticIterativeScheduler.__init__
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *CMStochasticIterativeScheduler.from_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .CMStochasticIterativeScheduler.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)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_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &CogVideoXDDIMScheduler.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)CogVideoXDPMScheduleri
  r   c                     [        U S/5        g r
   r   r   s      r   r   CogVideoXDPMScheduler.__init__
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !CogVideoXDPMScheduler.from_config
  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %CogVideoXDPMScheduler.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)DDIMInverseScheduleri
  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  
  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   r  r  
  r'   r   r  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   r  r  
  r'   r   r  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   r  r  	  r'   r   r  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   r  r    r'   r   r  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_pretrained1  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)"DPMSolverMultistepInverseScheduleri6  r   c                     [        U S/5        g r
   r   r   s      r   r   +DPMSolverMultistepInverseScheduler.__init__9  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   r+  r+  6  r'   r   r+  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DPMSolverMultistepScheduleriE  r   c                     [        U S/5        g r
   r   r   s      r   r   $DPMSolverMultistepScheduler.__init__H  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'DPMSolverMultistepScheduler.from_configK  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +DPMSolverMultistepScheduler.from_pretrainedO  r   r   r   Nr   r   r   r   r3  r3  E  r'   r   r3  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)DPMSolverSinglestepScheduleriT  r   c                     [        U S/5        g r
   r   r   s      r   r   %DPMSolverSinglestepScheduler.__init__W  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (DPMSolverSinglestepScheduler.from_configZ  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,DPMSolverSinglestepScheduler.from_pretrained^  r   r   r   Nr   r   r   r   r;  r;  T  r'   r   r;  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EDMDPMSolverMultistepScheduleric  r   c                     [        U S/5        g r
   r   r   s      r   r   'EDMDPMSolverMultistepScheduler.__init__f  r   r   c                     [        U S/5        g r
   r   r   s      r   r   *EDMDPMSolverMultistepScheduler.from_configi  r   r   c                     [        U S/5        g r
   r   r   s      r   r   .EDMDPMSolverMultistepScheduler.from_pretrainedm  r   r   r   Nr   r   r   r   rC  rC  c  r'   r   rC  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)EDMEulerSchedulerir  r   c                     [        U S/5        g r
   r   r   s      r   r   EDMEulerScheduler.__init__u  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EDMEulerScheduler.from_configx  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !EDMEulerScheduler.from_pretrained|  r   r   r   Nr   r   r   r   rK  rK  r  r'   r   rK  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__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +EulerAncestralDiscreteScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /EulerAncestralDiscreteScheduler.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)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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   &EulerDiscreteScheduler.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)FlowMatchEulerDiscreteScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   (FlowMatchEulerDiscreteScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   +FlowMatchEulerDiscreteScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   /FlowMatchEulerDiscreteScheduler.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)FlowMatchHeunDiscreteScheduleri  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_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)FlowMatchLCMScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   FlowMatchLCMScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   !FlowMatchLCMScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   %FlowMatchLCMScheduler.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)HeliosDMDScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   HeliosDMDScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   HeliosDMDScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   "HeliosDMDScheduler.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)HeliosScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r   HeliosScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   HeliosScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   HeliosScheduler.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)HeunDiscreteScheduleri  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    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_pretrained0  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)LCMScheduleri5  r   c                     [        U S/5        g r
   r   r   s      r   r   LCMScheduler.__init__8  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  5  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)LTXEulerAncestralRFScheduleriD  r   c                     [        U S/5        g r
   r   r   s      r   r   %LTXEulerAncestralRFScheduler.__init__G  r   r   c                     [        U S/5        g r
   r   r   s      r   r   (LTXEulerAncestralRFScheduler.from_configJ  r   r   c                     [        U S/5        g r
   r   r   s      r   r   ,LTXEulerAncestralRFScheduler.from_pretrainedN  r   r   r   Nr   r   r   r   r  r  D  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)PNDMScheduleriS  r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMScheduler.__init__V  r   r   c                     [        U S/5        g r
   r   r   s      r   r   PNDMScheduler.from_configY  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  S  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)RePaintSchedulerib  r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintScheduler.__init__e  r   r   c                     [        U S/5        g r
   r   r   s      r   r   RePaintScheduler.from_configh  r   r   c                     [        U S/5        g r
   r   r   s      r   r    RePaintScheduler.from_pretrainedl  r   r   r   Nr   r   r   r   r  r  b  r'   r   r  c                   @    \ rS rSrS/rS r\S 5       r\S 5       rSr	g)SASolverScheduleriq  r   c                     [        U S/5        g r
   r   r   s      r   r   SASolverScheduler.__init__t  r   r   c                     [        U S/5        g r
   r   r   s      r   r   SASolverScheduler.from_configw  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  q  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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   TCDScheduler.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)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_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   UnCLIPScheduler.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)UniPCMultistepScheduleri  r   c                     [        U S/5        g r
   r   r   s      r   r    UniPCMultistepScheduler.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   #UniPCMultistepScheduler.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   'UniPCMultistepScheduler.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)VQDiffusionScheduleri  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_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)EMAModeli  r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   EMAModel.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)VideoProcessori  r   c                     [        U S/5        g r
   r   r   s      r   r   VideoProcessor.__init__  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VideoProcessor.from_config  r   r   c                     [        U S/5        g r
   r   r   s      r   r   VideoProcessor.from_pretrained  r   r   r   Nr   r   r   r   r  r    r'   r   r  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  r"  r*  r2  r:  rB  rJ  rR  rZ  rb  rj  rr  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r"  r*  r2  r:  rB  rJ  rR  rZ  rb  rj  rr  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r"  r*  r2  r:  rB  rJ  rR  rZ  rb  rj  rr  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r
  r  r  r"  r*  r2  r:  rB  rJ  rR  rZ  rb  rj  rr  rz  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r#  r+  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   <module>r#     s
   2*+ **[ **[ **[ **{ **{ **; **; **+ **{ ** **+ **k **[ ** **{ ** **[ **k **+ *5:32D96* **{ **[ **+ **{ **+ **+ ** **[ **; **k **k **[ **{ **K **; **+ ** **+ **K **+ **{ **{ **{ **k **K **; **{ **[ ** **; **{ ** **+ **+ **; **{ **; ** **[ **K **{ **; **; **k **k ** **[ **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 **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   