• dg845's avatar
    Improve LCMScheduler (#5681) · aab6de22
    dg845 authored
    
    
    * Refactor LCMScheduler.step such that prev_sample == denoised at the last timestep in the schedule.
    
    * Make timestep scaling when calculating boundary conditions configurable.
    
    * Reparameterize timestep_scaling to be a multiplicative rather than division scaling.
    
    * make style
    
    * fix dtype conversion
    
    * make style
    
    ---------
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    aab6de22
scheduling_lcm.py 25.1 KB