code: def __init__(
      self, model, timestep_map, rescale_timesteps, original_num_steps
  ):
      self.model = model
      self.timestep_map = timestep_map
      self.rescale_timesteps = rescale_timesteps
      self.original_num_steps = original_num_steps
  
file_path: improved-diffusion/improved_diffusion/respace.py
imports: import numpy
  import torch
  from .gaussian_diffusion import GaussianDiffusion
label_desc: Use image node_id 1 to create a new _WrappedModel object  with example: obj = _WrappedModel(model, timestep_map, rescale_timesteps, original_num_steps)