DPM Scheduler

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

---

source

DPMSchedulerOutput

def DPMSchedulerOutput(
    prev_sample:FloatTensor, pred_original_sample:Optional=None
)->None:

---

source

DPMScheduler

def DPMScheduler(
    device:Union, num_train_timesteps:int=1000, beta_start:float=0.0001, beta_end:float=0.02,
    beta_schedule:str='linear', input_perturbation:float=0.1, prediction_type:str='epsilon',
    enable_zero_terminal_snr:bool=True, solver_order:int=2, kwargs:VAR_KEYWORD
)->None:

A Scheduler implementing (DPM-Solver++).