(
self,
weighting_config,
scaling_config,
num_idx,
discretization_config,
do_append_zero=False,
quantize_c_noise=True,
flip=True,
)
| 40 | |
| 41 | class DiscreteDenoiser(Denoiser): |
| 42 | def __init__( |
| 43 | self, |
| 44 | weighting_config, |
| 45 | scaling_config, |
| 46 | num_idx, |
| 47 | discretization_config, |
| 48 | do_append_zero=False, |
| 49 | quantize_c_noise=True, |
| 50 | flip=True, |
| 51 | ): |
| 52 | super().__init__(weighting_config, scaling_config) |
| 53 | sigmas = instantiate_from_config(discretization_config)(num_idx, do_append_zero=do_append_zero, flip=flip) |
| 54 | self.sigmas = sigmas |
| 55 | # self.register_buffer("sigmas", sigmas) |
| 56 | self.quantize_c_noise = quantize_c_noise |
| 57 | |
| 58 | def sigma_to_idx(self, sigma): |
| 59 | dists = sigma - self.sigmas.to(sigma.device)[:, None] |
nothing calls this directly
no test coverage detected