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