DDIMScheduler

Denoising diffusion implicit models (DDIM).

---

source

DDIMScheduler

 DDIMScheduler (device:Union[str,torch.device],
                num_train_timesteps:int=1000, beta_start:float=0.0001,
                beta_end:float=0.02, beta_schedule:str='linear',
                input_perturbation=0.1, eta:float=0)

A Scheduler implementing (DDIM).

---

source

DDIMSchedulerOutput

 DDIMSchedulerOutput (prev_sample:torch.FloatTensor,
                      pred_original_sample:Optional[torch.FloatTensor]=Non
                      e)