code: def __init__(self, diffusion_model, compile_model: bool = False):
      super().__init__()
      compile = (
          torch.compile
          if (
              version.parse(torch.__version__) >= version.parse("2.0.0")
          )
          and compile_model
          else lambda x: x
      )
      self.diffusion_model = compile(diffusion_model)
  
file_path: generative-models/sgm/modules/diffusionmodules/wrappers.py
imports: import torch
  import torch.nn
  from packaging import version
label_desc: Use image node_id 1 to create a new IdentityWrapper object from inherited base classes: nn with example: obj = IdentityWrapper(diffusion_model, compile_model)