(self, model, power, only_first_frame)
| 21 | INIT = False |
| 22 | |
| 23 | def apply(self, model, power, only_first_frame): |
| 24 | self.only_first_frame = only_first_frame |
| 25 | self.power = power |
| 26 | model = model.clone() |
| 27 | model.set_model_denoise_mask_function(self.forward) |
| 28 | return (model,) |
| 29 | |
| 30 | def find_step(self, sigma: torch.Tensor, step_sigmas: torch.Tensor): |
| 31 | for i, step_sigma in enumerate(step_sigmas): |
no outgoing calls
no test coverage detected