code: def __init__(
      self,
      unet_type,
      num_steps=50,
      sigma_min=0.002,
      sigma_max=80,
      rho=7,
      S_churn=0,
      S_min=0,
      S_max=float("inf"),
      S_noise=1,
  ):
      super().__init__()
      self.unet_type = unet_type
      self.sigma_min = sigma_min
      self.sigma_max = sigma_max
      self.rho = rho
      self.S_churn = S_churn
      self.S_min = S_min
      self.S_max = S_max
      self.S_noise = S_noise
      self.num_steps = num_steps
  
imports: import numpy
  import torch
  from einops import rearrange
  from tqdm import tqdm
  from nemo.collections.multimodal.modules.imagen.sampler.batch_ops import batch_div, batch_mul
  from nemo.collections.multimodal.modules.imagen.sampler.continuous_ddpm import GaussianDiffusionContinuousTimes
label_desc: Use image node_id 1 to create a new EDMSampler object from inherited base classes: Sampler with example: obj = EDMSampler(unet_type, num_steps, sigma_min, sigma_max, rho, S_churn, S_min, S_max, S_noise)