class_docstr: DPM-Solver. See https://arxiv.org/abs/2206.00927.
class_docstr_tok: ["DPM-Solver",".","See","https",":","\/\/arxiv.org\/abs\/2206.00927","."]
code: def __init__(
      self,
      model,
      extra_args=None,
      eps_callback=None,
      info_callback=None,
  ):
      super().__init__()
      self.model = model
      self.extra_args = {} if extra_args is None else extra_args
      self.eps_callback = eps_callback
      self.info_callback = info_callback
  
imports: import math
  import torch
  import torchsde
  from scipy import integrate
  from torch import nn
  from torchdiffeq import odeint
  from tqdm.auto import tqdm, trange
label_desc: Use image node_id 1 to create a new DPMSolver object from inherited base classes: nn with example: obj = DPMSolver(model, extra_args, eps_callback, info_callback)