• Pedro Cuenca's avatar
    Deprecate `predict_epsilon` (#1393) · d52388f4
    Pedro Cuenca authored
    
    
    * Adapt ddpm, ddpmsolver to prediction_type.
    
    * Deprecate predict_epsilon in __init__.
    
    * Bring FlaxDDIMScheduler up to date with DDIMScheduler.
    
    * Set prediction_type as an ivar for consistency.
    
    * Convert pipeline_ddpm
    
    * Adapt tests.
    
    * Adapt unconditional training script.
    
    * Adapt BitDiffusion example.
    
    * Add missing kwargs in dpmsolver_multistep
    
    * Ugly workaround to accept deprecated predict_epsilon when loading
    schedulers using from_pretrained.
    
    * make style
    
    * Remove import no longer in use.
    
    * Apply suggestions from code review
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Use config.prediction_type everywhere
    
    * Add a couple of Flax prediction type tests.
    
    * make style
    
    * fix register deprecated arg
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    d52388f4
bit_diffusion.py 10.6 KB