code: def __init__(self, diff_model_config, conditioning_key):
      super().__init__()
      if isinstance(diff_model_config, nn.Module):
          self.diffusion_model = diff_model_config
      else:
          self.diffusion_model = DiffusionWrapper.from_config_dict(
              diff_model_config
          )
      self.conditioning_key = conditioning_key
      assert self.conditioning_key in [
          None,
          "concat",
          "crossattn",
          "hybrid",
          "adm",
      ]
  
file_path: NeMo/nemo/collections/multimodal/models/text_to_image/dreambooth/util.py
imports: from functools import partial
  import numpy
  import torch
  import torch.nn
  from nemo.collections.multimodal.modules.stable_diffusion.diffusionmodules.util import extract_into_tensor, make_beta_schedule
  from nemo.collections.multimodal.parts.stable_diffusion.utils import default, exists
  from nemo.core.classes.common import Serialization
label_desc: Use image node_id 1 to create a new DiffusionWrapper object from inherited base classes: torch.nn, Serialization with example: obj = DiffusionWrapper(diff_model_config, conditioning_key)