[diffusion] feat: support LoRA (#13859)

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

@@ -55,7 +55,7 @@ class skip_init_modules:
             cls.reset_parameters = lambda self: None  # skip init
 
     def __exit__(self, exc_type, exc_value, traceback):
-        # Restore originals
+        # restore originals
         for cls, orig in self._orig_reset.items():
             cls.reset_parameters = orig
 
@@ -199,7 +199,7 @@ class TextEncoderLoader(ComponentLoader):
     def _get_weights_iterator(
         self, source: "Source", to_cpu: bool
     ) -> Generator[tuple[str, torch.Tensor], None, None]:
-        """Get an iterator for the model weights based on the load format."""
+        """get an iterator for the model weights based on the load format."""
         hf_folder, hf_weights_files, use_safetensors = self._prepare_weights(
             source.model_or_path,
             source.fall_back_to_pt,
@@ -214,7 +214,7 @@ class TextEncoderLoader(ComponentLoader):
 
         if self.counter_before_loading_weights == 0.0:
             self.counter_before_loading_weights = time.perf_counter()
-        # Apply the prefix.
+        # apply the prefix.
         return ((source.prefix + name, tensor) for (name, tensor) in weights_iterator)
 
     def _get_all_weights(

python/sglang/multimodal_gen/runtime/loader/utils.py

@@ -32,7 +32,6 @@ def get_param_names_mapping(
 
     Args:
         mapping_dict (Dict[str, str]): Dictionary mapping regex patterns to replacement patterns
-        param_name (str): The parameter name to be transformed
 
     Returns:
         Callable[[str], str]: A function that maps parameter names from source to target format
@@ -44,13 +43,13 @@ def get_param_names_mapping(
             match = re.match(pattern, name)
             if match:
                 merge_index = None
-                total_splitted_params = None
+                total_split_params = None
                 if isinstance(replacement, tuple):
                     merge_index = replacement[1]
-                    total_splitted_params = replacement[2]
+                    total_split_params = replacement[2]
                     replacement = replacement[0]
                 name = re.sub(pattern, replacement, name)
-                return name, merge_index, total_splitted_params
+                return name, merge_index, total_split_params
 
         # If no pattern matches, return the original name
         return name, None, None