o
    پi
                     @   sX   d Z ddlmZ ddlmZmZ ddlmZ ddlm	Z	 e	e
ZG dd deZeZdS )	z
Hunyuan video diffusion pipeline implementation.

This module contains an implementation of the Hunyuan video diffusion pipeline
using the modular pipeline architecture.
    )ComposedPipelineBase)InputValidationStageTextEncodingStage)
ServerArgs)init_loggerc                   @   s&   e Zd Zd Zg dZdefddZdS )HunyuanVideoPipeline)text_encodertext_encoder_2	tokenizertokenizer_2vaetransformer	schedulerserver_argsc                 C   sd   |  t  |  t| d| dg| d| dgdd |   |   |   |   d S )Nr   r	   r
   r   )text_encoders
tokenizersprompt_encoding_stage_primary)	add_stager   r   
get_module'add_standard_timestep_preparation_stage%add_standard_latent_preparation_stageadd_standard_denoising_stageadd_standard_decoding_stage)selfr    r   l/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/runtime/pipelines/hunyuan_pipeline.pycreate_pipeline_stages(   s    
z+HunyuanVideoPipeline.create_pipeline_stagesN)__name__
__module____qualname__pipeline_name_required_config_modulesr   r   r   r   r   r   r      s    
r   N)__doc__Csglang.multimodal_gen.runtime.pipelines_core.composed_pipeline_baser   3sglang.multimodal_gen.runtime.pipelines_core.stagesr   r   )sglang.multimodal_gen.runtime.server_argsr   1sglang.multimodal_gen.runtime.utils.logging_utilsr   r   loggerr   
EntryClassr   r   r   r   <module>   s   #