[diffusion] refactor: group component loaders under the component_loaders/ directory (#18438)

python/sglang/multimodal_gen/runtime/loader/component_loaders/adapter_loader.py

@@ -1,7 +1,9 @@
 from safetensors.torch import load_file as safetensors_load_file
 
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.utils import (
     _list_safetensors_files,
     set_default_torch_dtype,

python/sglang/multimodal_gen/runtime/loader/component_loaders/bridge_loader.py

@@ -3,7 +3,9 @@ from copy import deepcopy
 import torch
 
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.fsdp_load import maybe_load_fsdp_model
 from sglang.multimodal_gen.runtime.loader.utils import _list_safetensors_files
 from sglang.multimodal_gen.runtime.models.registry import ModelRegistry

python/sglang/multimodal_gen/runtime/loader/component_loaders/component_loader.py

@@ -190,7 +190,10 @@ class ComponentLoader(ABC):
             return
 
         package_dir = os.path.dirname(__file__)
-        package_name = __package__ or "sglang.multimodal_gen.runtime.loader"
+        package_name = (
+            __package__
+            or "sglang.multimodal_gen.runtime.loader.component_loaders.component_loaders"
+        )
 
         for _, name, _ in pkgutil.iter_modules([package_dir]):
             # skip importing self to avoid circular dependency issues

python/sglang/multimodal_gen/runtime/loader/component_loaders/image_encoder_loader.py

@@ -2,7 +2,9 @@ import json
 import os
 
 from sglang.multimodal_gen.configs.models import ModelConfig
-from sglang.multimodal_gen.runtime.loader.text_encoder_loader import TextEncoderLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.text_encoder_loader import (
+    TextEncoderLoader,
+)
 from sglang.multimodal_gen.runtime.loader.utils import _clean_hf_config_inplace
 from sglang.multimodal_gen.runtime.server_args import ServerArgs
 from sglang.multimodal_gen.runtime.utils.logging_utils import init_logger

python/sglang/multimodal_gen/runtime/loader/component_loaders/scheduler_loader.py

@@ -1,4 +1,6 @@
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.models.registry import ModelRegistry
 from sglang.multimodal_gen.runtime.server_args import ServerArgs
 from sglang.multimodal_gen.runtime.utils.hf_diffusers_utils import (

python/sglang/multimodal_gen/runtime/loader/component_loaders/text_encoder_loader.py

@@ -16,7 +16,9 @@ from sglang.multimodal_gen.configs.pipeline_configs.qwen_image import (
     QwenImageEditPipelineConfig,
 )
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.fsdp_load import shard_model
 from sglang.multimodal_gen.runtime.loader.utils import (
     _clean_hf_config_inplace,

python/sglang/multimodal_gen/runtime/loader/component_loaders/transformer_loader.py

@@ -4,7 +4,9 @@ from copy import deepcopy
 import torch
 
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.fsdp_load import maybe_load_fsdp_model
 from sglang.multimodal_gen.runtime.loader.utils import (
     _list_safetensors_files,

python/sglang/multimodal_gen/runtime/loader/component_loaders/vae_loader.py

@@ -4,7 +4,9 @@ import os
 from safetensors.torch import load_file as safetensors_load_file
 
 from sglang.multimodal_gen.configs.models import ModelConfig
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.utils import (
     _list_safetensors_files,
     set_default_torch_dtype,

python/sglang/multimodal_gen/runtime/loader/component_loaders/vl_encoder_loader.py

@@ -1,7 +1,9 @@
 from typing import Any
 
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.server_args import ServerArgs
 from sglang.multimodal_gen.runtime.utils.hf_diffusers_utils import get_hf_config
 

python/sglang/multimodal_gen/runtime/loader/component_loaders/vocoder_loader.py

@@ -1,7 +1,9 @@
 from safetensors.torch import load_file as safetensors_load_file
 
 from sglang.multimodal_gen.configs.models import ModelConfig
-from sglang.multimodal_gen.runtime.loader.component_loader import ComponentLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
+    ComponentLoader,
+)
 from sglang.multimodal_gen.runtime.loader.utils import (
     _list_safetensors_files,
     set_default_torch_dtype,

python/sglang/multimodal_gen/runtime/pipelines_core/composed_pipeline_base.py

@@ -14,7 +14,7 @@ from typing import Any, cast
 import torch
 from tqdm import tqdm
 
-from sglang.multimodal_gen.runtime.loader.component_loader import (
+from sglang.multimodal_gen.runtime.loader.component_loaders.component_loader import (
     PipelineComponentLoader,
 )
 from sglang.multimodal_gen.runtime.pipelines_core.executors.pipeline_executor import (

python/sglang/multimodal_gen/runtime/pipelines_core/stages/decoding.py

@@ -10,7 +10,7 @@ import weakref
 import torch
 
 from sglang.multimodal_gen.runtime.distributed import get_local_torch_device
-from sglang.multimodal_gen.runtime.loader.vae_loader import VAELoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.vae_loader import VAELoader
 from sglang.multimodal_gen.runtime.models.vaes.common import ParallelTiledVAE
 from sglang.multimodal_gen.runtime.pipelines_core.schedule_batch import OutputBatch, Req
 from sglang.multimodal_gen.runtime.pipelines_core.stages.base import (

python/sglang/multimodal_gen/runtime/pipelines_core/stages/denoising.py

@@ -55,7 +55,9 @@ from sglang.multimodal_gen.runtime.layers.attention.STA_configuration import (
     configure_sta,
     save_mask_search_results,
 )
-from sglang.multimodal_gen.runtime.loader.transformer_loader import TransformerLoader
+from sglang.multimodal_gen.runtime.loader.component_loaders.transformer_loader import (
+    TransformerLoader,
+)
 from sglang.multimodal_gen.runtime.managers.forward_context import set_forward_context
 from sglang.multimodal_gen.runtime.pipelines_core.schedule_batch import Req
 from sglang.multimodal_gen.runtime.pipelines_core.stages.base import (