(self, x, sigma, step_index, scale=None)
| 125 | ) |
| 126 | |
| 127 | def __call__(self, x, sigma, step_index, scale=None): |
| 128 | x_u, x_c = x.chunk(2) |
| 129 | |
| 130 | if isinstance(step_index, torch.Tensor): |
| 131 | step_index = step_index.item() |
| 132 | scale_value = self.scale_schedule(sigma, step_index) |
| 133 | x_pred = self.dyn_thresh(x_u, x_c, scale_value) |
| 134 | return x_pred |
| 135 | |
| 136 | class IdentityGuider: |
| 137 | def __call__(self, x, sigma): |
nothing calls this directly
no test coverage detected