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,