kvcache-ai/sglang
1
0
Fork 0
mirror of https://github.com/kvcache-ai/sglang.git synced 2026-07-08 08:17:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

[diffusion] chore: tiny fix model config (#16159)

Browse Source 
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
Yuhao Yang
2025-12-30 22:11:38 +08:00
committed by GitHub
parent 3449806727
commit 39ca57cd28
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/multimodal_gen/registry.py
View File

@@ -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,

4
python/sglang/multimodal_gen/runtime/loader/component_loader.py
View File

@@ -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,