Method __init__

src/utils/inversion.py:8–12

(self, model, scheduler, NUM_DDIM_STEPS)

class DDIMInversion:
    def __init__(self, model, scheduler, NUM_DDIM_STEPS):
        self.model = model
        self.scheduler = scheduler
        self.scheduler.set_timesteps(NUM_DDIM_STEPS)
        self.NUM_DDIM_STEPS = NUM_DDIM_STEPS

    def next_step(self, model_output: Union[torch.FloatTensor, np.ndarray], timestep: int,
                  sample: Union[torch.FloatTensor, np.ndarray]):