From 39ca57cd28f6bcb3e91908c244a0ebcffaa02087 Mon Sep 17 00:00:00 2001 From: Yuhao Yang <47235274+yhyang201@users.noreply.github.com> Date: Tue, 30 Dec 2025 22:11:38 +0800 Subject: [PATCH] [diffusion] chore: tiny fix model config (#16159) Co-authored-by: Mick --- python/sglang/multimodal_gen/registry.py | 2 +- .../sglang/multimodal_gen/runtime/loader/component_loader.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sglang/multimodal_gen/registry.py b/python/sglang/multimodal_gen/registry.py index e29261cfa..b9e91d044 100644 --- a/python/sglang/multimodal_gen/registry.py +++ b/python/sglang/multimodal_gen/registry.py @@ -429,7 +429,7 @@ def _register_configs(): register_configs( sampling_param_cls=QwenImageSamplingParams, pipeline_config_cls=QwenImagePipelineConfig, - hf_model_paths=["Qwen/Qwen-Image"], + hf_model_paths=["Qwen/Qwen-Image", "Qwen/Qwen-Image-2512"], ) register_configs( sampling_param_cls=QwenImageSamplingParams, diff --git a/python/sglang/multimodal_gen/runtime/loader/component_loader.py b/python/sglang/multimodal_gen/runtime/loader/component_loader.py index 3ae15135d..393f15ca9 100644 --- a/python/sglang/multimodal_gen/runtime/loader/component_loader.py +++ b/python/sglang/multimodal_gen/runtime/loader/component_loader.py @@ -156,7 +156,7 @@ class ComponentLoader(ABC): component = self.load_customized( component_model_path, server_args, module_name ) - source = "customized" + source = "sgl-diffusion" except Exception as e: if "Unsupported model architecture" in str(e): logger.info( @@ -190,7 +190,7 @@ class ComponentLoader(ABC): if consumed is None or consumed == 0.0: consumed = gpu_mem_before_loading - current_gpu_mem logger.info( - f"Loaded %s: %s from {source}. model size: %.2f GB, avail mem: %.2f GB", + f"Loaded %s: %s ({source} version). model size: %.2f GB, avail mem: %.2f GB", module_name, component.__class__.__name__, consumed,