(self, denoiser, x, cond, uc=None, num_steps=None)
| 142 | return x |
| 143 | |
| 144 | def __call__(self, denoiser, x, cond, uc=None, num_steps=None): |
| 145 | x, s_in, sigmas, num_sigmas, cond, uc = self.prepare_sampling_loop(x, cond, uc, num_steps) |
| 146 | |
| 147 | for i in self.get_sigma_gen(num_sigmas): |
| 148 | x = self.sampler_step( |
| 149 | s_in * sigmas[i], |
| 150 | s_in * sigmas[i + 1], |
| 151 | denoiser, |
| 152 | x, |
| 153 | cond, |
| 154 | uc, |
| 155 | self.s_noise, |
| 156 | ) |
| 157 | |
| 158 | return x |
| 159 | |
| 160 | |
| 161 | class AncestralSampler(SingleStepDiffusionSampler): |
nothing calls this directly
no test coverage detected