Function _collate_lora_metadata

src/diffusers/training_utils.py:352–357

(modules_to_save: dict[str, torch.nn.Module])

def _collate_lora_metadata(modules_to_save: dict[str, torch.nn.Module]) -> dict[str, Any]:
    metadatas = {}
    for module_name, module in modules_to_save.items():
        if module is not None:
            metadatas[f"{module_name}_lora_adapter_metadata"] = module.peft_config["default"].to_dict()
    return metadatas


def compute_density_for_timestep_sampling(