[diffusion] model: GLM-Image (#16894)
Co-authored-by: jianyingzhu <53300651@qq.com>
This commit is contained in:
co-authored by
jianyingzhu
parent
2a7b67adff
commit
a0b4ba9032
@@ -39,6 +39,9 @@ from sglang.multimodal_gen.configs.pipeline_configs import (
|
||||
)
|
||||
from sglang.multimodal_gen.configs.pipeline_configs.base import PipelineConfig
|
||||
from sglang.multimodal_gen.configs.pipeline_configs.flux import Flux2PipelineConfig
|
||||
from sglang.multimodal_gen.configs.pipeline_configs.glm_image import (
|
||||
GlmImagePipelineConfig,
|
||||
)
|
||||
from sglang.multimodal_gen.configs.pipeline_configs.qwen_image import (
|
||||
QwenImageEditPipelineConfig,
|
||||
QwenImageEditPlus_2511_PipelineConfig,
|
||||
@@ -56,6 +59,7 @@ from sglang.multimodal_gen.configs.pipeline_configs.wan import (
|
||||
Wan2_2_TI2V_5B_Config,
|
||||
)
|
||||
from sglang.multimodal_gen.configs.sample.flux import FluxSamplingParams
|
||||
from sglang.multimodal_gen.configs.sample.glmimage import GlmImageSamplingParams
|
||||
from sglang.multimodal_gen.configs.sample.hunyuan import (
|
||||
FastHunyuanSamplingParam,
|
||||
HunyuanSamplingParams,
|
||||
@@ -581,5 +585,11 @@ def _register_configs():
|
||||
hf_model_paths=["Qwen/Qwen-Image-Layered"],
|
||||
)
|
||||
|
||||
register_configs(
|
||||
sampling_param_cls=GlmImageSamplingParams,
|
||||
pipeline_config_cls=GlmImagePipelineConfig,
|
||||
model_detectors=[lambda hf_id: "glm-image" in hf_id.lower()],
|
||||
)
|
||||
|
||||
|
||||
_register_configs()
|
||||
|
||||
Reference in New Issue
Block a user