o
    پi                     @   s$   d dl Z d dlmZmZ dd ZdS )    N)get_is_diffusion_modelget_model_pathc                    s   t  fdddD r"ddlm} tjdd}|| |  d S t }t|}|rIddlm}m} tjdd}|| | }|| d S t	d	| )
Nc                 3   s    | ]}| v V  qd S )N ).0h
extra_argvr   G/home/ubuntu/.local/lib/python3.10/site-packages/sglang/cli/generate.py	<genexpr>   s    zgenerate.<locals>.<genexpr>)z-hz--helpr   ) add_multimodal_gen_generate_argszSGLang Multimodal Generation)description)r   generate_cmdz4Generate subcommand is not yet supported for model: )
any6sglang.multimodal_gen.runtime.entrypoints.cli.generater   argparseArgumentParser
parse_argsr   r   r   	Exception)argsr   r   parser
model_pathis_diffusion_modelr   parsed_argsr   r   r	   generate   s"   

r   )r   sglang.cli.utilsr   r   r   r   r   r   r	   <module>   s    