code: def combine_peft_state_dict(
      info: Dict[str, Union[LoRAInfo, LoHaInfo]]
  ) -> Dict[str, torch.Tensor]:
      result = {}
      for key_info in info.values():
          result.update(key_info.peft_state_dict())
      return result
  
imports: import argparse
  import json
  import logging
  import os
  from collections import Counter
  from dataclasses import dataclass
  from operator import attrgetter
  from typing import Dict, List, Optional, Union
  import safetensors
  import torch
  import torch.nn
  from diffusers import UNet2DConditionModel
  from transformers import CLIPTextModel
  from peft import LoHaConfig, LoKrConfig, LoraConfig, PeftType, get_peft_model, set_peft_model_state_dict
  from peft.tuners.lokr.layer import factorization
label_desc: Use image node_id 4 for calling a global function with example usage: combine_peft_state_dict(info) and returns: result