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

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
co-authored by Mick
parent 3449806727
commit 39ca57cd28
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -429,7 +429,7 @@ def _register_configs():
register_configs( register_configs(
sampling_param_cls=QwenImageSamplingParams, sampling_param_cls=QwenImageSamplingParams,
pipeline_config_cls=QwenImagePipelineConfig, pipeline_config_cls=QwenImagePipelineConfig,
hf_model_paths=["Qwen/Qwen-Image"], hf_model_paths=["Qwen/Qwen-Image", "Qwen/Qwen-Image-2512"],
) )
register_configs( register_configs(
sampling_param_cls=QwenImageSamplingParams, sampling_param_cls=QwenImageSamplingParams,
@@ -156,7 +156,7 @@ class ComponentLoader(ABC):
component = self.load_customized( component = self.load_customized(
component_model_path, server_args, module_name component_model_path, server_args, module_name
) )
source = "customized" source = "sgl-diffusion"
except Exception as e: except Exception as e:
if "Unsupported model architecture" in str(e): if "Unsupported model architecture" in str(e):
logger.info( logger.info(
@@ -190,7 +190,7 @@ class ComponentLoader(ABC):
if consumed is None or consumed == 0.0: if consumed is None or consumed == 0.0:
consumed = gpu_mem_before_loading - current_gpu_mem consumed = gpu_mem_before_loading - current_gpu_mem
logger.info( 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, module_name,
component.__class__.__name__, component.__class__.__name__,
consumed, consumed,