(self, scale, dyn_thresh_config=None)
| 11 | """ |
| 12 | |
| 13 | def __init__(self, scale, dyn_thresh_config=None): |
| 14 | scale_schedule = lambda scale, sigma: scale # independent of step |
| 15 | self.scale_schedule = partial(scale_schedule, scale) |
| 16 | self.dyn_thresh = instantiate_from_config( |
| 17 | default( |
| 18 | dyn_thresh_config, |
| 19 | { |
| 20 | "target": "sgm.modules.diffusionmodules.sampling_utils.NoDynamicThresholding" |
| 21 | }, |
| 22 | ) |
| 23 | ) |
| 24 | |
| 25 | def __call__(self, x, sigma): |
| 26 | x_u, x_c = x.chunk(2) |
nothing calls this directly
no test coverage detected