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
  
imports: import enum
  import math
  import numpy
  import torch
  import torch
  from tqdm import tqdm
  from TTS.tts.layers.tortoise.dpm_solver import DPM_Solver, NoiseScheduleVP, model_wrapper
label_desc: Use image node_id 1 to create a new _WrappedAutoregressiveModel object  with example: obj = _WrappedAutoregressiveModel(model, timestep_map, rescale_timesteps, original_num_steps)