DPM Scheduler

DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models (DPM-Solver) (DPM-Solver++).

source

DPMSchedulerOutput

 DPMSchedulerOutput (prev_sample:torch.FloatTensor,
                     pred_original_sample:Optional[torch.FloatTensor]=None
                     )

source

DPMScheduler

 DPMScheduler (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, prediction_type='epsilon',
               enable_zero_terminal_snr=True, solver_order:int=2,
               **kwargs)

A Scheduler implementing (DPM-Solver++).

Back to top