• Jonathan Whitaker's avatar
    Adding pred_original_sample to SchedulerOutput for some samplers (#614) · 91db8189
    Jonathan Whitaker authored
    * Adding pred_original_sample to SchedulerOutput of DDPMScheduler, DDIMScheduler, LMSDiscreteScheduler, KarrasVeScheduler step methods so we can access the predicted denoised outputs
    
    * Gave DDPMScheduler, DDIMScheduler and LMSDiscreteScheduler their own output dataclasses so the default SchedulerOutput in scheduling_utils does not need pred_original_sample as an optional extra
    
    * Reordered library imports to follow standard
    
    * didnt get import order quite right apparently
    
    * Forgot to change name of LMSDiscreteSchedulerOutput
    
    * Aha, needed some extra libs for make style to fully work
    91db8189
scheduling_ddim.py 13.6 KB