(self, scale, dyn_thresh_config=None)
| 25 | """ |
| 26 | |
| 27 | def __init__(self, scale, dyn_thresh_config=None): |
| 28 | self.scale = scale |
| 29 | scale_schedule = lambda scale, sigma: scale # independent of step |
| 30 | self.scale_schedule = partial(scale_schedule, scale) |
| 31 | self.dyn_thresh = instantiate_from_config( |
| 32 | default( |
| 33 | dyn_thresh_config, |
| 34 | {"target": "sgm.modules.diffusionmodules.sampling_utils.NoDynamicThresholding"}, |
| 35 | ) |
| 36 | ) |
| 37 | |
| 38 | def __call__(self, x, sigma, scale=None): |
| 39 | x_u, x_c = x.chunk(2) |
no test coverage detected