• Kashif Rasul's avatar
    [Pytorch] Pytorch only schedulers (#534) · bd8df2da
    Kashif Rasul authored
    
    
    * pytorch only schedulers
    
    * fix style
    
    * remove match_shape
    
    * pytorch only ddpm
    
    * remove SchedulerMixin
    
    * remove numpy from karras_ve
    
    * fix types
    
    * remove numpy from lms_discrete
    
    * remove numpy from pndm
    
    * fix typo
    
    * remove mixin and numpy from sde_vp and ve
    
    * remove remaining tensor_format
    
    * fix style
    
    * sigmas has to be torch tensor
    
    * removed set_format in readme
    
    * remove set format from docs
    
    * remove set_format from pipelines
    
    * update tests
    
    * fix typo
    
    * continue to use mixin
    
    * fix imports
    
    * removed unsed imports
    
    * match shape instead of assuming image shapes
    
    * remove import typo
    
    * update call to add_noise
    
    * use math instead of numpy
    
    * fix t_index
    
    * removed commented out numpy tests
    
    * timesteps needs to be discrete
    
    * cast timesteps to int in flax scheduler too
    
    * fix device mismatch issue
    
    * small fix
    
    * Update src/diffusers/schedulers/scheduling_pndm.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    bd8df2da
scheduling_utils.py 1.26 KB