[diffusion] chore: tiny fix model config (#16159)
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user