(self, model, extra_args=None, eps_callback=None, info_callback=None)
| 334 | """DPM-Solver. See https://arxiv.org/abs/2206.00927.""" |
| 335 | |
| 336 | def __init__(self, model, extra_args=None, eps_callback=None, info_callback=None): |
| 337 | super().__init__() |
| 338 | self.model = model |
| 339 | self.extra_args = {} if extra_args is None else extra_args |
| 340 | self.eps_callback = eps_callback |
| 341 | self.info_callback = info_callback |
| 342 | |
| 343 | def t(self, sigma): |
| 344 | return -sigma.log() |
nothing calls this directly
no outgoing calls
no test coverage detected