Method find_step

dynamic_conditioning.py:30–34

(self, sigma: torch.Tensor, step_sigmas: torch.Tensor)

        return (model,)

    def find_step(self, sigma: torch.Tensor, step_sigmas: torch.Tensor):
        for i, step_sigma in enumerate(step_sigmas):
            if step_sigma <= sigma:
                return i
        return len(step_sigmas) - 1

    def forward(
        self, sigma: torch.Tensor, denoise_mask: torch.Tensor, extra_options: dict