o
    پi                     @   sl   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	 d dl
mZ eeZG dd deeZegZd	S )
    )LoRAPipeline)ComposedPipelineBase)DenoisingStage)GlmImageBeforeDenoisingStage)
ServerArgs)init_loggerc                   @   s&   e Zd Zd Zg dZdefddZdS )GlmImagePipeline)text_encoder	tokenizervaevision_language_encoder	processortransformer	schedulerserver_argsc                 C   sr   |  t| d| d| d| d| d| d| ddd	 |  t| d| dd
 |   d S )Nr   r	   r
   r   r   r   r   )r   r	   r
   r   r   r   r    glm_image_before_denoising_stage)r   r   )	add_stager   
get_moduler   add_standard_decoding_stage)selfr    r   e/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/runtime/pipelines/glm_image.pycreate_pipeline_stages   s&   	z'GlmImagePipeline.create_pipeline_stagesN)__name__
__module____qualname__pipeline_name_required_config_modulesr   r   r   r   r   r   r      s    
r   N),sglang.multimodal_gen.runtime.pipelines_corer   Csglang.multimodal_gen.runtime.pipelines_core.composed_pipeline_baser   3sglang.multimodal_gen.runtime.pipelines_core.stagesr   Ssglang.multimodal_gen.runtime.pipelines_core.stages.model_specific_stages.glm_imager   )sglang.multimodal_gen.runtime.server_argsr   1sglang.multimodal_gen.runtime.utils.logging_utilsr   r   loggerr   
EntryClassr   r   r   r   <module>   s    
%