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

This module contains an implementation of the Wan video diffusion pipeline
using the modular pipeline architecture.
    )FlowUniPCMultistepScheduler)ComposedPipelineBase)LoRAPipeline)
ServerArgs)init_loggerc                   @   s<   e Zd ZdZd Zg dZdefddZdeddfdd	ZdS )
WanPipelinez9
    Wan video diffusion pipeline with LoRA support.
    )text_encoder	tokenizervaetransformer	schedulerserver_argsc                 C   s   t |jjd| jd< d S )N)shiftr   )r   pipeline_config
flow_shiftmodulesselfr    r   h/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/runtime/pipelines/wan_pipeline.pyinitialize_pipeline'   s   zWanPipeline.initialize_pipelinereturnNc                 C   s   |    d S )N)add_standard_t2i_stagesr   r   r   r   create_pipeline_stages-   s   z"WanPipeline.create_pipeline_stages)	__name__
__module____qualname____doc__pipeline_name_required_config_modulesr   r   r   r   r   r   r   r      s    r   N)r   Osglang.multimodal_gen.runtime.models.schedulers.scheduling_flow_unipc_multistepr   Csglang.multimodal_gen.runtime.pipelines_core.composed_pipeline_baser   :sglang.multimodal_gen.runtime.pipelines_core.lora_pipeliner   )sglang.multimodal_gen.runtime.server_argsr   1sglang.multimodal_gen.runtime.utils.logging_utilsr   r   loggerr   
EntryClassr   r   r   r   <module>   s   