(self, edit_ratio=0.5, *args, **kwargs)
| 440 | |
| 441 | class SdeditEDMSampler(EulerEDMSampler): |
| 442 | def __init__(self, edit_ratio=0.5, *args, **kwargs): |
| 443 | super().__init__(*args, **kwargs) |
| 444 | |
| 445 | self.edit_ratio = edit_ratio |
| 446 | |
| 447 | def __call__(self, denoiser, image, randn, cond, uc=None, num_steps=None, edit_ratio=None): |
| 448 | randn_unit = randn.clone() |