DDPMScheduler

Denoising diffusion probabilistic models (DDPM): reverse beta is fixed and diagonal.


source

DDPMScheduler

 DDPMScheduler (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)

A Scheduler implementing (DDPM)


source

DDPMSchedulerOutput

 DDPMSchedulerOutput (prev_sample:torch.FloatTensor,
                      pred_original_sample:Optional[torch.FloatTensor]=Non
                      e)
Back to top