o پir @s6ddlmZddlmZddlmZGdddZdS))Any) ModelConfig) ServerArgsc @sDeZdZdedeeefdededef ddZede fd d Z d S) DllmConfig algorithmalgorithm_config block_sizemask_idmax_running_requestscCs"||_||_||_||_||_dS)Nrrrr r )selfrrrr r r J/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/dllm/config.py__init__s  zDllmConfig.__init__ server_argsc Cs|jdurdStj||j|jd}dddddddddd}|jjd}||vr7||}|d }|d }ntd ||jdurEd n|j}i}|j durzddl } Wn t y`t d wt |j d } | | }Wdn1swwY|d |}t|j||||dS)N) model_pathmodel_revision id)rr iuP)LLaDA2MoeModelLMSDARForCausalLMSDARMoeForCausalLMrrr zUnknown diffusion LLM: zDPlease install PyYAML to use YAML config files. `pip install pyyaml`rr )dllm_algorithmrfrom_server_argsrrevision hf_config architectures RuntimeErrorr dllm_algorithm_configyaml ImportErroropen safe_loadgetr) r model_config DLLM_PARAMSarchparamsrr r rr!fr r rrsR         zDllmConfig.from_server_argsN) __name__ __module__ __qualname__strdictrintr staticmethodrrr r r rrs   rN)typingrsglang.srt.configs.model_configrsglang.srt.server_argsrrr r r rs