[diffusion] chore: minor refactor by streamlining the VAE class hierarchy (#16069)

2025-12-29 23:37:59 +08:00
parent 88f3de2514
commit b5d9fc873b
7 changed files with 22 additions and 22 deletions
--- a/python/sglang/multimodal_gen/runtime/loader/component_loader.py
+++ b/python/sglang/multimodal_gen/runtime/loader/component_loader.py
@@ -192,11 +192,11 @@ 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}. avail mem: %.2f GB, %.2f GB consumed",
+                f"Loaded %s: %s from {source}. model size: %.2f GB, avail mem: %.2f GB",
                module_name,
                component.__class__.__name__,
-                current_gpu_mem,
                consumed,
+                current_gpu_mem,
            )
        return component, consumed