
    
3jz                         S SK Jr  S SKJr  SSKJr  \R                  " \5      r " S S\\5      r	 " S S\	5      r
 " S	 S
\	5      rg)   )HeliosLoraLoaderMixin)logging   )ModularPipelinec                   \    \ rS rSrSrSr\S 5       r\S 5       r\S 5       r	\S 5       r
Srg	)
HeliosModularPipeline   z
A ModularPipeline for Helios text-to-video generation.

> [!WARNING] > This is an experimental feature and is likely to change in the future.
HeliosAutoBlocksc                     Sn[        U S5      (       a-  U R                  b   U R                  R                  R                  nU$ )N   vae)hasattrr   configscale_factor_spatialselfvae_scale_factors     m/home/wildlama/miniconda3/lib/python3.13/site-packages/diffusers/modular_pipelines/helios/modular_pipeline.pyvae_scale_factor_spatial.HeliosModularPipeline.vae_scale_factor_spatial$   s8    4DHH$8#xxCC    c                     Sn[        U S5      (       a-  U R                  b   U R                  R                  R                  nU$ )N   r   )r   r   r   scale_factor_temporalr   s     r   vae_scale_factor_temporal/HeliosModularPipeline.vae_scale_factor_temporal+   s8    4DHH$8#xxDDr   c                     Sn[        U S5      (       a-  U R                  b   U R                  R                  R                  nU$ )N   transformer)r   r   r   in_channels)r   num_channels_latentss     r   r!   *HeliosModularPipeline.num_channels_latents2   s@      "4''D,<,<,H#'#3#3#:#:#F#F ##r   c                     Sn[        U S5      (       aC  U R                  b6  U R                  R                  =(       a    U R                  R                  S:  nU$ )NFguider   )r   r$   _enablednum_conditions)r   requires_unconditional_embedss     r   r(   3HeliosModularPipeline.requires_unconditional_embeds:   sI    (-%4""t{{'>,0KK,@,@,cT[[E_E_bcEc),,r    N)__name__
__module____qualname____firstlineno____doc__default_blocks_namepropertyr   r   r!   r(   __static_attributes__r*   r   r   r   r      s]     -        $ $ - -r   r   c                       \ rS rSrSrSrSrg)HeliosPyramidModularPipelineD   z
A ModularPipeline for Helios pyramid (progressive resolution) video generation.

> [!WARNING] > This is an experimental feature and is likely to change in the future.
HeliosPyramidAutoBlocksr*   Nr+   r,   r-   r.   r/   r0   r2   r*   r   r   r4   r4   D   s     4r   r4   c                       \ rS rSrSrSrSrg)%HeliosPyramidDistilledModularPipelineN   a
  
A ModularPipeline for Helios distilled pyramid video generation using DMD scheduler.

Uses guidance_scale=1.0 (no CFG) and supports is_amplify_first_chunk for the DMD scheduler.

> [!WARNING] > This is an experimental feature and is likely to change in the future.
 HeliosPyramidDistilledAutoBlocksr*   Nr7   r*   r   r   r9   r9   N   s     =r   r9   N)loadersr   utilsr   modular_pipeliner   
get_loggerr+   loggerr   r4   r9   r*   r   r   <module>rA      sM     -  . 
		H	%)-)-X4#8 4	=,A 	=r   