• Dhruv Naik's avatar
    Add Flax stable diffusion img2img pipeline (#1355) · a9190bad
    Dhruv Naik authored
    
    
    * add flax img2img pipeline
    
    * update pipeline
    
    * black format file
    
    * remove argg from get_timesteps
    
    * update get_timesteps
    
    * fix bug: make use of timesteps for for_loop
    
    * black file
    
    * black, isort, flake8
    
    * update docstring
    
    * update readme
    
    * update flax img2img readme
    
    * update sd pipeline init
    
    * Update src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    
    * Update src/diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_img2img.py
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    
    * update inits
    
    * revert change
    
    * update var name to image, typo
    
    * update readme
    
    * return new t_start instead of modified timestep
    
    * black format
    
    * isort files
    
    * update docs
    
    * fix-copies
    
    * update prng_seed typing
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    a9190bad
__init__.py 5.44 KB