code: def combine_peft_state_dict(
      info: Dict[str, LoRAInfo]
  ) -> Dict[str, torch.Tensor]:
      result = {}
      for key_name, key_info in info.items():
          result[
              f"base_model.model.{key_name}.lora_A.weight"
          ] = key_info.lora_A
          result[
              f"base_model.model.{key_name}.lora_B.weight"
          ] = key_info.lora_B
      return result
  
file_path: peft/examples/lora_dreambooth/convert_kohya_ss_sd_lora_to_peft.py
imports: import argparse
  import os
  from collections import Counter
  from dataclasses import dataclass
  from typing import Dict, Optional
  import safetensors
  import torch
  from diffusers import UNet2DConditionModel
  from transformers import CLIPTextModel
  from peft import LoraConfig, get_peft_model, get_peft_model_state_dict, set_peft_model_state_dict
label_desc: Use image node_id 2 for calling a global function with example usage: combine_peft_state_dict(info) and returns: result