• Chi's avatar
    Solve missing clip_sample implementation in FlaxDDIMScheduler. (#7017) · 46fac824
    Chi authored
    
    
    * I added a new doc string to the class. This is more flexible to understanding other developers what are doing and where it's using.
    
    * Update src/diffusers/models/unet_2d_blocks.py
    
    This changes suggest by maintener.
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Update src/diffusers/models/unet_2d_blocks.py
    
    Add suggested text
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    
    * Update unet_2d_blocks.py
    
    I changed the Parameter to Args text.
    
    * Update unet_2d_blocks.py
    
    proper indentation set in this file.
    
    * Update unet_2d_blocks.py
    
    a little bit of change in the act_fun argument line.
    
    * I run the black command to reformat style in the code
    
    * Update unet_2d_blocks.py
    
    similar doc-string add to have in the original diffusion repository.
    
    * Fix bug for mention in this issue section #6901
    
    * Update src/diffusers/schedulers/scheduling_ddim_flax.py
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    
    * Fix linter
    
    * Restore empty line
    
    ---------
    Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    46fac824
scheduling_ddim_flax.py 13 KB